ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tests): Fix google-genai streaming mocks in api_events tests

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-05 17:22:54 -05:00
parent 0c6cfa21d4
commit e2ca7db7ab
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_api_events.py
View File

@@ -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