Spaces:
Sleeping
Sleeping
| from fastapi.testclient import TestClient | |
| from app.api import roadmap_routes | |
| from app.main import app | |
| from app.schemas.roadmap_schema import RoadmapResponse | |
| from app.services.roadmap_service import RoadmapServiceFailure, RoadmapServiceSuccess | |
| client = TestClient(app) | |
| def test_health_endpoint() -> None: | |
| response = client.get("/health") | |
| assert response.status_code == 200 | |
| assert response.json() == {"status": "healthy"} | |
| def test_api_health_endpoint() -> None: | |
| response = client.get("/api/health") | |
| assert response.status_code == 200 | |
| body = response.json() | |
| assert body["status"] == "healthy" | |
| assert "supabase_enabled" in body | |
| def test_generate_endpoint_success_envelope() -> None: | |
| class FakeService: | |
| async def generate(self, _request): | |
| data = RoadmapResponse.model_validate( | |
| { | |
| "goal": "Machine Learning", | |
| "duration_days": 7, | |
| "milestones": [{"title": "Kickoff", "start_day": 1, "end_day": 7}], | |
| "first_week_plan": { | |
| 1: [{"title": "Install Python", "duration_minutes": 60}], | |
| 2: [{"title": "Learn syntax", "duration_minutes": 60}], | |
| 3: [{"title": "Practice loops", "duration_minutes": 60}], | |
| 4: [{"title": "Functions intro", "duration_minutes": 60}], | |
| 5: [{"title": "Data types", "duration_minutes": 60}], | |
| 6: [{"title": "Simple exercise", "duration_minutes": 60}], | |
| 7: [{"title": "Mini project", "duration_minutes": 90}], | |
| }, | |
| } | |
| ) | |
| return RoadmapServiceSuccess(data=data) | |
| class FakeStore: | |
| async def save(self, _request, _roadmap): | |
| return True | |
| def health(self): | |
| return {"supabase_enabled": False} | |
| app.dependency_overrides[roadmap_routes.get_service] = lambda: FakeService() | |
| app.dependency_overrides[roadmap_routes.get_store] = lambda: FakeStore() | |
| response = client.post( | |
| "/api/generate", | |
| json={ | |
| "goal": "Machine Learning", | |
| "duration_days": 7, | |
| "daily_minutes": 120, | |
| "level": "beginner", | |
| }, | |
| ) | |
| app.dependency_overrides.clear() | |
| assert response.status_code == 200 | |
| body = response.json() | |
| assert body["ok"] is True | |
| assert "data" in body | |
| assert body["data"]["goal"] == "Machine Learning" | |
| def test_generate_endpoint_failure_envelope() -> None: | |
| class FakeService: | |
| async def generate(self, _request): | |
| return RoadmapServiceFailure(reason="Gemini request timed out") | |
| class FakeStore: | |
| async def save(self, _request, _roadmap): | |
| return True | |
| def health(self): | |
| return {"supabase_enabled": False} | |
| app.dependency_overrides[roadmap_routes.get_service] = lambda: FakeService() | |
| app.dependency_overrides[roadmap_routes.get_store] = lambda: FakeStore() | |
| response = client.post( | |
| "/api/generate", | |
| json={ | |
| "goal": "Machine Learning", | |
| "duration_days": 7, | |
| "daily_minutes": 120, | |
| "level": "beginner", | |
| }, | |
| ) | |
| app.dependency_overrides.clear() | |
| assert response.status_code == 200 | |
| body = response.json() | |
| assert body == {"ok": False, "reason": "Gemini request timed out"} | |