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

2
project_history.toml
View File

@@ -8,5 +8,5 @@ active = "main"
[discussions.main] [discussions.main]
git_commit = "" git_commit = ""
last_updated = "2026-03-05T17:14:44" last_updated = "2026-03-05T17:16:56"
history = [] history = []

5
tests/test_api_events.py
View File

@@ -43,7 +43,7 @@ def test_send_emits_events_proper() -> None:
mock_response = MagicMock() mock_response = MagicMock()
mock_response.candidates = [MockCandidate([MockPart("gemini response", None)])] mock_response.candidates = [MockCandidate([MockPart("gemini response", None)])]
mock_response.usage_metadata = MockUsage() 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() start_callback = MagicMock()
response_callback = MagicMock() response_callback = MagicMock()
ai_client.events.on("request_start", start_callback) 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 = MagicMock()
mock_response_final.candidates = [MockCandidate([MockPart("final answer", None)])] mock_response_final.candidates = [MockCandidate([MockPart("final answer", None)])]
mock_response_final.usage_metadata = MockUsage() 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_chat.send_message.side_effect = [mock_response_with_tool, mock_response_final]
mock_dispatch.return_value = "file content" mock_dispatch.return_value = "file content"
ai_client.set_provider("gemini", "gemini-2.5-flash-lite") 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}") print(f"DEBUG_TOOL_EVENT: {args} {kwargs}")
tool_callback(*args, **kwargs) tool_callback(*args, **kwargs)
ai_client.events.on("tool_execution", debug_tool) 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}") print(f"DEBUG_RESULT: {result}")
# Should be called twice: once for 'started', once for 'completed' # Should be called twice: once for 'started', once for 'completed'
assert tool_callback.call_count == 2 assert tool_callback.call_count == 2