Skip to content

Commit d5c22e3

Browse files
committed
test loaders: fix file:// prefix
1 parent 1eb0f96 commit d5c22e3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

pywb/utils/test/test_loaders.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
100
2626
2727
# no length specified, read full amount requested
28-
>>> len(BlockLoader().load('file:' + pathname2url(test_cdx_dir + 'example.cdx'), 0, -1).read(400))
28+
>>> len(BlockLoader().load(to_local_url(test_cdx_dir + 'example.cdx'), 0, -1).read(400))
2929
400
3030
3131
# HMAC Cookie Maker
@@ -61,6 +61,7 @@
6161

6262
#=================================================================
6363
import re
64+
import os
6465
from io import BytesIO
6566
from pywb.utils.loaders import BlockLoader, HMACCookieMaker
6667
from pywb.utils.loaders import LimitReader, extract_client_cookie
@@ -83,7 +84,9 @@ def seek_read_full(seekable_reader, offset):
8384
seekable_reader.readline() #skip
8485
return seekable_reader.readline()
8586

86-
87+
def to_local_url(filename):
88+
filename = os.path.abspath(filename)
89+
return 'file://' + pathname2url(filename)
8790

8891
if __name__ == "__main__":
8992
import doctest

0 commit comments

Comments
 (0)