From e30ada162d08d6b0751ac515a7da7dc229ca44f4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= Date: Sun, 26 May 2019 20:22:04 +0200 Subject: [PATCH] fix cookie tests update _get_extractor(): - always return an Extractor instance with a _login_impl() method - use Extractor.from_url() --- test/test_cookies.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/test_cookies.py b/test/test_cookies.py index 39dfb0ca..a786df63 100644 --- a/test/test_cookies.py +++ b/test/test_cookies.py @@ -17,7 +17,6 @@ from os.path import join import gallery_dl.config as config import gallery_dl.extractor as extractor -from gallery_dl.extractor.message import Message CKEY = ("cookies",) @@ -121,9 +120,10 @@ class TestCookieLogin(unittest.TestCase): def _get_extractor(category): - for msg in extractor.find("test:" + category): - if msg[0] == Message.Queue: - return extractor.find(msg[1]) + for extr in extractor.extractors(): + if extr.category == category and hasattr(extr, "_login_impl"): + url = next(extr._get_tests())[0] + return extr.from_url(url) if __name__ == "__main__":