ed
ada9617308
Batch rename of 22 test files. 62 references renamed total. The full test suite is now GREEN again, matching the pre-rename baseline from Task 1.1. Pure mechanical rename. No behavior change. Files affected: test_ai_cache_tracking, test_ai_client_cli, test_ai_client_result, test_api_events, test_context_pruner, test_deepseek_provider, test_gemini_cli_* (3 files), test_gui2_mcp, test_headless_* (2 files), test_live_gui_integration_v2, test_orchestration_logic, test_phase6_engine, test_rag_integration, test_run_worker_lifecycle_abort, test_spawn_interception_v2, test_symbol_parsing, test_tier4_interceptor, test_tiered_aggregation, test_token_usage. Note: spec estimated 24 files; actual is 22 (test_deprecation_warnings no longer exists, and 1 fewer file than spec's list). Refs: conductor/tracks/send_result_to_send_20260616/
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
import unittest.mock
|
|
from unittest.mock import patch, MagicMock
|
|
from src.gui_2 import App
|
|
from src import ai_client
|
|
from src.result_types import Result
|
|
|
|
def test_mcp_tool_call_is_dispatched(app_instance: App) -> None:
|
|
"""
|
|
|
|
|
|
This test verifies that when the AI returns a tool call for an MCP function,
|
|
the ai_client correctly dispatches it to mcp_client.
|
|
This will fail until mcp_client is properly integrated.
|
|
"""
|
|
# 1. Define the mock tool call from the AI
|
|
mock_fc = MagicMock()
|
|
mock_fc.name = "read_file"
|
|
mock_fc.args = {"file_path": "test.txt"}
|
|
# 2. Construct the mock AI response (Gemini format)
|
|
mock_response_with_tool = MagicMock()
|
|
mock_response_with_tool.text = ""
|
|
mock_part = MagicMock()
|
|
mock_part.text = ""
|
|
mock_part.function_call = mock_fc
|
|
mock_candidate = MagicMock()
|
|
mock_candidate.content.parts = [mock_part]
|
|
mock_candidate.finish_reason.name = "TOOL_CALLING"
|
|
mock_response_with_tool.candidates = [mock_candidate]
|
|
|
|
class DummyUsage:
|
|
prompt_token_count = 100
|
|
candidates_token_count = 10
|
|
cached_content_token_count = 0
|
|
mock_response_with_tool.usage_metadata = DummyUsage()
|
|
# 3. Create a mock for the final AI response after the tool call
|
|
mock_response_final = MagicMock()
|
|
mock_response_final.text = "Final answer"
|
|
mock_response_final.candidates = []
|
|
mock_response_final.usage_metadata = DummyUsage()
|
|
# 4. Patch the necessary components
|
|
with patch("src.ai_client._ensure_gemini_client"), \
|
|
patch("src.ai_client._gemini_client") as mock_client, \
|
|
patch("src.mcp_client.async_dispatch", new_callable=unittest.mock.AsyncMock, 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")
|
|
# 5. Call the send function
|
|
result = ai_client.send(
|
|
md_content="some context",
|
|
user_message="read the file",
|
|
base_dir=".",
|
|
file_items=[],
|
|
discussion_history=""
|
|
)
|
|
assert result.ok
|
|
# 6. Assert that the MCP dispatch function was called
|
|
mock_dispatch.assert_called_once_with("read_file", {"file_path": "test.txt"}) |