chg: Fix tests for generix types.

Lookyloo · Dec 13, 2024 · 0850089 · 0850089
1 parent b842da2
commit 0850089
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/tests/capture_samples b/tests/capture_samples
diff --git a/tests/simple_test.py b/tests/simple_test.py
@@ -131,8 +131,8 @@ def test_rebuild_url_partial_double_slash(self) -> None:
         self.assertEqual(rebuild_url_double_slash, 'https://www.youtube.com/watch?v=iwGFalTRHDA')
 
     def test_hostname_tree_features(self) -> None:
-        self.assertEqual(self.http_redirect_ct.root_hartree.hostname_tree.features, {'name', 'js', 'html', 'pdf', 'json', 'text', 'video', 'css', 'iframe', 'http_content', 'https_content', 'support', 'dist', 'octet_stream', 'font', 'redirect',
-                                                                                     'unknown_mimetype', 'contains_rendered_urlnode', 'urls', 'uuid', 'redirect_to_nothing', 'unset_mimetype', 'image'})
+        self.assertEqual(self.http_redirect_ct.root_hartree.hostname_tree.features, {'name', 'http_content', 'https_content', 'support', 'dist',
+                                                                                     'contains_rendered_urlnode', 'urls', 'uuid'})
         self.assertTrue('meta_refresh' in self.http_redirect_ct.root_hartree.url_tree.external_ressources)
         self.assertEqual(self.http_redirect_ct.root_hartree.url_tree.external_ressources['meta_refresh'][0], 'https://www.youtube.com/watch?v=iwGFalTRHDA')
 
@@ -195,9 +195,6 @@ def test_urls_in_rendered_page(self) -> None:
                 'https://policies.google.com/terms?hl=en&utm_soure=ucb']
         )
 
-    def test_iframe_feature(self) -> None:
-        self.assertTrue('iframe' in self.iframe_ct.root_hartree.hostname_tree.features)
-
     def test_iframe_capture_name(self) -> None:
         self.assertEqual(self.iframe_ct.root_hartree.har.initial_title, '!! No title found !!')