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
96 lines
3.3 KiB
Python
96 lines
3.3 KiB
Python
import anyio
|
|
import pytest
|
|
from mcp_types import (
|
|
ClientCapabilities,
|
|
Implementation,
|
|
InitializeRequestParams,
|
|
JSONRPCMessage,
|
|
JSONRPCNotification,
|
|
JSONRPCRequest,
|
|
JSONRPCResponse,
|
|
NotificationParams,
|
|
)
|
|
from mcp_types.version import LATEST_HANDSHAKE_VERSION
|
|
|
|
from mcp.server.lowlevel import NotificationOptions, Server
|
|
from mcp.server.models import InitializationOptions
|
|
from mcp.shared.message import SessionMessage
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_request_id_match() -> None:
|
|
"""Test that the server preserves request IDs in responses."""
|
|
server = Server("test")
|
|
custom_request_id = "test-123"
|
|
|
|
# Create memory streams for communication
|
|
client_writer, client_reader = anyio.create_memory_object_stream[SessionMessage | Exception](1)
|
|
server_writer, server_reader = anyio.create_memory_object_stream[SessionMessage | Exception](1)
|
|
|
|
# Server task to process the request
|
|
async def run_server():
|
|
async with client_reader, server_writer:
|
|
await server.run(
|
|
client_reader,
|
|
server_writer,
|
|
InitializationOptions(
|
|
server_name="test",
|
|
server_version="1.0.0",
|
|
capabilities=server.get_capabilities(
|
|
notification_options=NotificationOptions(),
|
|
experimental_capabilities={},
|
|
),
|
|
),
|
|
raise_exceptions=True,
|
|
)
|
|
|
|
# Start server task
|
|
async with (
|
|
anyio.create_task_group() as tg,
|
|
client_writer,
|
|
client_reader,
|
|
server_writer,
|
|
server_reader,
|
|
):
|
|
tg.start_soon(run_server)
|
|
|
|
# Send initialize request
|
|
init_req = JSONRPCRequest(
|
|
id="init-1",
|
|
method="initialize",
|
|
params=InitializeRequestParams(
|
|
protocol_version=LATEST_HANDSHAKE_VERSION,
|
|
capabilities=ClientCapabilities(),
|
|
client_info=Implementation(name="test-client", version="1.0.0"),
|
|
).model_dump(by_alias=True, exclude_none=True),
|
|
jsonrpc="2.0",
|
|
)
|
|
|
|
await client_writer.send(SessionMessage(init_req))
|
|
response = await server_reader.receive() # Get init response but don't need to check it
|
|
|
|
# Send initialized notification
|
|
initialized_notification = JSONRPCNotification(
|
|
method="notifications/initialized",
|
|
params=NotificationParams().model_dump(by_alias=True, exclude_none=True),
|
|
jsonrpc="2.0",
|
|
)
|
|
await client_writer.send(SessionMessage(initialized_notification))
|
|
|
|
# Send ping request with custom ID
|
|
ping_request = JSONRPCRequest(id=custom_request_id, method="ping", params={}, jsonrpc="2.0")
|
|
|
|
await client_writer.send(SessionMessage(ping_request))
|
|
|
|
# Read response
|
|
response = await server_reader.receive()
|
|
|
|
# Verify response ID matches request ID
|
|
assert isinstance(response, SessionMessage)
|
|
assert isinstance(response.message, JSONRPCMessage)
|
|
assert isinstance(response.message, JSONRPCResponse)
|
|
assert response.message.id == custom_request_id, "Response ID should match request ID"
|
|
|
|
# Cancel server task
|
|
tg.cancel_scope.cancel()
|