Fix mock endpoint tests

Author: tanzilahmed0

backend/tests/test_mock_endpoints.py

@@ -6,6 +6,7 @@
 from fastapi.testclient import TestClient
 
 from main import app
+from middleware.auth_middleware import verify_token
 from models.user import GoogleOAuthData, UserInDB
 from services.auth_service import AuthService
 
@@ -15,6 +16,11 @@
 auth_service = AuthService()
 
 
+def mock_verify_token():
+    """Mock verify_token that returns user_001"""
+    return "user_001"
+
+
 @pytest.fixture
 def sample_user():
     """Sample user for testing - uses UUID that matches our mock project ownership"""
@@ -75,7 +81,8 @@ def test_get_current_user(test_client, sample_user, test_access_token):
 
 def test_get_projects(test_client, test_access_token):
     """Test get projects endpoint"""
-    with patch("api.projects.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/projects?page=1&limit=10",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -86,11 +93,14 @@ def test_get_projects(test_client, test_access_token):
         assert "items" in data["data"]
         assert "total" in data["data"]
         assert len(data["data"]["items"]) >= 0
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_create_project(test_client, test_access_token):
     """Test create project endpoint"""
-    with patch("api.projects.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.post(
             "/projects",
             json={"name": "Test Project", "description": "Test description"},
@@ -101,11 +111,14 @@ def test_create_project(test_client, test_access_token):
         assert data["success"] is True
         assert data["data"]["project"]["name"] == "Test Project"
         assert "upload_url" in data["data"]
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_get_project(test_client, test_access_token):
     """Test get single project endpoint"""
-    with patch("api.projects.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/projects/project_001",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -115,11 +128,14 @@ def test_get_project(test_client, test_access_token):
         assert data["success"] is True
         assert data["data"]["id"] == "project_001"
         assert data["data"]["name"] == "Sales Data Analysis"
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_csv_preview(test_client, test_access_token):
     """Test CSV preview endpoint"""
-    with patch("api.chat.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/chat/project_001/preview",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -130,11 +146,14 @@ def test_csv_preview(test_client, test_access_token):
         assert "columns" in data["data"]
         assert "sample_data" in data["data"]
         assert len(data["data"]["columns"]) > 0
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_send_message(test_client, test_access_token):
     """Test send chat message endpoint"""
-    with patch("api.chat.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.post(
             "/chat/project_001/message",
             json={"message": "Show me total sales by product"},
@@ -146,11 +165,14 @@ def test_send_message(test_client, test_access_token):
         assert "message" in data["data"]
         assert "result" in data["data"]
         assert data["data"]["result"]["result_type"] in ["table", "chart", "summary"]
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_query_suggestions(test_client, test_access_token):
     """Test query suggestions endpoint"""
-    with patch("api.chat.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/chat/project_001/suggestions",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -160,12 +182,14 @@ def test_query_suggestions(test_client, test_access_token):
         assert data["success"] is True
         assert len(data["data"]) > 0
         assert all("text" in suggestion for suggestion in data["data"])
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_unauthorized_access(test_client):
     """Test that endpoints require authentication"""
     response = test_client.get("/projects")
-    assert response.status_code == 403
+    assert response.status_code == 401
 
 
 def test_invalid_token(test_client):
@@ -210,7 +234,8 @@ def test_refresh_token(test_client, sample_user):
 
 def test_project_status(test_client, test_access_token):
     """Test project status endpoint"""
-    with patch("api.projects.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/projects/project_001/status",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -220,26 +245,31 @@ def test_project_status(test_client, test_access_token):
         assert data["success"] is True
         assert "status" in data["data"]
         assert "progress" in data["data"]
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_get_upload_url(test_client, test_access_token):
     """Test get upload URL endpoint"""
-    with patch("api.projects.verify_token"):
-        response = test_client.post(
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
+        response = test_client.get(
             "/projects/project_001/upload-url",
-            json={"filename": "new_data.csv", "content_type": "text/csv"},
             headers={"Authorization": f"Bearer {test_access_token}"},
         )
         assert response.status_code == 200
         data = response.json()
         assert data["success"] is True
         assert "upload_url" in data["data"]
-        assert "object_path" in data["data"]
+        assert "upload_fields" in data["data"]
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_get_messages(test_client, test_access_token):
     """Test get chat messages endpoint"""
-    with patch("api.chat.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/chat/project_001/messages",
             headers={"Authorization": f"Bearer {test_access_token}"},
@@ -249,6 +279,8 @@ def test_get_messages(test_client, test_access_token):
         assert data["success"] is True
         assert "items" in data["data"]
         assert len(data["data"]["items"]) >= 0
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_invalid_google_token(test_client):
@@ -265,17 +297,21 @@ def test_invalid_google_token(test_client):
 
 def test_project_not_found(test_client, test_access_token):
     """Test project not found error"""
-    with patch("api.projects.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.get(
             "/projects/nonexistent_project",
             headers={"Authorization": f"Bearer {test_access_token}"},
         )
         assert response.status_code == 404
+    finally:
+        app.dependency_overrides.clear()
 
 
 def test_chart_query_response(test_client, test_access_token):
     """Test chart query response type"""
-    with patch("api.chat.verify_token"):
+    app.dependency_overrides[verify_token] = mock_verify_token
+    try:
         response = test_client.post(
             "/chat/project_001/message",
             json={"message": "show me a chart"},
@@ -285,3 +321,5 @@ def test_chart_query_response(test_client, test_access_token):
         data = response.json()
         assert data["data"]["result"]["result_type"] == "chart"
         assert "chart_config" in data["data"]["result"]
+    finally:
+        app.dependency_overrides.clear()