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"