wehub-resource-sync
49b9bb6724
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Waiting to run
40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
"""Drive `handle_one` directly to assert the raw result-dict shape, then over the wire."""
|
|
|
|
import mcp_types as types
|
|
from mcp_types.version import LATEST_MODERN_VERSION
|
|
|
|
from mcp.client import Client
|
|
from stories._harness import Target, run_client
|
|
from stories.serve_one.server import build_server, handle_one
|
|
|
|
|
|
async def main(target: Target, *, mode: str = "auto") -> None:
|
|
# ── direct: the namesake recipe — Connection.from_envelope + serve_one → raw result dict.
|
|
# The entry enters lifespan once and threads it to every per-request handle_one().
|
|
server = build_server()
|
|
params = {
|
|
"name": "add",
|
|
"arguments": {"a": 2, "b": 3},
|
|
"_meta": {
|
|
types.PROTOCOL_VERSION_META_KEY: LATEST_MODERN_VERSION,
|
|
types.CLIENT_INFO_META_KEY: {"name": "serve-one-probe", "version": "0.0.0"},
|
|
types.CLIENT_CAPABILITIES_META_KEY: {},
|
|
},
|
|
}
|
|
async with server.lifespan(server) as lifespan_state:
|
|
raw = await handle_one(server, "tools/call", params, lifespan_state=lifespan_state)
|
|
assert raw["structuredContent"] == {"result": 5}, raw
|
|
assert raw["content"][0] == {"type": "text", "text": "5"}, raw
|
|
|
|
# ── over the wire: the loop-mode driver behind the connected client.
|
|
async with Client(target, mode=mode) as client:
|
|
listed = await client.list_tools()
|
|
assert [t.name for t in listed.tools] == ["add"]
|
|
|
|
result = await client.call_tool("add", {"a": 2, "b": 3})
|
|
assert result.structured_content == {"result": 5}, result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_client(main)
|