@@ -1303,18 +1303,18 @@ async def run_tool():
13031303 tg .cancel_scope .cancel ()
13041304
13051305 # Verify we received exactly one notification
1306- assert len (captured_notifications ) == 1 # pragma: lax no cover
1307- assert isinstance (captured_notifications [0 ], types .LoggingMessageNotification ) # pragma: lax no cover
1308- assert captured_notifications [0 ].params .data == "First notification before lock" # pragma: lax no cover
1306+ assert len (captured_notifications ) == 1
1307+ assert isinstance (captured_notifications [0 ], types .LoggingMessageNotification )
1308+ assert captured_notifications [0 ].params .data == "First notification before lock"
13091309
13101310 # Clear notifications for the second phase
1311- captured_notifications = [] # pragma: lax no cover
1311+ captured_notifications = []
13121312
13131313 # Now resume the session with the same mcp-session-id and protocol version
1314- headers : dict [str , Any ] = {} # pragma: lax no cover
1315- if captured_session_id : # pragma: lax no cover
1314+ headers : dict [str , Any ] = {}
1315+ if captured_session_id :
13161316 headers [MCP_SESSION_ID_HEADER ] = captured_session_id
1317- if captured_protocol_version : # pragma: lax no cover
1317+ if captured_protocol_version :
13181318 headers [MCP_PROTOCOL_VERSION_HEADER ] = captured_protocol_version
13191319
13201320 async with create_mcp_http_client (headers = headers ) as httpx_client :
@@ -1346,8 +1346,8 @@ async def run_tool():
13461346 # We should have received the remaining notifications
13471347 assert len (captured_notifications ) == 1
13481348
1349- assert isinstance (captured_notifications [0 ], types .LoggingMessageNotification ) # pragma: lax no cover
1350- assert captured_notifications [0 ].params .data == "Second notification after lock" # pragma: lax no cover
1349+ assert isinstance (captured_notifications [0 ], types .LoggingMessageNotification )
1350+ assert captured_notifications [0 ].params .data == "Second notification after lock"
13511351
13521352
13531353@pytest .mark .anyio
0 commit comments