From 4d9a494e3511f8238aacd262b93705ef63dec68c Mon Sep 17 00:00:00 2001 From: Karim shoair Date: Sat, 23 Nov 2024 15:55:50 +0200 Subject: [PATCH] Covering more parts in StealthyFetcher tests --- tests/fetchers/test_camoufox.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/fetchers/test_camoufox.py b/tests/fetchers/test_camoufox.py index 0a75c09..5a0fe17 100644 --- a/tests/fetchers/test_camoufox.py +++ b/tests/fetchers/test_camoufox.py @@ -36,6 +36,7 @@ class TestStealthyFetcher(unittest.TestCase): def test_waiting_selector(self): """Test if waiting for a selector make page does not finish loading or not""" self.assertEqual(self.fetcher.fetch(self.html_url, wait_selector='h1').status, 200) + self.assertEqual(self.fetcher.fetch(self.html_url, wait_selector='h1', wait_selector_state='visible').status, 200) def test_cookies_loading(self): """Test if cookies are set after the request""" @@ -56,6 +57,7 @@ class TestStealthyFetcher(unittest.TestCase): self.assertEqual(self.fetcher.fetch(self.html_url, block_webrtc=True, allow_webgl=True).status, 200) self.assertEqual(self.fetcher.fetch(self.html_url, block_webrtc=False, allow_webgl=True).status, 200) self.assertEqual(self.fetcher.fetch(self.html_url, block_webrtc=True, allow_webgl=False).status, 200) + self.assertEqual(self.fetcher.fetch(self.html_url, extra_headers={'ayo': ''}, os_randomize=True).status, 200) def test_infinite_timeout(self): """Test if infinite timeout breaks the code or not"""