Skip to content

Commit 5547ba1

Browse files
committed
Gemini suggestion
1 parent 43618f0 commit 5547ba1

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

tests/server/request_handlers/test_default_request_handler.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1242,14 +1242,11 @@ def sync_get_event_stream_gen(*args, **kwargs):
12421242
)
12431243

12441244
# Mock current_result as an async property returning events sequentially.
1245-
async def get_current_result_coro1():
1246-
return event1_task_update
1247-
1248-
async def get_current_result_coro2():
1249-
return event2_final_task
1245+
async def to_coro(val):
1246+
return val
12501247

12511248
type(mock_result_aggregator_instance).current_result = PropertyMock(
1252-
side_effect=[get_current_result_coro1(), get_current_result_coro2()]
1249+
side_effect=[to_coro(event1_task_update), to_coro(event2_final_task)]
12531250
)
12541251

12551252
context = create_server_call_context()

0 commit comments

Comments
 (0)