update extractor test script

pull/13/head
Mike Fährmann 8 years ago
parent c84e975dcb
commit 331b413c60
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88

@ -68,6 +68,7 @@ class KissmangaChapterExtractor(KissmangaExtractor):
def items(self): def items(self):
page = self.request(self.url).text page = self.request(self.url).text
print(page)
data = self.get_job_metadata(page) data = self.get_job_metadata(page)
imgs = self.get_image_urls(page) imgs = self.get_image_urls(page)
data["count"] = len(imgs) data["count"] = len(imgs)

@ -48,18 +48,21 @@ def _generate_test(extr, tcase):
# enable selective testing for direct calls # enable selective testing for direct calls
extractors = extractor.extractors() skip = ["exhentai", "mangafox"]
if __name__ == '__main__' and len(sys.argv) > 1: if __name__ == '__main__' and len(sys.argv) > 1:
extractors = [ extractors = [
extr for extr in extractors extr for extr in extractor.extractors()
if extr.category in sys.argv if extr.category in sys.argv
] ]
del sys.argv[1:] del sys.argv[1:]
else:
extractors = [
extr for extr in extractor.extractors()
if extr.category not in skip
]
skip = ["exhentai", "kissmanga", "mangafox"]
for extr in extractors: for extr in extractors:
if extr.category in skip:
continue
if hasattr(extr, "test") and extr.test: if hasattr(extr, "test") and extr.test:
name = "test_" + extr.__name__ + "_" name = "test_" + extr.__name__ + "_"
for num, tcase in enumerate(extr.test, 1): for num, tcase in enumerate(extr.test, 1):

Loading…
Cancel
Save