Bootstrap pyro_mcp v0.0.1 with MCP static tool and Ollama demo

tests/test_demo.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+import asyncio
+from collections.abc import Sequence
+from typing import Any
+
+import pytest
+from mcp.types import TextContent
+
+import pyro_mcp.demo as demo_module
+from pyro_mcp.demo import run_demo
+from pyro_mcp.server import HELLO_STATIC_PAYLOAD
+
+
+def test_run_demo_returns_static_payload() -> None:
+    payload = asyncio.run(run_demo())
+    assert payload == HELLO_STATIC_PAYLOAD
+
+
+def test_run_demo_raises_for_non_text_blocks(monkeypatch: pytest.MonkeyPatch) -> None:
+    class StubServer:
+        async def call_tool(
+            self,
+            name: str,
+            arguments: dict[str, Any],
+        ) -> tuple[Sequence[int], dict[str, str]]:
+            assert name == "hello_static"
+            assert arguments == {}
+            return [123], HELLO_STATIC_PAYLOAD
+
+    monkeypatch.setattr(demo_module, "create_server", lambda: StubServer())
+
+    with pytest.raises(TypeError, match="unexpected MCP content block output"):
+        asyncio.run(demo_module.run_demo())
+
+
+def test_run_demo_raises_for_non_dict_payload(monkeypatch: pytest.MonkeyPatch) -> None:
+    class StubServer:
+        async def call_tool(
+            self,
+            name: str,
+            arguments: dict[str, Any],
+        ) -> tuple[list[TextContent], str]:
+            assert name == "hello_static"
+            assert arguments == {}
+            return [TextContent(type="text", text="x")], "bad"
+
+    monkeypatch.setattr(demo_module, "create_server", lambda: StubServer())
+
+    with pytest.raises(TypeError, match="expected a structured dictionary payload"):
+        asyncio.run(demo_module.run_demo())
+
+
+def test_run_demo_raises_for_unexpected_payload(monkeypatch: pytest.MonkeyPatch) -> None:
+    class StubServer:
+        async def call_tool(
+            self,
+            name: str,
+            arguments: dict[str, Any],
+        ) -> tuple[list[TextContent], dict[str, str]]:
+            assert name == "hello_static"
+            assert arguments == {}
+            return [TextContent(type="text", text="x")], {
+                "message": "different",
+                "status": "ok",
+                "version": "0.0.1",
+            }
+
+    monkeypatch.setattr(demo_module, "create_server", lambda: StubServer())
+
+    with pytest.raises(ValueError, match="static payload did not match expected value"):
+        asyncio.run(demo_module.run_demo())
+
+
+def test_demo_main_prints_json(
+    monkeypatch: pytest.MonkeyPatch,
+    capsys: pytest.CaptureFixture[str],
+) -> None:
+    async def fake_run_demo() -> dict[str, str]:
+        return HELLO_STATIC_PAYLOAD
+
+    monkeypatch.setattr(demo_module, "run_demo", fake_run_demo)
+    demo_module.main()
+
+    output = capsys.readouterr().out
+    assert '"message": "hello from pyro_mcp"' in output