60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import pytest
|
|
from src.context_presets import ContextPresetManager
|
|
from src.models import ContextPreset, ContextFileEntry
|
|
|
|
def test_save_context_preset():
|
|
manager = ContextPresetManager()
|
|
project_dict = {}
|
|
preset_name = "test_preset"
|
|
files = [ContextFileEntry(path="file1.py"), ContextFileEntry(path="file2.py")]
|
|
screenshots = ["screenshot1.png"]
|
|
|
|
preset = ContextPreset(name=preset_name, files=files, screenshots=screenshots)
|
|
manager.save_preset(project_dict, preset)
|
|
|
|
assert "context_presets" in project_dict
|
|
assert preset_name in project_dict["context_presets"]
|
|
assert project_dict["context_presets"][preset_name]["files"] == [f.to_dict() for f in files]
|
|
assert project_dict["context_presets"][preset_name]["screenshots"] == screenshots
|
|
|
|
def test_load_all_context_presets():
|
|
manager = ContextPresetManager()
|
|
project_dict = {
|
|
"context_presets": {
|
|
"test_preset": {
|
|
"name": "test_preset",
|
|
"files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
|
|
"screenshots": ["screenshot1.png"]
|
|
}
|
|
}
|
|
}
|
|
|
|
presets = manager.load_all(project_dict)
|
|
|
|
assert "test_preset" in presets
|
|
assert presets["test_preset"].files[0].path == "file1.py"
|
|
assert presets["test_preset"].screenshots == ["screenshot1.png"]
|
|
|
|
def test_delete_context_preset():
|
|
manager = ContextPresetManager()
|
|
project_dict = {
|
|
"context_presets": {
|
|
"test_preset": {
|
|
"name": "test_preset",
|
|
"files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
|
|
"screenshots": []
|
|
}
|
|
}
|
|
}
|
|
|
|
manager.delete_preset(project_dict, "test_preset")
|
|
|
|
assert "test_preset" not in project_dict["context_presets"]
|
|
|
|
def test_delete_nonexistent_preset_no_error():
|
|
manager = ContextPresetManager()
|
|
project_dict = {"context_presets": {}}
|
|
# Should not raise error if it doesn't exist
|
|
manager.delete_preset(project_dict, "nonexistent")
|
|
assert "nonexistent" not in project_dict["context_presets"]
|