@@ -45,7 +45,7 @@ async def test_no_progress_no_reset_timeout_fires():
4545
4646 async with create_client_server_memory_streams () as (client_streams , server_streams ):
4747 client_read , client_write = client_streams
48- server_read , server_write = server_streams
48+ server_read , _server_write = server_streams
4949
5050 async def mock_server ():
5151 """Receive the request but never respond."""
@@ -84,6 +84,7 @@ async def mock_server():
8484 """Receive the request, send progress at ~50% of the timeout window,
8585 then respond after the original timeout would have expired."""
8686 msg = await server_read .receive ()
87+ assert isinstance (msg , SessionMessage )
8788 assert isinstance (msg .message , JSONRPCRequest )
8889 request_id = msg .message .id
8990
@@ -137,6 +138,7 @@ async def mock_server():
137138 """Send multiple progress notifications that keep resetting the
138139 per-window timeout but eventually exceed max_total_timeout."""
139140 msg = await server_read .receive ()
141+ assert isinstance (msg , SessionMessage )
140142 assert isinstance (msg .message , JSONRPCRequest )
141143 request_id = msg .message .id
142144
@@ -182,6 +184,7 @@ async def test_progress_stops_timeout_fires():
182184 async def mock_server ():
183185 """Send a few progress notifications, then stop."""
184186 msg = await server_read .receive ()
187+ assert isinstance (msg , SessionMessage )
185188 assert isinstance (msg .message , JSONRPCRequest )
186189 request_id = msg .message .id
187190
@@ -224,6 +227,7 @@ async def mock_server():
224227 """Send progress every 80ms with a 0.3s timeout (3 progress
225228 notifications), then respond."""
226229 msg = await server_read .receive ()
230+ assert isinstance (msg , SessionMessage )
227231 assert isinstance (msg .message , JSONRPCRequest )
228232 request_id = msg .message .id
229233
@@ -278,6 +282,7 @@ async def test_reset_timeout_false_by_default():
278282 async def mock_server ():
279283 """Send progress before the timeout, then wait."""
280284 msg = await server_read .receive ()
285+ assert isinstance (msg , SessionMessage )
281286 assert isinstance (msg .message , JSONRPCRequest )
282287 request_id = msg .message .id
283288
0 commit comments