1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-22 10:42:34 +01:00
gallery-dl/test/test_extractors.py
2015-12-15 23:45:40 +01:00

35 lines
1.0 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2015 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, jobs, config
class TestExtractors(unittest.TestCase):
def test_extractors(self):
config.load()
for extr in extractor.extractors():
if not hasattr(extr, "test"):
continue
print(extr)
for url, result in extr.test:
print(url)
self.run_test(url, result)
def run_test(self, url, result):
hjob = jobs.HashJob(url)
hjob.run()
if "url" in result:
self.assertEqual(hjob.hash_url.hexdigest(), result["url"])
if "keyword" in result:
self.assertEqual(hjob.hash_keyword.hexdigest(), result["keyword"])
if __name__ == '__main__':
unittest.main(warnings='ignore')