feat(context): Add view_mode to FileItem model
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user