feat: 修复 AI 风险语义并隔离 Mock 数据

ai-service/tests/test_api.py

@@ -6,7 +6,9 @@ os.environ.setdefault("ALLOWED_STREAM_HOSTS", "localhost,127.0.0.1,100.83.103.1"
 
 from fastapi.testclient import TestClient
 
+from app import main as main_module
 from app.main import app
+from app.detector import MockDetector
 
 # 1x1 透明 PNG
 TINY_PNG = base64.b64decode(
@@ -22,6 +24,8 @@ def test_health():
     body = r.json()
     assert body["status"] == "ok"
     assert body["model"] in ("mock", "onnx")
+    assert body["modelVersion"]
+    assert body["isMock"] is True
 
 
 def test_detect_ok():
@@ -29,6 +33,10 @@ def test_detect_ok():
     assert r.status_code == 200
     body = r.json()
     assert body["model"] == "mock"
+    assert body["modelVersion"]
+    assert body["isMock"] is True
+    assert body["status"] == "healthy"
+    assert body["abnormalProbability"] == 0
     assert len(body["detections"]) >= 1
     d = body["detections"][0]
     assert d["class"] in ("healthy", "sick")
@@ -36,6 +44,28 @@ def test_detect_ok():
     assert d["bbox"]["w"] > 0
 
 
+def test_detect_uses_abnormal_class_confidence(monkeypatch):
+    monkeypatch.setattr(main_module, "detector", MockDetector(class_name="sick", confidence=0.92))
+    r = client.post("/detect", files={"file": ("a.png", TINY_PNG, "image/png")})
+    assert r.status_code == 200
+    body = r.json()
+    assert body["status"] == "abnormal"
+    assert body["abnormalProbability"] == 0.92
+
+
+def test_detect_empty_result_is_unknown(monkeypatch):
+    class EmptyDetector:
+        def detect(self, image_bytes):
+            return []
+
+    monkeypatch.setattr(main_module, "detector", EmptyDetector())
+    r = client.post("/detect", files={"file": ("a.png", TINY_PNG, "image/png")})
+    assert r.status_code == 200
+    body = r.json()
+    assert body["status"] == "unknown"
+    assert body["abnormalProbability"] == 0
+
+
 def test_detect_empty_file_rejected():
     r = client.post("/detect", files={"file": ("a.png", b"", "image/png")})
     assert r.status_code == 400