46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import pytest
|
|
from src.models import ContextPreset, FileViewPreset
|
|
|
|
def test_file_view_preset_serialization():
|
|
p = FileViewPreset(path="test.py", view_mode="skeleton")
|
|
d = p.to_dict()
|
|
assert d == {"path": "test.py", "view_mode": "skeleton"}
|
|
|
|
p2 = FileViewPreset.from_dict(d)
|
|
assert p2.path == "test.py"
|
|
assert p2.view_mode == "skeleton"
|
|
|
|
def test_context_preset_serialization():
|
|
f1 = FileViewPreset(path="a.py", view_mode="full")
|
|
f2 = FileViewPreset(path="b.py", view_mode="summary")
|
|
|
|
preset = ContextPreset(
|
|
name="test_preset",
|
|
files=[f1, f2],
|
|
screenshots=["shot1.png"]
|
|
)
|
|
|
|
d = preset.to_dict()
|
|
assert len(d["files"]) == 2
|
|
assert d["files"][0]["path"] == "a.py"
|
|
assert d["screenshots"] == ["shot1.png"]
|
|
|
|
p2 = ContextPreset.from_dict("test_preset", d)
|
|
assert p2.name == "test_preset"
|
|
assert len(p2.files) == 2
|
|
assert p2.files[0].view_mode == "full"
|
|
assert p2.screenshots == ["shot1.png"]
|
|
|
|
def test_context_preset_from_dict_legacy():
|
|
# Test handling of legacy string paths in preset files
|
|
d = {
|
|
"files": ["legacy.py", {"path": "new.py", "view_mode": "skeleton"}],
|
|
"screenshots": []
|
|
}
|
|
preset = ContextPreset.from_dict("legacy_test", d)
|
|
assert len(preset.files) == 2
|
|
assert preset.files[0].path == "legacy.py"
|
|
assert preset.files[0].view_mode == "summary" # Default
|
|
assert preset.files[1].path == "new.py"
|
|
assert preset.files[1].view_mode == "skeleton"
|