Skip to content

Commit bdc4795

Browse files
committed
fix: resolve pyright type errors in test file
Add isinstance(msg, SessionMessage) guard before accessing .message to satisfy pyright's type narrowing. Mark unused server_write with underscore prefix.
1 parent 580564e commit bdc4795

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

tests/shared/test_progress_reset_timeout.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)