import pytest
from src.context_presets import ContextPresetManager
from src.models import ContextPreset, ContextFileEntry

def test_save_context_preset():
 manager = ContextPresetManager()
 project_dict = {}
 preset_name = "test_preset"
 files = [ContextFileEntry(path="file1.py"), ContextFileEntry(path="file2.py")]
 screenshots = ["screenshot1.png"]
 
 preset = ContextPreset(name=preset_name, files=files, screenshots=screenshots)
 manager.save_preset(project_dict, preset)
 
 assert "context_presets" in project_dict
 assert preset_name in project_dict["context_presets"]
 assert project_dict["context_presets"][preset_name]["files"] == [f.to_dict() for f in files]
 assert project_dict["context_presets"][preset_name]["screenshots"] == screenshots

def test_load_all_context_presets():
 manager = ContextPresetManager()
 project_dict = {
  "context_presets": {
   "test_preset": {
    "name": "test_preset",
    "files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
    "screenshots": ["screenshot1.png"]
   }
  }
 }
 
 presets = manager.load_all(project_dict)
 
 assert "test_preset" in presets
 assert presets["test_preset"].files[0].path == "file1.py"
 assert presets["test_preset"].screenshots == ["screenshot1.png"]

def test_delete_context_preset():
 manager = ContextPresetManager()
 project_dict = {
  "context_presets": {
   "test_preset": {
    "name": "test_preset",
    "files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
    "screenshots": []
   }
  }
 }
 
 manager.delete_preset(project_dict, "test_preset")
 
 assert "test_preset" not in project_dict["context_presets"]

def test_delete_nonexistent_preset_no_error():
 manager = ContextPresetManager()
 project_dict = {"context_presets": {}}
 # Should not raise error if it doesn't exist
 manager.delete_preset(project_dict, "nonexistent")
 assert "nonexistent" not in project_dict["context_presets"]