temp

pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+pythonpath = .
+testpaths = tests
+python_files = test_*.py

tests/conftest.py

@@ -0,0 +1,7 @@
+import os
+import sys
+from pathlib import Path
+
+# Add the project root directory to Python path
+project_root = str(Path(__file__).parent.parent)
+sys.path.append(project_root)

tests/test_check_fact.py

@@ -0,0 +1,161 @@
+from fastapi.testclient import TestClient
+from unittest.mock import patch, Mock
+import sys
+from pathlib import Path
+
+# Import the FastAPI app
+from main import app
+from app.models.fact_check_models import (
+    FactCheckResponse,
+    UnverifiedFactCheckResponse,
+    VerdictEnum,
+    ConfidenceEnum,
+    Source
+)
+
+# Create test client
+client = TestClient(app)
+
+def test_check_facts_missing_query():
+    """Test the endpoint with a missing query."""
+    response = client.post("/check-facts", json={})
+    assert response.status_code == 422  # Validation error
+
+def test_check_facts_short_query():
+    """Test the endpoint with a query that's too short."""
+    response = client.post("/check-facts", json={"query": "ab"})
+    assert response.status_code == 422  # Query too short
+
+def test_check_facts_valid_query_no_sources():
+    """Test the endpoint with a valid query but no sources found."""
+    test_query = "Does drinking water cure all diseases?"
+    
+    # Mock the search_websites function
+    with patch('app.api.scrap_websites.search_websites') as mock_search:
+        mock_search.return_value = {
+            "status": "no_results",
+            "urls_found": 0,
+            "verification_result": {
+                "no_sources_found": True,
+                "reason": "No relevant fact-checking sources found"
+            }
+        }
+        
+        response = client.post("/check-facts", json={"query": test_query})
+        
+        assert response.status_code == 200
+        data = response.json()
+        assert data["claim"] == test_query
+        assert data["verdict"] == VerdictEnum.UNVERIFIED
+        assert data["confidence"] == ConfidenceEnum.LOW
+        assert data["sources"] == []
+        assert "No fact-checking sources" in data["evidence"]
+
+def test_check_facts_valid_query_with_sources():
+    """Test the endpoint with a valid query and sources found."""
+    test_query = "Did NASA find aliens on Mars?"
+    mock_fact_check_response = {
+        "claims": [
+            {
+                "text": test_query,
+                "claimReview": [
+                    {
+                        "publisher": {
+                            "name": "Fact Check Organization",
+                            "site": "https://factcheck.org"
+                        },
+                        "textualRating": "False",
+                        "title": "NASA Mars Claim",
+                        "url": "https://factcheck.org/mars-claim"
+                    }
+                ]
+            }
+        ]
+    }
+    
+    # Mock both the Google Fact Check API call and search_websites
+    with patch('httpx.AsyncClient.get') as mock_get:
+        mock_get.return_value.status_code = 200
+        mock_get.return_value.json.return_value = mock_fact_check_response
+        
+        response = client.post("/check-facts", json={"query": test_query})
+        
+        assert response.status_code == 200
+        data = response.json()
+        assert data["claim"] == test_query
+        assert data["verdict"] in [v.value for v in VerdictEnum]
+        assert data["confidence"] in [c.value for c in ConfidenceEnum]
+        assert len(data["sources"]) >= 1
+        assert all(isinstance(source, dict) for source in data["sources"])
+        assert all("url" in source and "name" in source for source in data["sources"])
+
+def test_check_facts_api_error():
+    """Test the endpoint's handling of API errors."""
+    test_query = "Test query for API error"
+    
+    # Mock API error scenario
+    with patch('httpx.AsyncClient.get') as mock_get:
+        mock_get.side_effect = Exception("API Error")
+        
+        response = client.post("/check-facts", json={"query": test_query})
+        
+        assert response.status_code == 200  # We return 200 with UnverifiedFactCheckResponse
+        data = response.json()
+        assert data["verdict"] == VerdictEnum.UNVERIFIED
+        assert data["confidence"] == ConfidenceEnum.LOW
+        assert data["sources"] == []
+        assert "error" in data["evidence"].lower()
+
+def test_check_facts_missing_api_keys():
+    """Test the endpoint's handling of missing API keys."""
+    test_query = "Test query without API keys"
+    
+    # Mock missing API keys
+    with patch('app.api.fact_check.GOOGLE_API_KEY', None):
+        response = client.post("/check-facts", json={"query": test_query})
+        
+        assert response.status_code == 200  # We return 200 with UnverifiedFactCheckResponse
+        data = response.json()
+        assert data["verdict"] == VerdictEnum.UNVERIFIED
+        assert "configuration" in data["evidence"].lower()
+        assert data["sources"] == []
+
+def test_check_facts_rate_limit():
+    """Test the endpoint's handling of rate limiting."""
+    test_query = "Test query for rate limit"
+    
+    # Mock rate limit response
+    with patch('httpx.AsyncClient.get') as mock_get:
+        mock_get.return_value.status_code = 429
+        
+        response = client.post("/check-facts", json={"query": test_query})
+        
+        assert response.status_code == 200  # We return 200 with UnverifiedFactCheckResponse
+        data = response.json()
+        assert data["verdict"] == VerdictEnum.UNVERIFIED
+        assert data["confidence"] == ConfidenceEnum.LOW
+        assert data["sources"] == []
+
+def test_check_facts_empty_query():
+    """Test the endpoint with an empty query."""
+    response = client.post("/check-facts", json={"query": ""})
+    assert response.status_code == 422  # Validation error
+
+def test_check_facts_long_query():
+    """Test the endpoint with a query that exceeds maximum length."""
+    long_query = "a" * 501  # Create a string longer than max_length (500)
+    response = client.post("/check-facts", json={"query": long_query})
+    assert response.status_code == 422  # Validation error
+
+def test_check_facts_malformed_json():
+    """Test the endpoint with malformed JSON."""
+    response = client.post(
+        "/check-facts",
+        data="this is not json",
+        headers={"Content-Type": "application/json"}
+    )
+    assert response.status_code == 422  # JSON validation error
+
+if __name__ == "__main__":
+    import pytest
+    pytest.main([__file__, "-v"])