checkpoint(Saved system prompt presets)
This commit is contained in:
103
tests/test_saved_presets_sim.py
Normal file
103
tests/test_saved_presets_sim.py
Normal file
@@ -0,0 +1,103 @@
|
||||
import pytest
|
||||
import time
|
||||
import tomli_w
|
||||
import os
|
||||
import json
|
||||
from pathlib import Path
|
||||
from src.api_hook_client import ApiHookClient
|
||||
|
||||
def test_preset_switching(live_gui):
|
||||
client = ApiHookClient()
|
||||
|
||||
# Paths for presets
|
||||
temp_workspace = Path("tests/artifacts/live_gui_workspace")
|
||||
global_presets_path = temp_workspace / "presets.toml"
|
||||
project_presets_path = temp_workspace / "project_presets.toml"
|
||||
manual_slop_path = temp_workspace / "manual_slop.toml"
|
||||
|
||||
# Cleanup before test
|
||||
if global_presets_path.exists(): global_presets_path.unlink()
|
||||
if project_presets_path.exists(): project_presets_path.unlink()
|
||||
|
||||
try:
|
||||
# Create a global preset
|
||||
global_presets_path.write_text(tomli_w.dumps({
|
||||
"presets": {
|
||||
"TestGlobal": {
|
||||
"system_prompt": "Global Prompt",
|
||||
"temperature": 0.7
|
||||
}
|
||||
}
|
||||
}))
|
||||
|
||||
# Create a project preset
|
||||
project_presets_path.write_text(tomli_w.dumps({
|
||||
"presets": {
|
||||
"TestProject": {
|
||||
"system_prompt": "Project Prompt",
|
||||
"temperature": 0.3
|
||||
},
|
||||
"TestGlobal": { # Override
|
||||
"system_prompt": "Overridden Prompt",
|
||||
"temperature": 0.5
|
||||
}
|
||||
}
|
||||
}))
|
||||
|
||||
# Switch to the local project to ensure context is correct
|
||||
client.push_event("custom_callback", {
|
||||
"callback": "_switch_project",
|
||||
"args": [str(manual_slop_path.absolute())]
|
||||
})
|
||||
time.sleep(2)
|
||||
|
||||
# Trigger reload of presets (just in case)
|
||||
client.push_event("custom_callback", {
|
||||
"callback": "_refresh_from_project",
|
||||
"args": []
|
||||
})
|
||||
time.sleep(2) # Wait for processing
|
||||
|
||||
# Verify where it thinks the project is
|
||||
state = client.get_gui_state()
|
||||
print(f"DEBUG: ui_files_base_dir={state.get('files_base_dir')}")
|
||||
|
||||
# Apply Global Preset (should use override from project if available in merged list)
|
||||
client.push_event("custom_callback", {
|
||||
"callback": "_apply_preset",
|
||||
"args": ["TestGlobal", "global"]
|
||||
})
|
||||
time.sleep(1)
|
||||
|
||||
# Verify state
|
||||
state = client.get_gui_state()
|
||||
assert state["global_preset_name"] == "TestGlobal"
|
||||
assert state["global_system_prompt"] == "Overridden Prompt"
|
||||
assert state["temperature"] == 0.5
|
||||
|
||||
# Apply Project Preset
|
||||
client.push_event("custom_callback", {
|
||||
"callback": "_apply_preset",
|
||||
"args": ["TestProject", "project"]
|
||||
})
|
||||
time.sleep(1)
|
||||
|
||||
state = client.get_gui_state()
|
||||
assert state["project_preset_name"] == "TestProject"
|
||||
assert state["project_system_prompt"] == "Project Prompt"
|
||||
assert state["temperature"] == 0.3
|
||||
# Select "None"
|
||||
client.push_event("custom_callback", {
|
||||
"callback": "_apply_preset",
|
||||
"args": ["None", "global"]
|
||||
})
|
||||
time.sleep(1)
|
||||
state = client.get_gui_state()
|
||||
assert not state.get("global_preset_name") # Should be None or ""
|
||||
|
||||
# Prompt remains from previous application
|
||||
assert state["global_system_prompt"] == "Overridden Prompt"
|
||||
finally:
|
||||
# Cleanup
|
||||
if global_presets_path.exists(): global_presets_path.unlink()
|
||||
if project_presets_path.exists(): project_presets_path.unlink()
|
||||
Reference in New Issue
Block a user