chore: add test infrastructure and update risk router

- Add vitest config and unit tests for components, api, stores - Add Playwright e2e test for user flows - Add backend test files - Update risk.py with LOD grid KDTree optimization
2026-06-09 12:54:55 +08:00
parent 8ddd8e87bb
commit f092c3c550
14 changed files with 2910 additions and 12 deletions
--- a/backend/tests/test_api.py
+++ b/backend/tests/test_api.py
@@ -0,0 +1,265 @@
+"""US-001: API endpoint tests covering all 10 routers."""
+import pytest
+from fastapi.testclient import TestClient
+
+
+class TestRootAndHealth:
+    def test_root_returns_status(self, client: TestClient):
+        resp = client.get("/")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert data["status"] == "running"
+        assert data["version"] == "1.0.0"
+
+    def test_health_check(self, client: TestClient):
+        resp = client.get("/health")
+        assert resp.status_code == 200
+        assert resp.json()["status"] == "healthy"
+
+
+class TestRiskEndpoints:
+    def test_current_risk_map(self, client: TestClient):
+        resp = client.get("/api/risk/current")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "grids" in data
+        assert isinstance(data["grids"], list)
+        assert "total_count" in data
+        assert "timestamp" in data
+        if data["grids"]:
+            g = data["grids"][0]
+            assert "grid_id" in g
+            assert "risk_value" in g
+            assert "risk_level" in g
+            assert g["risk_level"] in ("high", "medium_high", "medium", "medium_low", "low")
+
+    def test_risk_map_with_date(self, client: TestClient):
+        resp = client.get("/api/risk/map?date=20231201")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert data["total_count"] > 0
+
+    def test_risk_map_missing_date(self, client: TestClient):
+        resp = client.get("/api/risk/map?date=20990101")
+        assert resp.status_code == 404
+
+    def test_forecast_1d(self, client: TestClient):
+        resp = client.get("/api/risk/forecast/1")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "grids" in data
+
+    def test_forecast_3d(self, client: TestClient):
+        resp = client.get("/api/risk/forecast/3")
+        assert resp.status_code == 200
+
+    def test_forecast_7d(self, client: TestClient):
+        resp = client.get("/api/risk/forecast/7")
+        assert resp.status_code == 200
+
+    def test_stats(self, client: TestClient):
+        resp = client.get("/api/risk/stats")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "total_grids" in data
+        assert "avg_risk" in data
+        assert "distribution" in data
+        assert "high_risk_count" in data
+        dist = data["distribution"]
+        for k in ("high", "medium_high", "medium", "medium_low", "low"):
+            assert k in dist
+
+    def test_lod_grid(self, client: TestClient):
+        resp = client.get("/api/risk/lod-grid?zoom=10&forecast_day=1")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "lod" in data
+        assert "grids" in data
+        assert "total_count" in data
+
+    def test_lod_tile(self, client: TestClient):
+        resp = client.get("/api/risk/lod-grid/tile?zoom=14&tile_x=0&tile_y=0&forecast_day=1")
+        assert resp.status_code in (200, 422)
+
+    def test_fullgrid(self, client: TestClient):
+        resp = client.get("/api/risk/fullgrid")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "columns" in data
+        assert data["columns"] == ["lat", "lon", "risk_1d", "risk_3d", "risk_7d"]
+
+    def test_precomputed(self, client: TestClient):
+        resp = client.get("/api/risk/precomputed")
+        assert resp.status_code in (200, 404)
+
+    def test_risk_history(self, client: TestClient):
+        resp = client.get("/api/risk/history/r0_c0?days=7")
+        assert resp.status_code in (200, 404)
+        if resp.status_code == 200:
+            data = resp.json()
+            assert "grid_id" in data
+            assert "history" in data
+
+
+class TestAlertEndpoints:
+    def test_list_alerts(self, client: TestClient):
+        resp = client.get("/api/alerts")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "alerts" in data
+        assert "total" in data
+        assert isinstance(data["alerts"], list)
+
+    def test_alerts_with_min_risk_filter(self, client: TestClient):
+        resp_all = client.get("/api/alerts")
+        resp_filtered = client.get("/api/alerts?min_risk=0.9")
+        assert resp_filtered.status_code == 200
+        assert resp_filtered.json()["total"] <= resp_all.json()["total"]
+
+    def test_alerts_with_priority_filter(self, client: TestClient):
+        resp = client.get("/api/alerts?priority=P1")
+        assert resp.status_code == 200
+        for alert in resp.json()["alerts"]:
+            assert alert["priority"] == "P1"
+
+    def test_p1_alerts(self, client: TestClient):
+        resp = client.get("/api/alerts/priority/p1")
+        assert resp.status_code == 200
+        for alert in resp.json()["alerts"]:
+            assert alert["priority"] == "P1"
+
+    def test_p2_alerts(self, client: TestClient):
+        resp = client.get("/api/alerts/priority/p2")
+        assert resp.status_code == 200
+        for alert in resp.json()["alerts"]:
+            assert alert["priority"] == "P2"
+
+    def test_get_single_alert(self, client: TestClient):
+        alerts_resp = client.get("/api/alerts")
+        alerts = alerts_resp.json().get("alerts", [])
+        if alerts:
+            alert_id = alerts[0]["alert_id"]
+            resp = client.get(f"/api/alerts/{alert_id}")
+            assert resp.status_code == 200
+            assert resp.json()["alert_id"] == alert_id
+
+    def test_alert_not_found(self, client: TestClient):
+        resp = client.get("/api/alerts/nonexistent_alert_id")
+        assert resp.status_code == 404
+
+
+class TestGridEndpoints:
+    def test_grids_geojson(self, client: TestClient):
+        resp = client.get("/api/grids/geojson?date=2022-12-15")
+        assert resp.status_code in (200, 500, 503)
+
+    def test_grid_history(self, client: TestClient):
+        resp = client.get("/api/grids/r100_c200/history?days=7")
+        assert resp.status_code in (200, 404, 500, 503)
+
+    def test_historical_aggregated(self, client: TestClient):
+        resp = client.get("/api/history/aggregated?start_date=2022-12-01&end_date=2022-12-31")
+        assert resp.status_code in (200, 500, 503)
+
+    def test_multi_day_prediction(self, client: TestClient):
+        resp = client.post("/api/predict/multi-day", json={"date": "2022-12-15", "days": 3})
+        assert resp.status_code in (200, 500, 503)
+
+
+class TestCaseEndpoints:
+    def test_case_trend(self, client: TestClient):
+        resp = client.get("/api/cases/trend")
+        assert resp.status_code in (200, 500, 503)
+
+    def test_case_trend_with_params(self, client: TestClient):
+        resp = client.get(
+            "/api/cases/trend?start_date=2022-12-01&end_date=2022-12-31&group_by=week"
+        )
+        assert resp.status_code in (200, 500, 503)
+
+    def test_case_districts(self, client: TestClient):
+        resp = client.get("/api/cases/districts")
+        assert resp.status_code in (200, 500, 503)
+
+    def test_case_stats(self, client: TestClient):
+        resp = client.get("/api/cases/stats")
+        assert resp.status_code in (200, 500, 503)
+
+    def test_case_diagnoses(self, client: TestClient):
+        resp = client.get("/api/cases/diagnoses")
+        assert resp.status_code in (200, 500, 503)
+
+
+class TestGeocodedEndpoints:
+    def test_geocoded_grid(self, client: TestClient):
+        resp = client.get("/api/geocoded/grid")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "grids" in data
+
+    def test_geocoded_cases(self, client: TestClient):
+        resp = client.get("/api/geocoded/geocoded?limit=10")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "cases" in data
+
+    def test_geocoded_count(self, client: TestClient):
+        resp = client.get("/api/geocoded/geocoded/count")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "total" in data
+
+
+class TestInsightsEndpoints:
+    def test_insights_overview(self, client: TestClient):
+        resp = client.get("/api/insights/overview")
+        assert resp.status_code in (200, 404, 500)
+
+    def test_insights_cards(self, client: TestClient):
+        resp = client.get("/api/insights/cards")
+        assert resp.status_code in (200, 404, 500)
+
+
+class TestReportsEndpoints:
+    def test_reports_list(self, client: TestClient):
+        resp = client.get("/api/reports/list")
+        assert resp.status_code == 200
+        data = resp.json()
+        assert "reports" in data
+        assert "total" in data
+
+    def test_report_by_id(self, client: TestClient):
+        list_resp = client.get("/api/reports/list")
+        reports = list_resp.json().get("reports", [])
+        if reports:
+            rid = reports[0]["report_id"]
+            resp = client.get(f"/api/reports/{rid}")
+            assert resp.status_code == 200
+            data = resp.json()
+            assert "metadata" in data
+
+    def test_report_not_found(self, client: TestClient):
+        resp = client.get("/api/reports/nonexistent")
+        assert resp.status_code in (400, 404)
+
+    def test_report_summary_latest(self, client: TestClient):
+        resp = client.get("/api/reports/summary/latest")
+        assert resp.status_code in (200, 404)
+
+
+class TestAnalysisEndpoints:
+    def test_analysis_trend(self, client: TestClient):
+        resp = client.get("/api/analysis/trend?days=7")
+        assert resp.status_code == 200
+
+    def test_analysis_districts(self, client: TestClient):
+        resp = client.get("/api/analysis/districts")
+        assert resp.status_code == 200
+
+
+class TestChatEndpoint:
+    def test_chat_post(self, client: TestClient):
+        resp = client.post("/api/chat", json={
+            "messages": [{"role": "user", "content": "你好"}]
+        })
+        assert resp.status_code in (200, 401, 403, 503)