Skip to content

Commit da24d49

Browse files
committed
test: use events variable instead of _ = [...] pattern in stream test (a2aproject#1091)
1 parent 4e522f2 commit da24d49

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

tests/compat/v0_3/test_rest_transport.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,14 +250,20 @@ async def mock_send_stream_request(method, path, context=None, json=None):
250250
req = SendMessageRequest(
251251
message=Message(message_id='msg-1', role=Role.ROLE_USER)
252252
)
253-
_ = [event async for event in transport.send_message_streaming(req)]
253+
events = [event async for event in transport.send_message_streaming(req)]
254+
assert len(events) == 1
255+
assert isinstance(events[0], StreamResponse)
256+
assert events[0].task.id == 'task-123'
254257

255258
assert captured_paths[-1] == '/v1/message:stream?alt=sse'
256259

257260
# Test subscribe
258261
captured_paths.clear()
259262
sub_req = SubscribeToTaskRequest(id='task-123')
260-
_ = [event async for event in transport.subscribe(sub_req)]
263+
events = [event async for event in transport.subscribe(sub_req)]
264+
assert len(events) == 1
265+
assert isinstance(events[0], StreamResponse)
266+
assert events[0].task.id == 'task-123'
261267

262268
assert captured_paths[-1] == '/v1/tasks/task-123:subscribe?alt=sse'
263269

0 commit comments

Comments
 (0)