feat(context): Implement ContextPreset and FileViewPreset infrastructure
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user