Skip to content

Commit a41d1be

Browse files
committed
Unify call_args pattern in tests
1 parent bb5ba72 commit a41d1be

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/client/test_base_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ async def test_send_message_callsite_config_overrides_non_streaming(
157157
task, _ = events[0]
158158
assert task.id == 'task-cfg-ns-1'
159159

160-
params = mock_transport.send_message.await_args.args[0]
160+
params = mock_transport.send_message.call_args[0][0]
161161
assert params.configuration.history_length == 2
162162
assert params.configuration.blocking is False
163163
assert params.configuration.accepted_output_modes == ['application/json']
@@ -197,7 +197,7 @@ async def create_stream(*args, **kwargs):
197197
task, _ = events[0]
198198
assert task.id == 'task-cfg-s-1'
199199

200-
params = mock_transport.send_message_streaming.call_args.args[0]
200+
params = mock_transport.send_message_streaming.call_args[0][0]
201201
assert params.configuration.history_length == 0
202202
assert params.configuration.blocking is True
203203
assert params.configuration.accepted_output_modes == ['text/plain']

0 commit comments

Comments
 (0)