69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
from typing import Generator
|
|
import pytest
|
|
from unittest.mock import patch, Mock
|
|
from src import ai_client
|
|
from src.gui_2 import App
|
|
|
|
@pytest.fixture
|
|
def app_instance() -> Generator[App, None, None]:
|
|
with (
|
|
patch('src.models.load_config', return_value={'ai': {'provider': 'gemini', 'model': 'gemini-2.5-flash-lite'}, 'projects': {}}),
|
|
patch('src.models.save_config'),
|
|
patch('src.gui_2.project_manager'),
|
|
patch('src.gui_2.session_logger'),
|
|
patch('src.gui_2.immapp.run'),
|
|
patch('src.app_controller.AppController._load_active_project'),
|
|
patch('src.app_controller.AppController._fetch_models'),
|
|
patch.object(App, '_load_fonts'),
|
|
patch.object(App, '_post_init'),
|
|
patch('src.app_controller.AppController._prune_old_logs'),
|
|
patch('src.app_controller.AppController.start_services'),
|
|
# Do not patch _init_ai_and_hooks to ensure _settable_fields is initialized
|
|
patch('src.api_hooks.HookServer'),
|
|
patch('src.ai_client.set_provider'),
|
|
patch('src.ai_client.reset_session')
|
|
):
|
|
app = App()
|
|
yield app
|
|
|
|
def test_redundant_calls_in_process_pending_gui_tasks(app_instance: App) -> None:
|
|
app_instance.controller._pending_gui_tasks = [
|
|
{'action': 'set_value', 'item': 'current_provider', 'value': 'anthropic'}
|
|
]
|
|
with patch('src.ai_client.set_provider') as mock_set_provider, \
|
|
patch('src.ai_client.reset_session') as mock_reset_session:
|
|
app_instance.controller._process_pending_gui_tasks()
|
|
assert mock_set_provider.call_count == 1
|
|
assert mock_reset_session.call_count == 1
|
|
|
|
def test_gcli_path_updates_adapter(app_instance: App) -> None:
|
|
app_instance.controller.current_provider = 'gemini_cli'
|
|
app_instance.controller._pending_gui_tasks = [
|
|
{'action': 'set_value', 'item': 'gcli_path', 'value': '/new/path/to/gemini'}
|
|
]
|
|
# Initialize adapter if it doesn't exist (it shouldn't in mock env)
|
|
ai_client._gemini_cli_adapter = None
|
|
app_instance.controller._process_pending_gui_tasks()
|
|
assert ai_client._gemini_cli_adapter is not None
|
|
assert ai_client._gemini_cli_adapter.binary_path == '/new/path/to/gemini'
|
|
|
|
def test_process_pending_gui_tasks_drag(app_instance: App) -> None:
|
|
"""Test that the drag action is correctly processed and dispatches to the registered callback."""
|
|
mock_callback = Mock()
|
|
app_instance.controller._drag_actions["src_id"] = mock_callback
|
|
app_instance.controller._pending_gui_tasks = [
|
|
{"action": "drag", "src_item": "src_id", "dst_item": "dst_id"}
|
|
]
|
|
app_instance.controller._process_pending_gui_tasks()
|
|
mock_callback.assert_called_once_with(dst_item="dst_id")
|
|
|
|
def test_process_pending_gui_tasks_right_click(app_instance: App) -> None:
|
|
"""Test that the right_click action is correctly processed and dispatches to the registered callback."""
|
|
mock_callback = Mock()
|
|
app_instance.controller._right_clickable_actions["item_id"] = mock_callback
|
|
app_instance.controller._pending_gui_tasks = [
|
|
{"action": "right_click", "item": "item_id"}
|
|
]
|
|
app_instance.controller._process_pending_gui_tasks()
|
|
mock_callback.assert_called_once()
|