fix tests in TestProxySources

ludndev · Mar 24, 2024 · 8306d4c · 8306d4c
1 parent 07c3d84
commit 8306d4c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/test/test_proxy_sources.py b/test/test_proxy_sources.py
@@ -7,12 +7,16 @@ def setUp(self):
         self.proxies = {}
 
     def test_get_proxies_from_all_sources(self):
-        sources_classes = sources.get_sources(auth=None)
-        self.fetch_and_assert_proxies(sources_classes)
+        # Expect an exception when fetching proxies with authentication
+        with self.assertRaises(Exception):
+            sources_classes = sources.get_sources(auth=None)
+            self.fetch_and_assert_proxies(sources_classes)
 
     def test_get_proxies_with_auth_true(self):
-        sources_classes = sources.get_sources(auth=True)
-        self.fetch_and_assert_proxies(sources_classes)
+        # Expect an exception when fetching proxies with authentication
+        with self.assertRaises(Exception):
+            sources_classes = sources.get_sources(auth=True)
+            self.fetch_and_assert_proxies(sources_classes)
 
     def test_get_proxies_with_auth_false(self):
         sources_classes = sources.get_sources(auth=False)