Adding tests for Fetcher class

2024-11-06 12:21:18 +02:00
parent 4400363146
commit 4222626ec1
1 changed files with 67 additions and 0 deletions
@@ -0,0 +1,67 @@
+import unittest
+import pytest_httpbin
+
+from scrapling import Fetcher
+
+
+@pytest_httpbin.use_class_based_httpbin
+class TestFetcher(unittest.TestCase):
+    def setUp(self):
+        self.fetcher = Fetcher(auto_match=False)
+        url = self.httpbin.url
+        self.status_200 = f'{url}/status/200'
+        self.status_404 = f'{url}/status/404'
+        self.status_501 = f'{url}/status/501'
+        self.basic_url = f'{url}/get'
+        self.post_url = f'{url}/post'
+        self.put_url = f'{url}/put'
+        self.delete_url = f'{url}/delete'
+        self.html_url = f'{url}/html'
+
+    def test_basic_get(self):
+        """Test doing basic get request with multiple statuses"""
+        self.assertEqual(self.fetcher.get(self.status_200).status, 200)
+        self.assertEqual(self.fetcher.get(self.status_404).status, 404)
+        self.assertEqual(self.fetcher.get(self.status_501).status, 501)
+
+    def test_get_properties(self):
+        """Test if different arguments with GET request breaks the code or not"""
+        self.assertEqual(self.fetcher.get(self.status_200, stealthy_headers=True).status, 200)
+        self.assertEqual(self.fetcher.get(self.status_200, follow_redirects=True).status, 200)
+        self.assertEqual(self.fetcher.get(self.status_200, timeout=None).status, 200)
+        self.assertEqual(
+            self.fetcher.get(self.status_200, stealthy_headers=True, follow_redirects=True, timeout=None).status,
+            200
+        )
+
+    def test_post_properties(self):
+        """Test if different arguments with POST request breaks the code or not"""
+        self.assertEqual(self.fetcher.post(self.post_url, data={'key': 'value'}).status, 200)
+        self.assertEqual(self.fetcher.post(self.post_url, data={'key': 'value'}, stealthy_headers=True).status, 200)
+        self.assertEqual(self.fetcher.post(self.post_url, data={'key': 'value'}, follow_redirects=True).status, 200)
+        self.assertEqual(self.fetcher.post(self.post_url, data={'key': 'value'}, timeout=None).status, 200)
+        self.assertEqual(
+            self.fetcher.post(self.post_url, data={'key': 'value'}, stealthy_headers=True, follow_redirects=True, timeout=None).status,
+            200
+        )
+
+    def test_put_properties(self):
+        """Test if different arguments with PUT request breaks the code or not"""
+        self.assertEqual(self.fetcher.put(self.put_url, data={'key': 'value'}).status, 200)
+        self.assertEqual(self.fetcher.put(self.put_url, data={'key': 'value'}, stealthy_headers=True).status, 200)
+        self.assertEqual(self.fetcher.put(self.put_url, data={'key': 'value'}, follow_redirects=True).status, 200)
+        self.assertEqual(self.fetcher.put(self.put_url, data={'key': 'value'}, timeout=None).status, 200)
+        self.assertEqual(
+            self.fetcher.put(self.put_url, data={'key': 'value'}, stealthy_headers=True, follow_redirects=True, timeout=None).status,
+            200
+        )
+
+    def test_delete_properties(self):
+        """Test if different arguments with DELETE request breaks the code or not"""
+        self.assertEqual(self.fetcher.delete(self.delete_url, stealthy_headers=True).status, 200)
+        self.assertEqual(self.fetcher.delete(self.delete_url, follow_redirects=True).status, 200)
+        self.assertEqual(self.fetcher.delete(self.delete_url, timeout=None).status, 200)
+        self.assertEqual(
+            self.fetcher.delete(self.delete_url, stealthy_headers=True, follow_redirects=True, timeout=None).status,
+            200
+        )