ed
6a3c142bfa
Removed 18 unused 'from src import models' imports. Migrated the explicit sub-imports: - 'from src.models import PROVIDERS' -> 'from src.ai_client import PROVIDERS' (test_minimax_provider x2, test_deepseek_infra) - 'from src.models import Metadata' -> 'from src.type_aliases import Metadata' (test_project_manager_tracks, test_track_state_persistence, test_track_state_schema) Kept the 2 'import src.models as models' in test_provider_curation + test_providers_source_of_truth (they verify the backward-compat shim still re-exports PROVIDERS) and the 2 self-tests (test_models_no_top_level_pydantic + test_models_no_top_level_tomli_w) that exercise the shim's no-leak invariants.
118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
import os
|
|
import pytest
|
|
import copy
|
|
from src.project_files import FileItem, NamedViewPreset
|
|
from src.app_controller import AppController
|
|
|
|
@pytest.fixture
|
|
def controller(tmp_path):
|
|
# Create a mock project file
|
|
proj_path = tmp_path / "test_project.toml"
|
|
proj_path.write_text("[project]\nname = 'test'\n")
|
|
|
|
ctrl = AppController()
|
|
ctrl.active_project_path = str(proj_path)
|
|
ctrl.project = {"project": {"name": "test"}}
|
|
ctrl.view_presets = []
|
|
|
|
# Initialize missing attributes needed for flush/refresh
|
|
ctrl.ui_output_dir = "./md_gen"
|
|
ctrl.ui_files_base_dir = "."
|
|
ctrl.ui_shots_base_dir = "."
|
|
ctrl.ui_project_git_dir = ""
|
|
ctrl.ui_project_conductor_dir = "conductor"
|
|
ctrl.ui_project_system_prompt = ""
|
|
ctrl.ui_project_preset_name = None
|
|
|
|
ctrl.ui_auto_add_history = False
|
|
ctrl.ui_auto_scroll_comms = True
|
|
ctrl.ui_auto_scroll_tool_calls = True
|
|
ctrl.ui_agent_tools = {}
|
|
ctrl.ui_epic_input = ""
|
|
ctrl.ui_active_context_preset = ""
|
|
ctrl.preset_manager = type('Mock', (), {'load_all': lambda self: {}, 'project_root': None})()
|
|
ctrl.tool_preset_manager = type('Mock', (), {'load_all_presets': lambda self: {}, 'load_all_bias_profiles': lambda self: {}, 'project_root': None})()
|
|
ctrl.persona_manager = type('Mock', (), {'load_all': lambda self: {}})()
|
|
|
|
return ctrl
|
|
|
|
def test_save_view_preset(controller):
|
|
f_item = FileItem(path="test.py", view_mode="skeleton")
|
|
f_item.ast_mask = {"test::func": "sig"}
|
|
f_item.custom_slices = [{"start_line": 1, "end_line": 10}]
|
|
|
|
controller._cb_save_view_preset("my_preset", f_item)
|
|
|
|
assert any(vp.name == "my_preset" for vp in controller.view_presets)
|
|
preset = next(vp for vp in controller.view_presets if vp.name == "my_preset")
|
|
assert preset.view_mode == "skeleton"
|
|
assert preset.ast_mask == {"test::func": "sig"}
|
|
assert preset.custom_slices == [{"start_line": 1, "end_line": 10}]
|
|
|
|
# Verify persistence
|
|
controller._flush_to_project()
|
|
assert "view_presets" in controller.project
|
|
assert isinstance(controller.project["view_presets"], list)
|
|
assert any(vp["name"] == "my_preset" for vp in controller.project["view_presets"])
|
|
|
|
def test_apply_view_preset(controller):
|
|
# Setup a preset
|
|
preset = NamedViewPreset(
|
|
name="my_preset",
|
|
view_mode="masked",
|
|
ast_mask={"main::run": "def"},
|
|
custom_slices=[{"start_line": 5, "end_line": 15}]
|
|
)
|
|
controller.view_presets.append(preset)
|
|
|
|
# Create a file item to apply to
|
|
f_item = FileItem(path="main.py", view_mode="summary")
|
|
|
|
controller._cb_apply_view_preset("my_preset", f_item)
|
|
|
|
assert f_item.view_mode == "masked"
|
|
assert f_item.ast_mask == {"main::run": "def"}
|
|
assert f_item.custom_slices == [{"start_line": 5, "end_line": 15}]
|
|
|
|
def test_delete_view_preset(controller):
|
|
preset = NamedViewPreset(name="to_del", view_mode="full")
|
|
controller.view_presets.append(preset)
|
|
|
|
controller._cb_delete_view_preset("to_del")
|
|
|
|
assert not any(vp.name == "to_del" for vp in controller.view_presets)
|
|
|
|
def test_load_presets_from_project_list(controller):
|
|
controller.project["view_presets"] = [
|
|
{
|
|
"name": "stored_preset",
|
|
"view_mode": "outline",
|
|
"ast_mask": {"a": "b"},
|
|
"custom_slices": []
|
|
}
|
|
]
|
|
|
|
controller._refresh_from_project()
|
|
|
|
assert any(vp.name == "stored_preset" for vp in controller.view_presets)
|
|
preset = next(vp for vp in controller.view_presets if vp.name == "stored_preset")
|
|
assert preset.view_mode == "outline"
|
|
assert preset.ast_mask == {"a": "b"}
|
|
|
|
def test_load_presets_from_project_legacy_dict(controller):
|
|
# Test backward compatibility
|
|
controller.project["view_presets"] = {
|
|
"legacy_preset": {
|
|
"view_mode": "full",
|
|
"ast_mask": {"c": "d"},
|
|
"custom_slices": []
|
|
}
|
|
}
|
|
|
|
controller._refresh_from_project()
|
|
|
|
assert any(vp.name == "legacy_preset" for vp in controller.view_presets)
|
|
preset = next(vp for vp in controller.view_presets if vp.name == "legacy_preset")
|
|
assert preset.view_mode == "full"
|
|
assert preset.ast_mask == {"c": "d"}
|