60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import pytest
|
|
from src.project_manager import (
|
|
save_context_preset,
|
|
load_context_preset,
|
|
delete_context_preset
|
|
)
|
|
|
|
def test_save_context_preset():
|
|
project_dict = {}
|
|
preset_name = "test_preset"
|
|
files = ["file1.py", "file2.py"]
|
|
screenshots = ["screenshot1.png"]
|
|
|
|
save_context_preset(project_dict, preset_name, files, screenshots)
|
|
|
|
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]["screenshots"] == screenshots
|
|
|
|
def test_load_context_preset():
|
|
project_dict = {
|
|
"context_presets": {
|
|
"test_preset": {
|
|
"files": ["file1.py"],
|
|
"screenshots": ["screenshot1.png"]
|
|
}
|
|
}
|
|
}
|
|
|
|
preset = load_context_preset(project_dict, "test_preset")
|
|
|
|
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")
|
|
|
|
def test_delete_context_preset():
|
|
project_dict = {
|
|
"context_presets": {
|
|
"test_preset": {
|
|
"files": ["file1.py"],
|
|
"screenshots": []
|
|
}
|
|
}
|
|
}
|
|
|
|
delete_context_preset(project_dict, "test_preset")
|
|
|
|
assert "test_preset" not in project_dict["context_presets"]
|
|
|
|
def test_delete_nonexistent_preset_no_error():
|
|
project_dict = {"context_presets": {}}
|
|
# Should not raise error if it doesn't exist
|
|
delete_context_preset(project_dict, "nonexistent")
|
|
assert "nonexistent" not in project_dict["context_presets"]
|