Skip to content

Commit bb135cb

Browse files
committed
test(client): improve readability in BaseClient send_message tests
1 parent 234e23c commit bb135cb

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tests/client/test_base_client.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,9 @@ async def test_send_message_uses_callsite_configuration_partial_override_non_str
139139

140140
mock_transport.send_message.assert_called_once()
141141
assert not mock_transport.send_message_streaming.called
142-
assert len(events) == 1 and events[0][0].id == 'task-cfg-ns-1'
142+
assert len(events) == 1
143+
task, _ = events[0]
144+
assert task.id == 'task-cfg-ns-1'
143145

144146
params = mock_transport.send_message.await_args.args[0]
145147
assert params.configuration.history_length == 2
@@ -170,7 +172,9 @@ async def test_send_message_ignores_none_fields_in_callsite_configuration_non_st
170172
]
171173

172174
mock_transport.send_message.assert_called_once()
173-
assert len(events) == 1 and events[0][0].id == 'task-cfg-ns-2'
175+
assert len(events) == 1
176+
task, _ = events[0]
177+
assert task.id == 'task-cfg-ns-2'
174178

175179
params = mock_transport.send_message.await_args.args[0]
176180
assert params.configuration.history_length is None
@@ -208,8 +212,8 @@ async def create_stream(*args, **kwargs):
208212
mock_transport.send_message_streaming.assert_called_once()
209213
assert not mock_transport.send_message.called
210214
assert len(events) == 1
211-
first = events[0][0] if isinstance(events[0], tuple) else events[0]
212-
assert first.id == 'task-cfg-s-1'
215+
task, _ = events[0]
216+
assert task.id == 'task-cfg-s-1'
213217

214218
params = mock_transport.send_message_streaming.call_args.args[0]
215219
assert params.configuration.history_length == 0

0 commit comments

Comments
 (0)