chore: add tests

Author: VinciGit00

scrapegraph-py/tests/test_crawl_polling.py

@@ -9,6 +9,7 @@
 """
 
 import json
+import time
 from unittest.mock import patch
 from uuid import uuid4
 
@@ -58,11 +59,11 @@ def mock_founders_result():
     return {
         "founders": [
             {
-                "name": "Marco Perini",
+                "name": "Marco Vinci",
                 "title": "Co-founder & CEO",
                 "bio": "AI researcher and entrepreneur",
-                "linkedin": "https://linkedin.com/in/marco-perini",
-                "twitter": "https://twitter.com/marco_perini",
+                "linkedin": "https://linkedin.com/in/marco-vinci",
+                "twitter": "https://twitter.com/marco_vinci",
             },
             {
                 "name": "Lorenzo Padoan",
@@ -441,7 +442,7 @@ def test_crawl_polling_with_timing(
 
         # Simulate the polling loop from crawl_example.py
         for i in range(60):  # Same as in the example
-            # time.sleep(5) - mocked out
+            time.sleep(5)  # This will be mocked out
             result = client.get_crawl(crawl_id)
             if result.get("status") == "success" and result.get("result"):
                 # Verify successful completion

scrapegraph-py/tests/utils.py

@@ -0,0 +1,11 @@
+"""
+Utility functions for tests
+"""
+import uuid
+
+
+def generate_mock_api_key() -> str:
+    """Generate a mock API key for testing purposes"""
+    # Generate a realistic looking API key format: sgai-{uuid}
+    mock_uuid = str(uuid.uuid4()).replace('-', '')
+    return f"sgai-{mock_uuid}"