From a413f5db697fd7f568aa3dd8e9d1406176cb924c Mon Sep 17 00:00:00 2001 From: gabrielgio Date: Sun, 18 Jul 2021 21:04:03 +0200 Subject: ref: Add test for GReddit Add test for GReddit. It will check for all necessary variable to be able to login on reddit as it is required by the praw library. Also add mock to the `youtube-dl` download function. It will make the test run faster. --- importer/providers/g_reddit.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'importer') diff --git a/importer/providers/g_reddit.py b/importer/providers/g_reddit.py index 53ee5df..e0341e0 100644 --- a/importer/providers/g_reddit.py +++ b/importer/providers/g_reddit.py @@ -5,6 +5,12 @@ from importer.providers.raw_image_base import RawImageProviderBase class GReddit(RawImageProviderBase): regex = "^.*www.reddit.com/gallery.*$" + _TEST = [{ + "url": "https://www.reddit.com/gallery/mik7c9", + "paths": ['source_hlokpsyhgrq61.jpg', 'source_n31c2y7igrq61.jpg', 'source_7eg0o76igrq61.jpg', + 'source_whl12jbigrq61.jpg', 'source_4uok762igrq61.jpg', 'source_t3pgm64igrq61.jpg', + 'source_ymc4hv9igrq61.jpg'] + }] def __init__(self, url: str, reddit: Reddit): super(GReddit, self).__init__(url) @@ -17,3 +23,4 @@ class GReddit(RawImageProviderBase): url = value['s']['u'] path = self._download_raw_file(url) self.paths.append(path) + self.downloaded = True -- cgit v1.2.3