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

make extractor unittest discoverable

Browse Source
This commit is contained in:
Mike Fährmann 2017-01-09 12:27:20 +01:00
parent 4a8d74973c
commit a1fb46d984
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
test/test_extractors.py
View File

@ -1,14 +1,15 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2015, 2016 Mike Fährmann
# Copyright 2015-2017 Mike Fährmann
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
import unittest
from gallery_dl import extractor, job, config, cache
from gallery_dl import extractor, job, config
class TestExtractors(unittest.TestCase):
@ -16,7 +17,7 @@ class TestExtractors(unittest.TestCase):
config.load()
config.set(("cache", "file"), ":memory:")
def run_test(self, extr, url, result):
def _run_test(self, extr, url, result):
hjob = job.HashJob(url, "content" in result)
self.assertEqual(extr, hjob.extractor.__class__)
if "exception" in result:
@ -31,24 +32,26 @@ class TestExtractors(unittest.TestCase):
self.assertEqual(hjob.hash_content.hexdigest(), result["content"])
def generate_test(extr):
# dynamically genetate tests
def _generate_test(extr, tcase):
def test(self):
print("\n", extr.__name__, sep="")
for url, result in extr.test:
print(url)
self.run_test(extr, url, result)
url, result = tcase
print("\n", url, sep="")
self._run_test(extr, url, result)
return test
for extr in extractor.extractors():
# disable extractors that require authentication for now
if hasattr(extr, "login"):
continue
if hasattr(extr, "test") and extr.test:
name = "test_" + extr.__name__ + "_"
for num, tcase in enumerate(extr.test, 1):
test = _generate_test(extr, tcase)
test.__name__ = name + str(num)
setattr(TestExtractors, test.__name__, test)
del test
if __name__ == '__main__':
import sys
extractors = extractor.extractors()
if len(sys.argv) > 1:
extractors = filter(lambda x: x.category in sys.argv, extractors)
for extr in extractors:
if hasattr(extr, "test") and extr.test:
name = "test_" + extr.__name__
test = generate_test(extr)
setattr(TestExtractors, name, test)
del sys.argv[1:]
unittest.main(warnings='ignore')