checkpoint(Saved system prompt presets)

tests/test_presets.py

@@ -0,0 +1,77 @@
+import os
+import unittest
+from pathlib import Path
+import tempfile
+import shutil
+from src.presets import PresetManager
+from src.models import Preset
+
+class TestPresetManager(unittest.TestCase):
+ def setUp(self):
+  self.test_dir = Path(tempfile.mkdtemp())
+  self.project_root = self.test_dir / "project"
+  self.project_root.mkdir()
+  
+  # Mocking global path is harder since it's hardcoded in paths.py
+  # But we can at least test project-specific ones and the manager's logic.
+  # For the sake of this test, we will only test what we can without 
+  # affecting the real global_presets.toml if possible.
+  
+  self.manager = PresetManager(project_root=self.project_root)
+  # Override paths for testing to avoid touching real files
+  self.manager.global_path = self.test_dir / "global_presets.toml"
+
+ def tearDown(self):
+  shutil.rmtree(self.test_dir)
+
+ def test_save_and_load_global(self):
+  preset = Preset(name="test_global", system_prompt="You are a global assistant")
+  self.manager.save_preset(preset, scope="global")
+  
+  presets = self.manager.load_all()
+  self.assertIn("test_global", presets)
+  self.assertEqual(presets["test_global"].system_prompt, "You are a global assistant")
+
+ def test_save_and_load_project(self):
+  preset = Preset(name="test_project", system_prompt="You are a project assistant")
+  self.manager.save_preset(preset, scope="project")
+  
+  presets = self.manager.load_all()
+  self.assertIn("test_project", presets)
+  self.assertEqual(presets["test_project"].system_prompt, "You are a project assistant")
+
+ def test_project_overwrites_global(self):
+  g_preset = Preset(name="shared", system_prompt="Global version")
+  p_preset = Preset(name="shared", system_prompt="Project version")
+  
+  self.manager.save_preset(g_preset, scope="global")
+  self.manager.save_preset(p_preset, scope="project")
+  
+  presets = self.manager.load_all()
+  self.assertEqual(presets["shared"].system_prompt, "Project version")
+
+ def test_delete_preset(self):
+  preset = Preset(name="to_delete", system_prompt="Delete me")
+  self.manager.save_preset(preset, scope="project")
+  
+  presets = self.manager.load_all()
+  self.assertIn("to_delete", presets)
+  
+  self.manager.delete_preset("to_delete", scope="project")
+  presets = self.manager.load_all()
+  self.assertNotIn("to_delete", presets)
+
+ def test_dynamic_project_path(self):
+  """Verifies that project_path updates when project_root changes."""
+  initial_root = self.test_dir / "project1"
+  initial_root.mkdir()
+  manager = PresetManager(project_root=initial_root)
+  self.assertEqual(manager.project_path, initial_root / "project_presets.toml")
+  
+  new_root = self.test_dir / "project2"
+  new_root.mkdir()
+  manager.project_root = new_root
+  self.assertEqual(manager.project_path, new_root / "project_presets.toml")
+
+if __name__ == "__main__":
+ unittest.main()