fix: harden realtime pipeline lifecycle

2026-09-07 08:27:52 +08:00
parent 43f2cc445e
commit 8025ed8aa0
12 changed files with 445 additions and 42 deletions
@@ -0,0 +1,38 @@
+import unittest
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, Mock
+
+from services.config_resolver import _agent_resource_for, _resource_for
+
+
+class DisabledResourceTest(unittest.IsolatedAsyncioTestCase):
+    async def test_disabled_binding_uses_enabled_fallback_for_models_and_agents(self):
+        for capability, interface_type in (("LLM", "openai-llm"), ("Agent", "dify")):
+            with self.subTest(capability=capability):
+                disabled = SimpleNamespace(
+                    id="disabled", capability=capability,
+                    interface_type=interface_type, enabled=False,
+                )
+                fallback = SimpleNamespace(id="enabled", enabled=True)
+                result = Mock()
+                result.scalar_one_or_none.return_value = fallback
+                session = SimpleNamespace(
+                    get=AsyncMock(side_effect=[SimpleNamespace(model_resource_id="disabled"), disabled]),
+                    execute=AsyncMock(return_value=result),
+                )
+                if capability == "Agent":
+                    selected = await _agent_resource_for(session, "assistant", interface_type)
+                else:
+                    selected = await _resource_for(session, "assistant", capability)
+                self.assertIs(selected, fallback)
+                statement = str(session.execute.await_args.args[0])
+                self.assertIn("model_resources.enabled IS true", statement)
+
+    async def test_enabled_binding_is_preserved(self):
+        resource = SimpleNamespace(id="enabled", capability="LLM", enabled=True)
+        session = SimpleNamespace(
+            get=AsyncMock(side_effect=[SimpleNamespace(model_resource_id=resource.id), resource]),
+            execute=AsyncMock(),
+        )
+        self.assertIs(await _resource_for(session, "assistant", "LLM"), resource)
+        session.execute.assert_not_awaited()