orchestrator: 全专家失败时不调用 aggregator provider

2026-09-19 08:42:09 +00:00
parent 6c24672e87
commit c0bcf1d851
2 changed files with 60 additions and 2 deletions
@@ -244,3 +244,57 @@ class TestPartialExpertsOk:
        assert captured_status.get("status") == "success", \
            f"期望 status=success,实际 {captured_status.get('status')}"
        print(f"PASS: 1 ok + 4 error → status={captured_status.get('status')}")
+
+
+class TestNoAggregatorCallOnDegraded:
+    """全专家失败时,aggregator provider 不应被调用。"""
+
+    @pytest.mark.asyncio
+    async def test_all_error_no_aggregator_call(self):
+        """5 个专家全 error → 不应调用 _agent_provider('aggregator')。"""
+        header = _make_header()
+        aggregator_called = []
+
+        async def mock_run_specialists(header, *, version, before):
+            return _all_error_reports()
+
+        async def mock_agent_provider(agent_id, *, tier):
+            aggregator_called.append((agent_id, tier))
+            return MagicMock(model="test-model")
+
+        async def mock_load_header(mid, db=None):
+            return header
+
+        captured_values = {}
+
+        async def mock_upsert(session, **kw):
+            captured_values.update(kw.get("values", {}))
+            mock_pred = MagicMock()
+            mock_pred.id = 1
+            mock_pred.provider = "test"
+            mock_pred.model = kw["values"].get("model")
+            return mock_pred
+
+        class FakeUow:
+            async def __aenter__(self):
+                return self
+            async def __aexit__(self, *a):
+                pass
+            async def get(self, cls, id):
+                return MagicMock()
+
+        with patch.object(orch_mod, "run_specialists", mock_run_specialists), \
+             patch.object(orch_mod, "_agent_provider", mock_agent_provider), \
+             patch.object(orch_mod, "load_match_header", mock_load_header), \
+             patch.object(orch_mod, "_upsert_prediction", mock_upsert), \
+             patch.object(orch_mod, "get_uow", FakeUow):
+
+            await orch_mod.predict_match_multi(999)
+
+        # 断言:aggregator provider 未被调用
+        assert len(aggregator_called) == 0, \
+            f"全失败时不应调用 aggregator provider,实际调用: {aggregator_called}"
+        # 断言:model 使用 settings 默认值
+        assert captured_values.get("model") is not None
+        assert captured_values.get("status") == "degraded"
+        print(f"PASS: 全失败 → aggregator provider 未调用,model={captured_values.get('model')}")