goodbye modes 2 working on it

tests/wf_mcp/test_server.py

@@ -0,0 +1,146 @@
+from __future__ import annotations
+
+import asyncio
+import sys
+from typing import Any
+
+from wf_mcp.models import BrokerConfig, ConnectionConfig
+from wf_mcp.server import create_server_client
+
+from .test_support import fixture_server_path, local_temp_root
+
+
+def _structured(result: Any) -> dict[str, Any]:
+    content = result.structured_content
+    assert isinstance(content, dict)
+    return content
+
+
+def test_server_exposes_upstream_admin_and_workflow_tools() -> None:
+    config = BrokerConfig(
+        store_root=local_temp_root() / "unified_server_store",
+        connections=[
+            ConnectionConfig(
+                id="fixture.personal",
+                server="fixture",
+                account="personal",
+                metadata={
+                    "transport": "stdio",
+                    "command": sys.executable,
+                    "args": [fixture_server_path()],
+                },
+            )
+        ],
+    )
+
+    async def run_proxy() -> None:
+        client = create_server_client(config)
+        async with client:
+            tools = await client.list_tools()
+            names = [tool.name for tool in tools]
+            assert "fixture.personal.echo_tool" in names
+            assert "wf.admin.list_connections" in names
+            assert "wf.admin.get_connection_statuses" in names
+            assert "wf.admin.refresh_connection_catalog" in names
+            assert "wf.admin.get_catalog" in names
+            assert "wf.admin.get_planner_catalog" in names
+            assert "wf.admin.list_spec_sources" in names
+            assert "wf.admin.list_sources" in names
+            assert "wf.admin.read_resource" in names
+            assert "wf.admin.render_prompt" in names
+            assert "wf.admin.invoke_method" in names
+            assert "wf.admin.call_tool" in names
+            assert "wf.admin.get_events" in names
+            assert "wf.workflow.list_artifacts" in names
+            assert "wf.workflow.run_deployment" in names
+
+            echo_result = await client.call_tool(
+                "fixture.personal.echo_tool",
+                {"text": "hello"},
+            )
+            artifacts_result = await client.call_tool("wf.workflow.list_artifacts")
+            sources_result = await client.call_tool("wf.admin.list_sources")
+
+            assert _structured(echo_result)["echoed"] == "hello"
+            assert _structured(artifacts_result)["nodes"] == []
+            source_ids = {
+                source["id"] for source in _structured(sources_result)["result"]
+            }
+            assert "wf.admin" in source_ids
+            assert "wf.mcp" in source_ids
+            assert "wf.std" in source_ids
+
+    asyncio.run(run_proxy())
+
+
+def test_server_can_hide_admin_tools() -> None:
+    config = BrokerConfig(
+        store_root=local_temp_root() / "unified_no_admin_store",
+        connections=[
+            ConnectionConfig(
+                id="fixture.personal",
+                server="fixture",
+                account="personal",
+                metadata={
+                    "transport": "stdio",
+                    "command": sys.executable,
+                    "args": [fixture_server_path()],
+                },
+            )
+        ],
+    )
+
+    async def run_proxy() -> None:
+        client = create_server_client(config, admin_tools=False)
+        async with client:
+            tools = await client.list_tools()
+            names = [tool.name for tool in tools]
+            assert "fixture.personal.echo_tool" in names
+            assert "wf.workflow.list_artifacts" in names
+            assert "wf.admin.list_connections" not in names
+
+    asyncio.run(run_proxy())
+
+
+def test_workflow_tools_have_human_metadata() -> None:
+    config = BrokerConfig(
+        store_root=local_temp_root() / "unified_metadata_store",
+        connections=[],
+    )
+
+    async def run_proxy() -> None:
+        client = create_server_client(config, admin_tools=False)
+        async with client:
+            tools = await client.list_tools()
+            by_name = {tool.name: tool for tool in tools}
+            list_artifacts = by_name["wf.workflow.list_artifacts"]
+            run_deployment = by_name["wf.workflow.run_deployment"]
+
+            assert list_artifacts.title == "List Workflow Artifacts"
+            assert "saved workflow artifacts" in (list_artifacts.description or "")
+            assert run_deployment.title == "Run Workflow Deployment"
+            assert "deployment_id" in (run_deployment.description or "")
+
+    asyncio.run(run_proxy())
+
+
+def test_admin_tools_have_human_metadata() -> None:
+    config = BrokerConfig(
+        store_root=local_temp_root() / "unified_admin_metadata_store",
+        connections=[],
+    )
+
+    async def run_proxy() -> None:
+        client = create_server_client(config)
+        async with client:
+            tools = await client.list_tools()
+            by_name = {tool.name: tool for tool in tools}
+            list_connections = by_name["wf.admin.list_connections"]
+            reload_config = by_name["wf.admin.reload_config"]
+
+            assert list_connections.title == "List Connections"
+            assert "configured MCP connections" in (list_connections.description or "")
+            assert reload_config.title == "Reload Config"
+            assert "remount" in (reload_config.description or "")
+
+    asyncio.run(run_proxy())