tests

Author: VinciGit00

tests/test_credits.py

@@ -1 +1,21 @@
- 
+import unittest
+from unittest.mock import patch
+from scrapegraphaiapisdk.credits import credits
+
+class TestCredits(unittest.TestCase):
+    
+    @patch('scrapegraphaiapisdk.credits.requests.get')
+    def test_credits_success(self, mock_get):
+        mock_get.return_value.status_code = 200
+        mock_get.return_value.text = '{"credits": 100}'
+        response = credits("test_api_key")
+        self.assertEqual(response, '{"credits": 100}')
+
+    @patch('scrapegraphaiapisdk.credits.requests.get')
+    def test_credits_http_error(self, mock_get):
+        mock_get.side_effect = requests.exceptions.HTTPError
+        response = credits("test_api_key")
+        self.assertIn("HTTP error occurred", response)
+
+if __name__ == '__main__':
+    unittest.main() 

tests/test_feedback.py

@@ -17,7 +17,6 @@ def test_feedback_http_error(self, mock_post):
         response = feedback("test_api_key", "Great service!")
         self.assertIn("HTTP error occurred", response)
 
-# ... additional tests can be added here ...
 
 if __name__ == '__main__':
     unittest.main() 

tests/test_scrape.py

@@ -17,7 +17,6 @@ def test_scrape_http_error(self, mock_post):
         response = scrape("test_api_key", "http://example.com", "Extract data")
         self.assertIn("HTTP error occurred", response)
 
-# ... additional tests can be added here ...
 
 if __name__ == '__main__':
     unittest.main()