fix(tests): Update tests and HotReloader to accommodate gui_2 refactoring

tests/test_context_presets.py

@@ -1,59 +1,59 @@
 import pytest
-from src.project_manager import (
- save_context_preset,
- load_context_preset,
- delete_context_preset
-)
+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 = ["file1.py", "file2.py"]
+ files = [ContextFileEntry(path="file1.py"), ContextFileEntry(path="file2.py")]
  screenshots = ["screenshot1.png"]
  
- save_context_preset(project_dict, preset_name, files, screenshots)
+ 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"] == files
+ 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_context_preset():
+def test_load_all_context_presets():
+ manager = ContextPresetManager()
  project_dict = {
   "context_presets": {
    "test_preset": {
-    "files": ["file1.py"],
+    "name": "test_preset",
+    "files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
     "screenshots": ["screenshot1.png"]
    }
   }
  }
  
- preset = load_context_preset(project_dict, "test_preset")
+ presets = manager.load_all(project_dict)
  
- assert preset["files"] == ["file1.py"]
- assert preset["screenshots"] == ["screenshot1.png"]
-
-def test_load_nonexistent_preset():
- project_dict = {"context_presets": {}}
- with pytest.raises(KeyError):
-  load_context_preset(project_dict, "nonexistent")
+ 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": {
-    "files": ["file1.py"],
+    "name": "test_preset",
+    "files": [{"path": "file1.py", "view_mode": "summary", "custom_slices": []}],
     "screenshots": []
    }
   }
  }
  
- delete_context_preset(project_dict, "test_preset")
+ 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
- delete_context_preset(project_dict, "nonexistent")
+ manager.delete_preset(project_dict, "nonexistent")
  assert "nonexistent" not in project_dict["context_presets"]