Add request context tests and pre-commit config

Author: dsp-ant

.pre-commit-config.yaml

@@ -0,0 +1,15 @@
+fail_fast: true
+
+repos:
+  - repo: https://github.com/pre-commit/mirrors-prettier
+    rev: v3.1.0
+    hooks:
+      - id: prettier
+        types_or: [yaml, json5]
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.8.1
+    hooks:
+      - id: ruff-format
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]

tests/server/test_request_context.py

@@ -0,0 +1,115 @@
+"""Test request context handling in the server."""
+
+import io
+import logging
+
+import anyio
+import pytest
+
+from mcp.server import Server
+from mcp.server.models import InitializationOptions
+from mcp.server.stdio import stdio_server
+from mcp.types import (
+    LATEST_PROTOCOL_VERSION,
+    CallToolRequest,
+    CallToolRequestParams,
+    ClientCapabilities,
+    ClientRequest,
+    Implementation,
+    InitializeRequest,
+    InitializeRequestParams,
+    JSONRPCMessage,
+    JSONRPCRequest,
+    RequestParams,
+    ServerCapabilities,
+    TextContent,
+    ToolsCapability,
+)
+
+logger = logging.getLogger()
+logger.setLevel(logging.DEBUG)
+
+
+@pytest.mark.anyio
+async def test_request_meta_simple():
+    """Test request_meta availability in server.request_context"""
+    server = Server("test_server")
+
+    # Add a tool handler to verify request context
+    tool_called = False
+    captured_meta = None
+
+    @server.call_tool()
+    async def handle_tool_call(name: str, args: dict | None = None):
+        nonlocal captured_meta, tool_called
+        # Capture the meta from request context
+        logger.info(server.request_context)
+        captured_meta = server.request_context.meta
+        tool_called = True
+        return [TextContent(type="text", text="test")]
+
+    # Initialize server with tool capabilities
+    init_request = InitializeRequest(
+        method="initialize",
+        params=InitializeRequestParams(
+            protocolVersion=LATEST_PROTOCOL_VERSION,
+            capabilities=ClientCapabilities(),
+            clientInfo=Implementation(name="test", version="1.0"),
+        ),
+    )
+
+    init_msg = JSONRPCMessage(
+        root=JSONRPCRequest(
+            jsonrpc="2.0",
+            id=1,
+            **init_request.model_dump(by_alias=True, exclude_none=True),
+        )
+    )
+
+    # Send initialized notification
+    init_notify = JSONRPCMessage(
+        root=dict(jsonrpc="2.0", method="notifications/initialized")
+    )
+
+    # Send tool call request with _meta
+    call_request = ClientRequest(
+        root=CallToolRequest(
+            method="tools/call",
+            params=CallToolRequestParams(
+                name="test", arguments={}, _meta=RequestParams.Meta(progressToken=123)
+            ),
+        )
+    )
+
+    call_msg = JSONRPCMessage(
+        root=JSONRPCRequest(
+            jsonrpc="2.0",
+            id=2,
+            **call_request.model_dump(by_alias=True, exclude_none=True),
+        )
+    )
+
+    stdin = io.StringIO()
+    stdout = io.StringIO()
+    stdin.write(init_msg.model_dump_json(by_alias=True, exclude_none=True) + "\n")
+    stdin.write(init_notify.model_dump_json(by_alias=True, exclude_none=True) + "\n")
+    stdin.write(call_msg.model_dump_json(by_alias=True, exclude_none=True) + "\n")
+    stdin.seek(0)
+
+    async with stdio_server(
+        stdin=anyio.AsyncFile(stdin), stdout=anyio.AsyncFile(stdout)
+    ) as (read_stream, write_stream):
+        await server.run(
+            read_stream,
+            write_stream,
+            InitializationOptions(
+                server_name="test",
+                server_version="1.0",
+                capabilities=ServerCapabilities(tools=ToolsCapability()),
+            ),
+            raise_exceptions=True,
+        )
+
+    assert tool_called
+    assert captured_meta is not None
+    assert captured_meta.progressToken == 123