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

test archive-id creation and uniqueness

Browse Source
This commit is contained in:
Mike Fährmann 2018-02-12 23:02:09 +01:00
parent 3cec533c28
commit be3ea4425d
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
gallery_dl/job.py
View File

@ -344,8 +344,10 @@ class TestJob(DownloadJob):
self.content = content
self.list_url = []
self.list_keyword = []
self.list_archive = []
self.hash_url = hashlib.sha1()
self.hash_keyword = hashlib.sha1()
self.hash_archive = hashlib.sha1()
self.hash_content = hashlib.sha1()
if content:
self.fileobj = self.HashIO(self.hash_content)
@ -358,6 +360,7 @@ class TestJob(DownloadJob):
def handle_url(self, url, keywords):
self.update_url(url)
self.update_keyword(keywords)
self.update_archive(keywords)
self.update_content(url)
def handle_urllist(self, urls, keywords):
@ -382,6 +385,12 @@ class TestJob(DownloadJob):
self.hash_keyword.update(
json.dumps(kwdict, sort_keys=True).encode())
def update_archive(self, kwdict):
"""Update the archive-id hash"""
archive_id = self.extractor.archive_fmt.format_map(kwdict)
self.list_archive.append(archive_id)
self.hash_archive.update(archive_id.encode())
def update_content(self, url):
"""Update the content hash"""
if self.content:

4
test/test_extractors.py
View File

@ -68,6 +68,10 @@ class TestExtractors(unittest.TestCase):
pass
raise
# test archive-id uniqueness
self.assertEqual(len(set(tjob.list_archive)), len(tjob.list_archive))
# test extraction results
if "url" in result:
self.assertEqual(result["url"], tjob.hash_url.hexdigest())