feat(context): Add view_mode to FileItem model

This commit is contained in:
2026-05-11 10:28:35 -04:00
parent a1343eebe6
commit 8addb97018
2 changed files with 11 additions and 3 deletions
+6 -1
View File
@@ -7,17 +7,19 @@ def test_file_item_fields():
assert item.path == "src/models.py"
assert item.auto_aggregate is True
assert item.force_full is False
assert item.view_mode == "full"
assert item.ast_mask == {}
assert item.custom_slices == []
assert item.injected_at is None
def test_file_item_to_dict():
"""Test that FileItem can be serialized to a dict."""
item = FileItem(path="test.py", auto_aggregate=False, force_full=True)
item = FileItem(path="test.py", auto_aggregate=False, force_full=True, view_mode="summary")
expected = {
"path": "test.py",
"auto_aggregate": False,
"force_full": True,
"view_mode": "summary",
"ast_signatures": False,
"ast_definitions": False,
"ast_mask": {},
@@ -32,12 +34,14 @@ def test_file_item_from_dict():
"path": "test.py",
"auto_aggregate": False,
"force_full": True,
"view_mode": "summary",
"injected_at": 123.456
}
item = FileItem.from_dict(data)
assert item.path == "test.py"
assert item.auto_aggregate is False
assert item.force_full is True
assert item.view_mode == "summary"
assert item.injected_at == 123.456
assert item.ast_mask == {}
@@ -48,6 +52,7 @@ def test_file_item_from_dict_defaults():
assert item.path == "test.py"
assert item.auto_aggregate is True
assert item.force_full is False
assert item.view_mode == "full"
assert item.ast_mask == {}
assert item.custom_slices == []
assert item.injected_at is None