diff options
Diffstat (limited to 'test/test_extractor.py')
| -rw-r--r-- | test/test_extractor.py | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/test/test_extractor.py b/test/test_extractor.py index abf122b..e622fa8 100644 --- a/test/test_extractor.py +++ b/test/test_extractor.py @@ -105,15 +105,23 @@ class TestExtractorModule(unittest.TestCase): def test_categories(self): for result in results.all(): url = result["#url"] - base, cat, sub = result["#category"] + cls = result["#class"] try: - extr = result["#class"].from_url(url) + extr = cls.from_url(url) except ImportError as exc: if exc.name in ("youtube_dl", "yt_dlp"): - print("Skipping '{}' category checks".format(cat)) + print("Skipping '{}' category checks".format(cls.category)) continue raise self.assertTrue(extr, url) + + categories = result.get("#category") + if categories: + base, cat, sub = categories + else: + cat = cls.category + sub = cls.subcategory + base = cls.basecategory self.assertEqual(extr.category, cat, url) self.assertEqual(extr.subcategory, sub, url) self.assertEqual(extr.basecategory, base, url) @@ -243,9 +251,12 @@ class TestExtractorWait(unittest.TestCase): def test_wait_until_datetime(self): extr = extractor.find("generic:https://example.org/") - until = datetime.utcnow() + timedelta(seconds=5) + until = util.datetime_utcnow() + timedelta(seconds=5) until_local = datetime.now() + timedelta(seconds=5) + if not until.microsecond: + until = until.replace(microsecond=until_local.microsecond) + with patch("time.sleep") as sleep, patch.object(extr, "log") as log: extr.wait(until=until) |