backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

test loaders: fix file:// prefix

Browse Source
This commit is contained in:
Ilya Kreymer 2015-01-10 15:27:45 -08:00
parent 1eb0f96f92
commit d5c22e3649
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pywb/utils/test/test_loaders.py
View File

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