allow multiple hashes for content tests

2020-01-15 23:46:37 +01:00
parent f498a9057f
commit 174117f827
2 changed files with 9 additions and 3 deletions
--- a/test/test_results.py
+++ b/test/test_results.py
@@ -109,7 +109,12 @@ class TestExtractorResults(unittest.TestCase):
            self.assertEqual(result["url"], tjob.url_hash.hexdigest())

        if "content" in result:
-            self.assertEqual(result["content"], tjob.content_hash.hexdigest())
+            expected = result["content"]
+            digest = tjob.content_hash.hexdigest()
+            if isinstance(expected, str):
+                self.assertEqual(digest, expected, "content")
+            else:  # assume iterable
+                self.assertIn(digest, expected, "content")

        if "keyword" in result:
            expected = result["keyword"]