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

[tests] use fallback URLs for content tests (#3163)

Browse Source
This commit is contained in:
Mike Fährmann 2023-09-30 21:00:55 +02:00
parent b92645cd37
commit c7bd9925d9
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
2 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
test/results/tumblr.py
View File

@ -219,10 +219,21 @@ __tests__ = (
"#comment" : "high-quality images (#1344)",
"#category": ("", "tumblr", "post"),
"#class" : tumblr.TumblrPostExtractor,
"#exception" : exception.NotFoundError,
"#count" : 2,
"#sha1_content": "6bc19a42787e46e1bba2ef4aeef5ca28fcd3cd34",
},
{
"#url" : "https://k-eke.tumblr.com/post/185341184856",
"#comment" : "wrong extension returned by api (#3095)",
"#category": ("", "tumblr", "post"),
"#class" : tumblr.TumblrPostExtractor,
"#options" : {"retries": 0},
"#urls" : "https://64.media.tumblr.com/5e9d760aba24c65beaf0e72de5aae4dd/tumblr_psj5yaqV871t1ig6no1_1280.gif",
"#sha1_content": "3508d894b6cc25e364d182a8e1ff370d706965fb",
},
{
"#url" : "https://mikf123.tumblr.com/image/689860196535762944",
"#category": ("", "tumblr", "post"),

17
test/test_results.py
View File

@ -222,6 +222,8 @@ class ResultJob(job.DownloadJob):
if content:
self.fileobj = TestPathfmt(self.content_hash)
else:
self._update_content = lambda url, kwdict: None
self.format_directory = TestFormatter(
"".join(self.extractor.directory_fmt)).format_map
@ -269,10 +271,17 @@ class ResultJob(job.DownloadJob):
self.archive_hash.update(archive_id.encode())
def _update_content(self, url, kwdict):
if self.content:
scheme = url.partition(":")[0]
self.fileobj.kwdict = kwdict
self.get_downloader(scheme).download(url, self.fileobj)
self.fileobj.kwdict = kwdict
downloader = self.get_downloader(url.partition(":")[0])
if downloader.download(url, self.fileobj):
return
for num, url in enumerate(kwdict.get("_fallback") or (), 1):
self.log.warning("Trying fallback URL #%d", num)
downloader = self.get_downloader(url.partition(":")[0])
if downloader.download(url, self.fileobj):
return
class TestPathfmt():