Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-25 04:02:32 +01:00
Code Issues Releases Wiki Activity

fix downloader tests

Browse Source
This commit is contained in:
Mike Fährmann 2018-06-07 22:27:36 +02:00
parent 37d97ff02c
commit b344f2290f
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/test_downloader.py
View File

@ -54,9 +54,8 @@ class TestDownloaderBase(unittest.TestCase):
pathfmt.set_keywords(kwdict)
if content:
path = pathfmt.realpath + (".part" if part else "")
mode = "w" + ("b" if isinstance(content, bytes) else "")
with open(path, mode) as file:
with pathfmt.open(mode) as file:
file.write(content)
return pathfmt
@ -65,20 +64,23 @@ class TestDownloaderBase(unittest.TestCase):
extension, expected_extension=None):
pathfmt = self._prepare_destination(input, extension=extension)
success = self.downloader.download(url, pathfmt)
path = pathfmt.realpath
# test successful download
self.assertTrue(success, "downloading '{}' failed".format(url))
# test content
mode = "r" + ("b" if isinstance(output, bytes) else "")
with open(path, mode) as file:
with pathfmt.open(mode) as file:
content = file.read()
self.assertEqual(content, output)
# test filename extension
self.assertEqual(
os.path.splitext(path)[1][1:],
pathfmt.keywords["extension"],
expected_extension,
)
self.assertEqual(
os.path.splitext(pathfmt.realpath)[1][1:],
expected_extension,
)