connection service

2026-06-02 23:34:50 +07:00
parent 83d97ecd62
commit e7f7ff4851
6 changed files with 845 additions and 43 deletions
@@ -0,0 +1,143 @@
+from __future__ import annotations
+
+from wf_mcp.broker import WfMcpService
+from wf_mcp.broker.service.connection_service import ConnectionService
+from wf_mcp.broker.service.events import BrokerEventRecorder
+from wf_mcp.broker.service.source_catalog import SourceCatalogService
+from wf_mcp.events import EventBus
+from wf_mcp.models import BrokerConfig, ConnectionConfig
+from wf_mcp.runtime import ToolExecutor
+from wf_mcp.storage import FileStore
+
+from ..test_support import local_temp_root
+
+
+def _source_catalog(service: ConnectionService) -> SourceCatalogService:
+    store = FileStore(local_temp_root() / "connection_service_catalog")
+
+    def _tool_executor_for(_connection: ConnectionConfig) -> ToolExecutor:
+        raise AssertionError("tool executor should not be needed in these tests")
+
+    catalog = SourceCatalogService(
+        store=store,
+        connection_lookup=service.get,
+        connection_list_enabled=service.list_enabled,
+        connection_list_all=service.list_all,
+        tool_executor_for=_tool_executor_for,
+        load_auth=lambda _connection_id: None,
+        emit_event=service.events.record_event,
+    )
+    service.bind_source_catalog(catalog)
+    return catalog
+
+
+def test_connection_service_rejects_reserved_connection_ids() -> None:
+    service = ConnectionService(events=BrokerEventRecorder(EventBus()))
+    _source_catalog(service)
+
+    for connection_id in ("wf.admin", "wf.mcp"):
+        try:
+            service.register_connection(
+                ConnectionConfig(id=connection_id, server="wf", account="reserved")
+            )
+        except ValueError as exc:
+            assert connection_id in str(exc)
+            assert "reserved by wf-mcp" in str(exc)
+        else:
+            raise AssertionError(f"expected {connection_id!r} to be rejected")
+
+
+def test_connection_service_registers_connection_and_empty_source() -> None:
+    service = ConnectionService(events=BrokerEventRecorder(EventBus()))
+    catalog = _source_catalog(service)
+
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+
+    assert service.get("demo.personal").server == "demo"
+    assert [connection.id for connection in service.list_enabled()] == ["demo.personal"]
+    source = catalog.capability_sources["demo.personal"]
+    assert source.enabled is True
+    assert source.description == "No catalog loaded for demo.personal."
+    assert service.events.list_events()[0].kind == "connection_registered"
+    assert service.events.list_events()[0].connection_id == "demo.personal"
+
+
+def test_connection_service_sync_removes_retired_connections_and_sources() -> None:
+    service = ConnectionService(events=BrokerEventRecorder(EventBus()))
+    catalog = _source_catalog(service)
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+
+    service.sync_connections_from_config(
+        BrokerConfig(store_root=local_temp_root(), connections=[])
+    )
+
+    assert service.list_all() == []
+    assert "demo.personal" not in catalog.capability_sources
+
+
+def test_connection_service_sync_updates_existing_source_enabled_flag() -> None:
+    service = ConnectionService(events=BrokerEventRecorder(EventBus()))
+    catalog = _source_catalog(service)
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+
+    service.sync_connections_from_config(
+        BrokerConfig(
+            store_root=local_temp_root(),
+            connections=[
+                ConnectionConfig(
+                    id="demo.personal",
+                    server="demo",
+                    account="personal",
+                    enabled=False,
+                )
+            ],
+        )
+    )
+
+    assert service.get("demo.personal").enabled is False
+    assert catalog.capability_sources["demo.personal"].enabled is False
+
+
+def test_wfmcpservice_exposes_connection_registry_from_connection_service() -> None:
+    service = WfMcpService(store=FileStore(local_temp_root() / "connection_facade"))
+
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+
+    assert service.connections is service.connection_service.connections
+    assert service.connections.get("demo.personal").account == "personal"
+    assert "demo.personal" in service.capability_sources
+
+
+def test_wfmcpservice_sync_connections_delegates_to_connection_service() -> None:
+    service = WfMcpService(store=FileStore(local_temp_root() / "connection_sync"))
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+
+    service.sync_connections_from_config(
+        BrokerConfig(
+            store_root=local_temp_root(),
+            connections=[
+                ConnectionConfig(
+                    id="demo.work",
+                    server="demo",
+                    account="work",
+                    enabled=True,
+                )
+            ],
+        )
+    )
+
+    assert [connection.id for connection in service.connections.list_all()] == [
+        "demo.work"
+    ]
+    assert "demo.personal" not in service.capability_sources
+    assert "demo.work" in service.capability_sources