WIP: STILL FIXING FUNDAMENTAL TRASH
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
from unittest.mock import patch, MagicMock
|
||||
from gui_2 import App
|
||||
import ai_client
|
||||
from src.gui_2 import App
|
||||
from src import ai_client
|
||||
|
||||
def test_mcp_tool_call_is_dispatched(app_instance: App) -> None:
|
||||
"""
|
||||
@@ -33,9 +33,9 @@ def test_mcp_tool_call_is_dispatched(app_instance: App) -> None:
|
||||
mock_response_final.candidates = []
|
||||
mock_response_final.usage_metadata = DummyUsage()
|
||||
# 4. Patch the necessary components
|
||||
with patch("ai_client._ensure_gemini_client"), \
|
||||
patch("ai_client._gemini_client") as mock_client, \
|
||||
patch('mcp_client.dispatch', return_value="file content") as mock_dispatch:
|
||||
with patch("src.ai_client._ensure_gemini_client"), \
|
||||
patch("src.ai_client._gemini_client") as mock_client, \
|
||||
patch("src.mcp_client.dispatch", return_value="file content") as mock_dispatch:
|
||||
mock_chat = mock_client.chats.create.return_value
|
||||
mock_chat.send_message.side_effect = [mock_response_with_tool, mock_response_final]
|
||||
ai_client.set_provider("gemini", "mock-model")
|
||||
|
||||
Reference in New Issue
Block a user