1
0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-15 00:03:28 +01:00

test loaders: fix file:// prefix

This commit is contained in:
Ilya Kreymer 2015-01-10 15:27:45 -08:00
parent 1eb0f96f92
commit d5c22e3649

View File

@ -25,7 +25,7 @@ True
100
# no length specified, read full amount requested
>>> len(BlockLoader().load('file:' + pathname2url(test_cdx_dir + 'example.cdx'), 0, -1).read(400))
>>> len(BlockLoader().load(to_local_url(test_cdx_dir + 'example.cdx'), 0, -1).read(400))
400
# HMAC Cookie Maker
@ -61,6 +61,7 @@ True
#=================================================================
import re
import os
from io import BytesIO
from pywb.utils.loaders import BlockLoader, HMACCookieMaker
from pywb.utils.loaders import LimitReader, extract_client_cookie
@ -83,7 +84,9 @@ def seek_read_full(seekable_reader, offset):
seekable_reader.readline() #skip
return seekable_reader.readline()
def to_local_url(filename):
filename = os.path.abspath(filename)
return 'file://' + pathname2url(filename)
if __name__ == "__main__":
import doctest