118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
import os
|
|
import pytest
|
|
import copy
|
|
from src import models
|
|
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_gemini_cli_path = "gemini"
|
|
ctrl.ui_word_wrap = True
|
|
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})()
|
|
|
|
return ctrl
|
|
|
|
def test_save_view_preset(controller):
|
|
f_item = models.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 = models.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 = models.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 = models.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"}
|