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
95 lines
4.3 KiB
Python
95 lines
4.3 KiB
Python
"""`docs/protocol-versions.md`: every claim the page makes, proved against the real SDK."""
|
|
|
|
import re
|
|
|
|
import pytest
|
|
from mcp_types import DiscoverResult, Implementation, ServerCapabilities
|
|
|
|
from docs_src.protocol_versions import tutorial001, tutorial002, tutorial003, tutorial004
|
|
from mcp import Client
|
|
|
|
# See test_index.py for why this is a per-module mark and not a conftest hook.
|
|
pytestmark = [pytest.mark.anyio, pytest.mark.filterwarnings("error::mcp.MCPDeprecationWarning")]
|
|
|
|
|
|
async def test_auto_lands_on_the_modern_version() -> None:
|
|
"""tutorial001: the default `mode="auto"` probes `server/discover` and adopts the result."""
|
|
async with Client(tutorial001.mcp) as client:
|
|
assert client.protocol_version == "2026-07-28"
|
|
assert client.server_info.name == "Bookshop"
|
|
assert client.session.discover_result is not None
|
|
assert client.session.initialize_result is None
|
|
|
|
|
|
async def test_legacy_forces_the_initialize_handshake() -> None:
|
|
"""tutorial002: `mode="legacy"` runs `initialize` against the very same server."""
|
|
async with Client(tutorial002.mcp, mode="legacy") as client:
|
|
assert client.protocol_version == "2025-11-25"
|
|
assert client.server_info.name == "Bookshop"
|
|
assert client.session.initialize_result is not None
|
|
assert client.session.discover_result is None
|
|
|
|
|
|
async def test_version_pin_sends_nothing_and_knows_nothing() -> None:
|
|
"""tutorial003: a pin adopts the version locally; `server_info` and capabilities are blank."""
|
|
async with Client(tutorial003.mcp, mode="2026-07-28") as client:
|
|
assert client.protocol_version == "2026-07-28"
|
|
assert client.server_info == Implementation(name="", version="")
|
|
# The `!!! check` fence is the literal `print(client.server_info)` output.
|
|
assert str(client.server_info) == "name='' title=None version='' description=None website_url=None icons=None"
|
|
assert client.server_capabilities == ServerCapabilities()
|
|
result = await client.call_tool("search_books", {"query": "dune"})
|
|
assert result.structured_content == {"result": "Found 3 books matching 'dune'."}
|
|
|
|
|
|
def test_handshake_era_version_is_not_a_valid_pin() -> None:
|
|
"""A pre-2026 version string is rejected at construction with the exact error the page shows."""
|
|
with pytest.raises(
|
|
ValueError,
|
|
match=re.escape(
|
|
"mode must be 'legacy', 'auto', or one of ['2026-07-28']; "
|
|
"got '2025-06-18' ('2025-06-18' is a handshake-era version; use mode='legacy')"
|
|
),
|
|
):
|
|
Client(tutorial003.mcp, mode="2025-06-18")
|
|
|
|
|
|
async def test_prior_discover_round_trips() -> None:
|
|
"""tutorial004: save `discover_result`, reconnect with it, and the identity comes back."""
|
|
async with Client(tutorial004.mcp) as client:
|
|
saved = client.session.discover_result
|
|
assert saved is not None
|
|
assert saved.supported_versions == ["2026-07-28"]
|
|
|
|
async with Client(tutorial004.mcp, mode="2026-07-28", prior_discover=saved) as client:
|
|
assert client.protocol_version == "2026-07-28"
|
|
assert client.server_info.name == "Bookshop"
|
|
assert client.server_capabilities.tools is not None
|
|
|
|
|
|
async def test_discover_result_survives_json() -> None:
|
|
"""`DiscoverResult` is a Pydantic model: dump it to JSON, validate it back, reconnect with it."""
|
|
async with Client(tutorial004.mcp) as client:
|
|
saved = client.session.discover_result
|
|
assert saved is not None
|
|
|
|
restored = DiscoverResult.model_validate_json(saved.model_dump_json())
|
|
assert restored == saved
|
|
|
|
async with Client(tutorial004.mcp, mode="2026-07-28", prior_discover=restored) as client:
|
|
assert client.server_info.name == "Bookshop"
|
|
|
|
|
|
async def test_prior_discover_is_ignored_unless_mode_is_a_pin() -> None:
|
|
"""The `!!! tip`: under `auto` the client probes anyway; under `legacy` it never discovers."""
|
|
stale = DiscoverResult(
|
|
supported_versions=["2026-07-28"],
|
|
capabilities=ServerCapabilities(),
|
|
server_info=Implementation(name="Stale", version="0.0.0"),
|
|
)
|
|
async with Client(tutorial004.mcp, prior_discover=stale) as client:
|
|
assert client.server_info.name == "Bookshop"
|
|
async with Client(tutorial004.mcp, mode="legacy", prior_discover=stale) as client:
|
|
assert client.session.discover_result is None
|
|
assert client.protocol_version == "2025-11-25"
|