46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from mcp.types import TextContent
|
|
|
|
import pyro_mcp.server as server_module
|
|
from pyro_mcp.server import HELLO_STATIC_PAYLOAD, create_server
|
|
|
|
|
|
def test_create_server_registers_static_tool() -> None:
|
|
async def _run() -> list[str]:
|
|
server = create_server()
|
|
tools = await server.list_tools()
|
|
return [tool.name for tool in tools]
|
|
|
|
tool_names = asyncio.run(_run())
|
|
assert "hello_static" in tool_names
|
|
|
|
|
|
def test_hello_static_returns_expected_payload() -> None:
|
|
async def _run() -> tuple[list[TextContent], dict[str, Any]]:
|
|
server = create_server()
|
|
blocks, structured = await server.call_tool("hello_static", {})
|
|
assert isinstance(blocks, list)
|
|
assert all(isinstance(block, TextContent) for block in blocks)
|
|
assert isinstance(structured, dict)
|
|
return blocks, structured
|
|
|
|
_, structured_output = asyncio.run(_run())
|
|
assert structured_output == HELLO_STATIC_PAYLOAD
|
|
|
|
|
|
def test_server_main_runs_stdio_transport(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
called: dict[str, str] = {}
|
|
|
|
class StubServer:
|
|
def run(self, transport: str) -> None:
|
|
called["transport"] = transport
|
|
|
|
monkeypatch.setattr(server_module, "create_server", lambda: StubServer())
|
|
server_module.main()
|
|
|
|
assert called == {"transport": "stdio"}
|