@@ -208,7 +208,11 @@ async def test_runner_gates_requests_before_initialize(server: SrvT):
208208 async with connected_runner (server , initialized = False ) as (client , _ ):
209209 with pytest .raises (MCPError ) as exc :
210210 await client .send_raw_request ("tools/list" , None )
211- assert exc .value .error == ErrorData (code = INVALID_PARAMS , message = "Invalid request parameters" , data = "" )
211+ assert exc .value .error == ErrorData (
212+ code = INVALID_REQUEST ,
213+ message = "Received request before initialization was complete" ,
214+ data = "The client must send an 'initialize' request before any other request." ,
215+ )
212216 # ping is exempt from the gate
213217 assert await client .send_raw_request ("ping" , None ) == {}
214218
@@ -249,7 +253,11 @@ async def greet(ctx: Ctx, params: RequestParams | None) -> Any:
249253 async with connected_runner (server , initialized = False ) as (client , _ ):
250254 with pytest .raises (MCPError ) as exc :
251255 await client .send_raw_request ("custom/greet" , None )
252- assert exc .value .error == ErrorData (code = INVALID_PARAMS , message = "Invalid request parameters" , data = "" )
256+ assert exc .value .error == ErrorData (
257+ code = INVALID_REQUEST ,
258+ message = "Received request before initialization was complete" ,
259+ data = "The client must send an 'initialize' request before any other request." ,
260+ )
253261
254262
255263@pytest .mark .anyio
@@ -632,7 +640,11 @@ async def reject_initialize(ctx: Ctx, call_next: Any) -> Any:
632640 assert exc .value .error .message == "rejected by middleware"
633641 with pytest .raises (MCPError ) as gate_exc :
634642 await client .send_raw_request ("tools/list" , None )
635- assert gate_exc .value .error == ErrorData (code = INVALID_PARAMS , message = "Invalid request parameters" , data = "" )
643+ assert gate_exc .value .error == ErrorData (
644+ code = INVALID_REQUEST ,
645+ message = "Received request before initialization was complete" ,
646+ data = "The client must send an 'initialize' request before any other request." ,
647+ )
636648 # ping passes through the middleware untouched
637649 assert await client .send_raw_request ("ping" , None ) == {}
638650 assert runner .connection .initialize_accepted is False
0 commit comments