78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
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()
|