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/
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
from pathlib import Path
|
|
from src.app_controller import AppController
|
|
from src.events import UserRequestEvent
|
|
from src import events
|
|
from src.result_types import Result
|
|
|
|
@pytest.fixture
|
|
def controller():
|
|
with (
|
|
patch('src.app_controller.AppController.load_config', return_value={
|
|
"ai": {"provider": "gemini", "model": "model-1"},
|
|
"projects": {"paths": [], "active": ""},
|
|
"gui": {"show_windows": {}}
|
|
}),
|
|
patch('src.project_manager.load_project', return_value={}),
|
|
patch('src.project_manager.migrate_from_legacy_config', return_value={}),
|
|
patch('src.project_manager.save_project'),
|
|
patch('src.session_logger.open_session'),
|
|
patch('src.session_logger.reset_session'),
|
|
patch('src.app_controller.AppController._init_ai_and_hooks'),
|
|
patch('src.app_controller.AppController._fetch_models')
|
|
):
|
|
c = AppController()
|
|
# Mock necessary state
|
|
c.ui_files_base_dir = "."
|
|
c.event_queue = MagicMock()
|
|
return c
|
|
|
|
def test_handle_request_event_appends_definitions(controller):
|
|
# Setup
|
|
file_items = [{"path": "src/models.py", "entry": "src/models.py"}]
|
|
event = UserRequestEvent(
|
|
prompt="Explain @Track object",
|
|
stable_md="stable_md",
|
|
file_items=file_items,
|
|
disc_text="disc_text",
|
|
base_dir="."
|
|
)
|
|
|
|
# Mock symbol helpers
|
|
with (
|
|
patch('src.app_controller.parse_symbols', return_value=["Track"]) as mock_parse,
|
|
patch('src.app_controller.get_symbol_definition', return_value=("src/models.py", "class Track: pass", 42)) as mock_get_def,
|
|
patch('src.ai_client.send', return_value=Result(data="mocked response")) as mock_send
|
|
):
|
|
# Execute
|
|
controller._handle_request_event(event)
|
|
|
|
# Verify
|
|
mock_parse.assert_called_once_with("Explain @Track object")
|
|
mock_get_def.assert_called_once()
|
|
|
|
# Check if enriched prompt was sent to AI
|
|
expected_suffix = "\n\n[Definition: Track from src/models.py (line 42)]\n```python\nclass Track: pass\n```"
|
|
mock_send.assert_called_once()
|
|
args, kwargs = mock_send.call_args
|
|
sent_prompt = args[1]
|
|
assert sent_prompt == "Explain @Track object" + expected_suffix
|
|
|
|
def test_handle_request_event_no_symbols(controller):
|
|
# Setup
|
|
file_items = [{"path": "src/models.py", "entry": "src/models.py"}]
|
|
event = UserRequestEvent(
|
|
prompt="Just a normal prompt",
|
|
stable_md="stable_md",
|
|
file_items=file_items,
|
|
disc_text="disc_text",
|
|
base_dir="."
|
|
)
|
|
|
|
with (
|
|
patch('src.app_controller.parse_symbols', return_value=[]) as mock_parse,
|
|
patch('src.ai_client.send', return_value=Result(data="mocked response")) as mock_send
|
|
):
|
|
# Execute
|
|
controller._handle_request_event(event)
|
|
|
|
# Verify
|
|
mock_send.assert_called_once()
|
|
args, kwargs = mock_send.call_args
|
|
sent_prompt = args[1]
|
|
assert sent_prompt == "Just a normal prompt"
|