From e2ca7db7aba5252528c82ad3f793de3d6e28be01 Mon Sep 17 00:00:00 2001 From: Ed_ Date: Thu, 5 Mar 2026 17:22:54 -0500 Subject: [PATCH] fix(tests): Fix google-genai streaming mocks in api_events tests --- project_history.toml | 2 +- tests/test_api_events.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/project_history.toml b/project_history.toml index 722824f..7700366 100644 --- a/project_history.toml +++ b/project_history.toml @@ -8,5 +8,5 @@ active = "main" [discussions.main] git_commit = "" -last_updated = "2026-03-05T17:14:44" +last_updated = "2026-03-05T17:16:56" history = [] diff --git a/tests/test_api_events.py b/tests/test_api_events.py index 549d04b..8a4057f 100644 --- a/tests/test_api_events.py +++ b/tests/test_api_events.py @@ -43,7 +43,7 @@ def test_send_emits_events_proper() -> None: mock_response = MagicMock() mock_response.candidates = [MockCandidate([MockPart("gemini response", None)])] mock_response.usage_metadata = MockUsage() - mock_chat.send_message.return_value = mock_response + mock_chat.send_message_stream.return_value = mock_response start_callback = MagicMock() response_callback = MagicMock() ai_client.events.on("request_start", start_callback) @@ -73,6 +73,7 @@ def test_send_emits_tool_events() -> None: mock_response_final = MagicMock() mock_response_final.candidates = [MockCandidate([MockPart("final answer", None)])] mock_response_final.usage_metadata = MockUsage() + mock_chat.send_message_stream.side_effect = [mock_response_with_tool, mock_response_final] mock_chat.send_message.side_effect = [mock_response_with_tool, mock_response_final] mock_dispatch.return_value = "file content" ai_client.set_provider("gemini", "gemini-2.5-flash-lite") @@ -81,7 +82,7 @@ def test_send_emits_tool_events() -> None: print(f"DEBUG_TOOL_EVENT: {args} {kwargs}") tool_callback(*args, **kwargs) ai_client.events.on("tool_execution", debug_tool) - result = ai_client.send("context", "message") + result = ai_client.send("context", "message", stream_callback=lambda x: None) print(f"DEBUG_RESULT: {result}") # Should be called twice: once for 'started', once for 'completed' assert tool_callback.call_count == 2