refactor(api): unify scheme query endpoints

tests/api/test_leakage_endpoints.py

@@ -1,7 +1,5 @@
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
-from types import SimpleNamespace
-
 from app.api.v1.endpoints import leakage as leakage_endpoint
 
 
@@ -35,34 +33,3 @@ def test_identify_leakage_success(monkeypatch):
     )
     assert response.status_code == 200
     assert response.json()["area_count"] == 0
-
-
-def test_query_leakage_schemes_success(monkeypatch):
-    monkeypatch.setattr(
-        leakage_endpoint,
-        "list_leakage_identify_schemes",
-        lambda network, query_date=None: [
-            {"scheme_name": "dma_001", "scheme_type": "dma_leak_identification"}
-        ],
-    )
-    client = _build_client()
-    response = client.get("/api/v1/leakage/schemes/", params={"network": "demo"})
-    assert response.status_code == 200
-    assert response.json()[0]["scheme_name"] == "dma_001"
-
-
-def test_query_leakage_scheme_detail_success(monkeypatch):
-    monkeypatch.setattr(
-        leakage_endpoint,
-        "get_leakage_identify_scheme_detail",
-        lambda network, scheme_name: {
-            "scheme_name": scheme_name,
-            "rows": [{"Area": "1", "LeakageFlow_m3_per_s": 0.1}],
-        },
-    )
-    client = _build_client()
-    response = client.get(
-        "/api/v1/leakage/schemes/dma_001", params={"network": "demo"}
-    )
-    assert response.status_code == 200
-    assert response.json()["scheme_name"] == "dma_001"

tests/api/test_schemes_endpoints.py

@@ -0,0 +1,102 @@
+from datetime import date
+
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+
+from app.api.v1.endpoints import schemes as schemes_endpoint
+
+
+def _build_client() -> TestClient:
+    app = FastAPI()
+    app.include_router(schemes_endpoint.router, prefix="/api/v1")
+    return TestClient(app)
+
+
+def test_get_schemes_forwards_optional_scheme_type(monkeypatch):
+    captured = {}
+
+    def fake_get_all_schemes(network, scheme_type=None, query_date=None):
+        captured["network"] = network
+        captured["scheme_type"] = scheme_type
+        captured["query_date"] = query_date
+        return [
+            {
+                "scheme_id": 1,
+                "scheme_name": "burst_case",
+                "scheme_type": scheme_type,
+            }
+        ]
+
+    monkeypatch.setattr(schemes_endpoint, "get_all_schemes", fake_get_all_schemes)
+
+    response = _build_client().get(
+        "/api/v1/schemes",
+        params={"network": "demo", "scheme_type": "burst_analysis"},
+    )
+
+    assert response.status_code == 200
+    assert captured == {
+        "network": "demo",
+        "scheme_type": "burst_analysis",
+        "query_date": None,
+    }
+    assert response.json()[0]["scheme_type"] == "burst_analysis"
+
+
+def test_get_schemes_forwards_query_date(monkeypatch):
+    captured = {}
+
+    def fake_get_all_schemes(network, scheme_type=None, query_date=None):
+        captured["network"] = network
+        captured["scheme_type"] = scheme_type
+        captured["query_date"] = query_date
+        return []
+
+    monkeypatch.setattr(schemes_endpoint, "get_all_schemes", fake_get_all_schemes)
+
+    response = _build_client().get(
+        "/api/v1/schemes",
+        params={
+            "network": "demo",
+            "scheme_type": "dma_leak_identification",
+            "query_date": "2026-01-02T00:00:00+08:00",
+        },
+    )
+
+    assert response.status_code == 200
+    assert captured == {
+        "network": "demo",
+        "scheme_type": "dma_leak_identification",
+        "query_date": date(2026, 1, 2),
+    }
+
+
+def test_get_scheme_detail_forwards_scheme_type(monkeypatch):
+    captured = {}
+
+    def fake_query_scheme_detail(name, scheme_name, scheme_type=None):
+        captured["name"] = name
+        captured["scheme_name"] = scheme_name
+        captured["scheme_type"] = scheme_type
+        return {
+            "scheme_name": scheme_name,
+            "scheme_type": scheme_type,
+            "rows": [{"Area": "1", "LeakageFlow_m3_per_s": 0.1}],
+        }
+
+    monkeypatch.setattr(
+        schemes_endpoint, "query_scheme_detail", fake_query_scheme_detail
+    )
+
+    response = _build_client().get(
+        "/api/v1/schemes/dma_001",
+        params={"network": "demo", "scheme_type": "dma_leak_identification"},
+    )
+
+    assert response.status_code == 200
+    assert captured == {
+        "name": "demo",
+        "scheme_name": "dma_001",
+        "scheme_type": "dma_leak_identification",
+    }
+    assert response.json()["scheme_name"] == "dma_001"