mirror of
https://github.com/vegu-ai/talemate.git
synced 2026-05-18 05:05:39 +02:00
0.36.0 (#255)
Major Features - API key encryption at rest using Fernet (OS keyring with file fallback) - Prompt Manager: unified UI with template groups, priority ordering, override tracking, response extractors - Scene context history review panel with token budgets and best-fit mode - Multiple concurrent director chats with auto-generated titles - Granular scene state reset dialog - Time passage insert/edit/delete in scene view - Image analysis via OpenAI-compatible and Talemate Client backends - Volatile context placement after scene history for improved prompt caching Improvements - Configurable narrator generation length per narration type - AI Aware conversation mode - Summarizer: custom instructions, writing style inclusion, short line filtering - Anthropic: adaptive thinking support, updated model list (opus-4-5/4-6, haiku-4-5) - Google: gemini-3.1 support - World editor: generate from topic, quick create state reinforcement, reorganized menus - Node editor: promote scene modules to global - Frontend: version mismatch detection, hideable bracket content, required scene name - TTS: improved pause handling, audio tag support for vocal markers (ElevenLabs v3) - Writing style template for AI-generated instructions - Added Kimi.jinja2 LLM prompt template - Option to disable character names in stopping strings - Client response length enforcement options - Graduated token count sliders - Increased summarizer token threshold max Bugfixes - Fix bracket/paren/brace terminators stripped from message ends - Fix colon in conversation causing content loss - Fix "Use as reference" navigating to blank page - Fix avatar regeneration and manual regenerate - Fix conversation agent ignoring generation length - Fix duplicate length instructions with reasoning enabled - Fix trailing newline on message edits - Fix summarize dialogue sending too much context with layered history - Fix layered history inspection and construction issues - Fix empty response handling in summarization - Fix context ID dot notation with dotted character names - Fix recursive retry in focal agent - Fix leading whitespace causing duplicate prepared responses - Fix summarization not stripping ANALYSIS OF lines - Fix template group selection/removal in prompt manager - Fix multiline text in parentheses/brackets parser - Fix determine_character_name resolution - Fix character activate/deactivate desyncing creative menu - Fix character image generation missing context - Fix LMStudio client not sending token limits - Fix Recent Scene images on newer Chromium - Fix sequential reinforcement messages cut off at first linebreak - Fix reinforcement removal not clearing state - Fixes #252, #256, #258 Deprecations - Removed context investigations (replaced by AI-assisted RAG mixin) - Removed deprecated prompt templates (fix-continuity-errors, fix-exposition, etc.) - Removed conversation/edit.jinja2, auto break repetition, CLI reset layered history --------- Co-authored-by: theDTV2 <47825738+theDTV2@users.noreply.github.com>
This commit is contained in:
veguAI
0
tests/server/__init__.py
Normal file
0
tests/server/__init__.py
Normal file
653
tests/server/test_prompts.py
Normal file
653
tests/server/test_prompts.py
Normal file
@@ -0,0 +1,653 @@
|
||||
"""
|
||||
Unit tests for WebSocket prompts handler.
|
||||
|
||||
Tests the PromptsPlugin handlers for template management API.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import Mock, patch, MagicMock, AsyncMock
|
||||
|
||||
from talemate.server.prompts import (
|
||||
PromptsPlugin,
|
||||
parse_template_uid,
|
||||
validate_jinja2_syntax,
|
||||
)
|
||||
|
||||
|
||||
class TestParseTemplateUid:
|
||||
"""Tests for parse_template_uid helper."""
|
||||
|
||||
def test_parses_valid_uid(self):
|
||||
"""Parses agent.template_name format correctly."""
|
||||
agent, name = parse_template_uid("narrator.narrate-scene")
|
||||
assert agent == "narrator"
|
||||
assert name == "narrate-scene"
|
||||
|
||||
def test_handles_multiple_dots(self):
|
||||
"""Handles template names with dots."""
|
||||
agent, name = parse_template_uid("narrator.my.template.name")
|
||||
assert agent == "narrator"
|
||||
assert name == "my.template.name"
|
||||
|
||||
def test_raises_on_invalid_format(self):
|
||||
"""Raises ValueError on invalid UID format."""
|
||||
with pytest.raises(ValueError, match="Invalid template UID"):
|
||||
parse_template_uid("no-dot-here")
|
||||
|
||||
|
||||
class TestValidateJinja2Syntax:
|
||||
"""Tests for validate_jinja2_syntax helper."""
|
||||
|
||||
def test_valid_template(self):
|
||||
"""Returns True for valid Jinja2 syntax."""
|
||||
valid, errors = validate_jinja2_syntax("Hello {{ name }}!")
|
||||
assert valid is True
|
||||
assert errors == []
|
||||
|
||||
def test_invalid_template(self):
|
||||
"""Returns False with errors for invalid syntax."""
|
||||
valid, errors = validate_jinja2_syntax("Hello {{ name }!")
|
||||
assert valid is False
|
||||
assert len(errors) > 0
|
||||
|
||||
def test_complex_valid_template(self):
|
||||
"""Handles complex valid templates."""
|
||||
template = """
|
||||
{% for item in items %}
|
||||
{{ item.name }}: {{ item.value }}
|
||||
{% endfor %}
|
||||
"""
|
||||
valid, errors = validate_jinja2_syntax(template)
|
||||
assert valid is True
|
||||
assert errors == []
|
||||
|
||||
def test_unclosed_block(self):
|
||||
"""Detects unclosed blocks."""
|
||||
valid, errors = validate_jinja2_syntax("{% for x in y %}")
|
||||
assert valid is False
|
||||
assert len(errors) > 0
|
||||
|
||||
|
||||
class MockWebsocketHandler:
|
||||
"""Mock websocket handler for testing."""
|
||||
|
||||
def __init__(self, scene=None):
|
||||
self._scene = scene
|
||||
self.messages = []
|
||||
|
||||
@property
|
||||
def scene(self):
|
||||
return self._scene
|
||||
|
||||
def queue_put(self, data):
|
||||
self.messages.append(data)
|
||||
|
||||
|
||||
class TestPromptsPluginListGroups:
|
||||
"""Tests for handle_list_groups handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_lists_groups_without_scene(self, plugin, tmp_path):
|
||||
"""Lists groups when no scene is loaded."""
|
||||
mock_groups = [
|
||||
Mock(
|
||||
name="user",
|
||||
is_active=True,
|
||||
is_readonly=False,
|
||||
template_count=5,
|
||||
path=str(tmp_path / "user"),
|
||||
model_dump=lambda: {
|
||||
"name": "user",
|
||||
"is_active": True,
|
||||
"is_readonly": False,
|
||||
"template_count": 5,
|
||||
"path": str(tmp_path / "user"),
|
||||
},
|
||||
),
|
||||
Mock(
|
||||
name="default",
|
||||
is_active=True,
|
||||
is_readonly=True,
|
||||
template_count=100,
|
||||
path=str(tmp_path / "default"),
|
||||
model_dump=lambda: {
|
||||
"name": "default",
|
||||
"is_active": True,
|
||||
"is_readonly": True,
|
||||
"template_count": 100,
|
||||
"path": str(tmp_path / "default"),
|
||||
},
|
||||
),
|
||||
]
|
||||
|
||||
with patch("talemate.server.prompts.list_groups", return_value=mock_groups):
|
||||
await plugin.handle_list_groups({})
|
||||
|
||||
assert len(plugin.websocket_handler.messages) == 1
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["type"] == "prompts"
|
||||
assert response["action"] == "list_groups"
|
||||
assert response["data"]["scene_loaded"] is False
|
||||
assert len(response["data"]["groups"]) == 2
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_lists_groups_with_scene(self, tmp_path):
|
||||
"""Lists groups when scene is loaded."""
|
||||
mock_scene = Mock()
|
||||
mock_scene.name = "test-scene"
|
||||
mock_scene.template_dir = str(tmp_path)
|
||||
|
||||
handler = MockWebsocketHandler(scene=mock_scene)
|
||||
plugin = PromptsPlugin(handler)
|
||||
|
||||
# Create mock groups - need to use MagicMock and configure_mock for 'name'
|
||||
# because Mock uses 'name' internally
|
||||
scene_group_mock = MagicMock()
|
||||
scene_group_mock.configure_mock(name="scene")
|
||||
scene_group_mock.is_active = True
|
||||
scene_group_mock.is_readonly = False
|
||||
scene_group_mock.template_count = 2
|
||||
scene_group_mock.path = str(tmp_path)
|
||||
scene_group_mock.model_dump = lambda: {
|
||||
"name": "scene",
|
||||
"is_active": True,
|
||||
"is_readonly": False,
|
||||
"template_count": 2,
|
||||
"path": str(tmp_path),
|
||||
}
|
||||
|
||||
user_group_mock = MagicMock()
|
||||
user_group_mock.configure_mock(name="user")
|
||||
user_group_mock.is_active = True
|
||||
user_group_mock.is_readonly = False
|
||||
user_group_mock.template_count = 5
|
||||
user_group_mock.path = str(tmp_path / "user")
|
||||
user_group_mock.model_dump = lambda: {
|
||||
"name": "user",
|
||||
"is_active": True,
|
||||
"is_readonly": False,
|
||||
"template_count": 5,
|
||||
"path": str(tmp_path / "user"),
|
||||
}
|
||||
|
||||
mock_groups = [scene_group_mock, user_group_mock]
|
||||
|
||||
with patch("talemate.server.prompts.list_groups", return_value=mock_groups):
|
||||
await plugin.handle_list_groups({})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["scene_loaded"] is True
|
||||
# Scene group should have is_scene flag
|
||||
scene_group = next(
|
||||
g for g in response["data"]["groups"] if g["name"] == "scene"
|
||||
)
|
||||
assert scene_group["is_scene"] is True
|
||||
|
||||
|
||||
class TestPromptsPluginCreateGroup:
|
||||
"""Tests for handle_create_group handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_creates_group_successfully(self, plugin, tmp_path):
|
||||
"""Creates a new group successfully."""
|
||||
mock_group = Mock(
|
||||
name="my-new-group",
|
||||
is_active=False,
|
||||
is_readonly=False,
|
||||
template_count=0,
|
||||
path=str(tmp_path),
|
||||
model_dump=lambda: {
|
||||
"name": "my-new-group",
|
||||
"is_active": False,
|
||||
"is_readonly": False,
|
||||
"template_count": 0,
|
||||
"path": str(tmp_path),
|
||||
},
|
||||
)
|
||||
|
||||
with patch("talemate.server.prompts.create_group", return_value=mock_group):
|
||||
await plugin.handle_create_group({"name": "my-new-group"})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
assert response["data"]["group"]["name"] == "my-new-group"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handles_reserved_name_error(self, plugin):
|
||||
"""Handles error when trying to create reserved group name."""
|
||||
with patch(
|
||||
"talemate.server.prompts.create_group",
|
||||
side_effect=ValueError("Cannot create reserved"),
|
||||
):
|
||||
await plugin.handle_create_group({"name": "default"})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert "error" in response["data"]
|
||||
|
||||
|
||||
class TestPromptsPluginDeleteGroup:
|
||||
"""Tests for handle_delete_group handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_deletes_group_successfully(self, plugin):
|
||||
"""Deletes a group successfully."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.group_priority = ["user"]
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.delete_group", return_value=True):
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
await plugin.handle_delete_group({"name": "my-group", "force": True})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_removes_from_priority_on_delete(self, plugin):
|
||||
"""Removes deleted group from priority list."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.group_priority = ["user", "my-group"]
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.delete_group", return_value=True):
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
await plugin.handle_delete_group({"name": "my-group"})
|
||||
|
||||
assert "my-group" not in mock_config.prompts.group_priority
|
||||
mock_config.set_dirty.assert_called_once()
|
||||
|
||||
|
||||
class TestPromptsPluginSetGroupPriority:
|
||||
"""Tests for handle_set_group_priority handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_sets_priority_order(self, plugin):
|
||||
"""Sets group priority order."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.group_priority = []
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
await plugin.handle_set_group_priority({"priority": ["user", "my-group"]})
|
||||
|
||||
assert mock_config.prompts.group_priority == ["user", "my-group"]
|
||||
mock_config.set_dirty.assert_called_once()
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
|
||||
|
||||
class TestPromptsPluginListTemplates:
|
||||
"""Tests for handle_list_templates handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_lists_all_templates(self, plugin):
|
||||
"""Lists all templates with source info."""
|
||||
mock_templates = [
|
||||
Mock(
|
||||
uid="narrator.narrate-scene",
|
||||
agent="narrator",
|
||||
name="narrate-scene",
|
||||
source_group="default",
|
||||
available_in=["default", "user"],
|
||||
model_dump=lambda: {
|
||||
"uid": "narrator.narrate-scene",
|
||||
"agent": "narrator",
|
||||
"name": "narrate-scene",
|
||||
"source_group": "default",
|
||||
"available_in": ["default", "user"],
|
||||
},
|
||||
),
|
||||
]
|
||||
|
||||
with patch(
|
||||
"talemate.server.prompts.list_templates", return_value=mock_templates
|
||||
):
|
||||
await plugin.handle_list_templates({})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["type"] == "prompts"
|
||||
assert response["action"] == "list_templates"
|
||||
assert len(response["data"]["templates"]) == 1
|
||||
assert response["data"]["templates"][0]["uid"] == "narrator.narrate-scene"
|
||||
|
||||
|
||||
class TestPromptsPluginListGroupTemplates:
|
||||
"""Tests for handle_list_group_templates handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_lists_templates_for_group(self, plugin):
|
||||
"""Lists templates showing which exist in specific group."""
|
||||
mock_templates = [
|
||||
Mock(
|
||||
uid="narrator.narrate-scene",
|
||||
available_in=["default", "user"],
|
||||
model_dump=lambda: {
|
||||
"uid": "narrator.narrate-scene",
|
||||
"available_in": ["default", "user"],
|
||||
},
|
||||
),
|
||||
Mock(
|
||||
uid="narrator.other-template",
|
||||
available_in=["default"],
|
||||
model_dump=lambda: {
|
||||
"uid": "narrator.other-template",
|
||||
"available_in": ["default"],
|
||||
},
|
||||
),
|
||||
]
|
||||
|
||||
with patch(
|
||||
"talemate.server.prompts.list_templates", return_value=mock_templates
|
||||
):
|
||||
await plugin.handle_list_group_templates({"group": "user"})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
templates = response["data"]["templates"]
|
||||
assert len(templates) == 2
|
||||
|
||||
narrate_scene = next(
|
||||
t for t in templates if t["uid"] == "narrator.narrate-scene"
|
||||
)
|
||||
other_template = next(
|
||||
t for t in templates if t["uid"] == "narrator.other-template"
|
||||
)
|
||||
|
||||
assert narrate_scene["exists"] is True # exists in user
|
||||
assert other_template["exists"] is False # doesn't exist in user
|
||||
|
||||
|
||||
class TestPromptsPluginGetTemplate:
|
||||
"""Tests for handle_get_template handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_gets_template_from_specific_group(self, plugin):
|
||||
"""Gets template content from specified group."""
|
||||
with patch(
|
||||
"talemate.server.prompts.get_template_content",
|
||||
return_value="template content",
|
||||
):
|
||||
await plugin.handle_get_template({"uid": "narrator.test", "group": "user"})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["content"] == "template content"
|
||||
assert response["data"]["group"] == "user"
|
||||
assert response["data"]["readonly"] is False
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_default_group_is_readonly(self, plugin):
|
||||
"""Default group templates are marked readonly."""
|
||||
with patch(
|
||||
"talemate.server.prompts.get_template_content",
|
||||
return_value="template content",
|
||||
):
|
||||
await plugin.handle_get_template(
|
||||
{"uid": "narrator.test", "group": "default"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["readonly"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_resolves_template_when_no_group(self, plugin, tmp_path):
|
||||
"""Resolves template when no group specified."""
|
||||
mock_path = tmp_path / "test.jinja2"
|
||||
mock_path.write_text("resolved content")
|
||||
|
||||
with patch(
|
||||
"talemate.server.prompts.resolve_template",
|
||||
return_value=(mock_path, "default"),
|
||||
):
|
||||
await plugin.handle_get_template({"uid": "narrator.test"})
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["content"] == "resolved content"
|
||||
assert response["data"]["readonly"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handles_not_found(self, plugin):
|
||||
"""Handles template not found."""
|
||||
with patch("talemate.server.prompts.get_template_content", return_value=None):
|
||||
await plugin.handle_get_template(
|
||||
{"uid": "narrator.nonexistent", "group": "user"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert "error" in response["data"]
|
||||
|
||||
|
||||
class TestPromptsPluginSaveTemplate:
|
||||
"""Tests for handle_save_template handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_saves_template_successfully(self, plugin):
|
||||
"""Saves template content successfully."""
|
||||
with patch("talemate.server.prompts.write_template") as mock_write:
|
||||
await plugin.handle_save_template(
|
||||
{
|
||||
"uid": "narrator.test",
|
||||
"group": "user",
|
||||
"content": "new content {{ var }}",
|
||||
}
|
||||
)
|
||||
|
||||
mock_write.assert_called_once()
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
assert response["data"]["syntax_valid"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_reports_syntax_errors(self, plugin):
|
||||
"""Reports syntax errors but still saves."""
|
||||
with patch("talemate.server.prompts.write_template") as mock_write:
|
||||
await plugin.handle_save_template(
|
||||
{"uid": "narrator.test", "group": "user", "content": "invalid {{ var"}
|
||||
)
|
||||
|
||||
mock_write.assert_called_once() # Still saved
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
assert response["data"]["syntax_valid"] is False
|
||||
assert len(response["data"]["syntax_errors"]) > 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handles_write_error(self, plugin):
|
||||
"""Handles error when writing template."""
|
||||
with patch(
|
||||
"talemate.server.prompts.write_template",
|
||||
side_effect=ValueError("Cannot write to default"),
|
||||
):
|
||||
await plugin.handle_save_template(
|
||||
{"uid": "narrator.test", "group": "default", "content": "content"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert "error" in response["data"]
|
||||
|
||||
|
||||
class TestPromptsPluginDeleteTemplate:
|
||||
"""Tests for handle_delete_template handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_deletes_template_successfully(self, plugin):
|
||||
"""Deletes template override successfully."""
|
||||
with patch("talemate.server.prompts.delete_template", return_value=True):
|
||||
await plugin.handle_delete_template(
|
||||
{"uid": "narrator.test", "group": "user"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handles_not_found(self, plugin):
|
||||
"""Handles template not found."""
|
||||
with patch("talemate.server.prompts.delete_template", return_value=False):
|
||||
await plugin.handle_delete_template(
|
||||
{"uid": "narrator.nonexistent", "group": "user"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert response["data"]["error"] == "Template not found"
|
||||
|
||||
|
||||
class TestPromptsPluginCreateTemplate:
|
||||
"""Tests for handle_create_template handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_creates_new_template(self, plugin):
|
||||
"""Creates a new template file."""
|
||||
with patch("talemate.server.prompts.get_template_content", return_value=None):
|
||||
with patch("talemate.server.prompts.write_template") as mock_write:
|
||||
await plugin.handle_create_template(
|
||||
{
|
||||
"uid": "narrator.my-helper",
|
||||
"group": "user",
|
||||
"content": "helper content",
|
||||
}
|
||||
)
|
||||
|
||||
mock_write.assert_called_once()
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cannot_create_in_default(self, plugin):
|
||||
"""Cannot create template in default group."""
|
||||
await plugin.handle_create_template(
|
||||
{"uid": "narrator.test", "group": "default", "content": "content"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert "default" in response["data"]["error"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_cannot_create_existing(self, plugin):
|
||||
"""Cannot create template that already exists."""
|
||||
with patch(
|
||||
"talemate.server.prompts.get_template_content",
|
||||
return_value="existing content",
|
||||
):
|
||||
await plugin.handle_create_template(
|
||||
{"uid": "narrator.existing", "group": "user", "content": "new content"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert "already exists" in response["data"]["error"]
|
||||
|
||||
|
||||
class TestPromptsPluginSetTemplateSource:
|
||||
"""Tests for handle_set_template_source handler."""
|
||||
|
||||
@pytest.fixture
|
||||
def plugin(self):
|
||||
handler = MockWebsocketHandler()
|
||||
return PromptsPlugin(handler)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_sets_explicit_source(self, plugin):
|
||||
"""Sets explicit source for template."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.template_sources = {}
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
with patch(
|
||||
"talemate.server.prompts.get_template_content", return_value="content"
|
||||
):
|
||||
await plugin.handle_set_template_source(
|
||||
{"uid": "narrator.test", "group": "my-custom"}
|
||||
)
|
||||
|
||||
assert mock_config.prompts.template_sources["narrator.test"] == "my-custom"
|
||||
mock_config.set_dirty.assert_called_once()
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_removes_override_with_null_group(self, plugin):
|
||||
"""Removes override when group is null."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.template_sources = {"narrator.test": "my-custom"}
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
await plugin.handle_set_template_source(
|
||||
{"uid": "narrator.test", "group": None}
|
||||
)
|
||||
|
||||
assert "narrator.test" not in mock_config.prompts.template_sources
|
||||
mock_config.set_dirty.assert_called_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_validates_template_exists_in_group(self, plugin):
|
||||
"""Validates template exists in target group before setting source."""
|
||||
mock_config = Mock()
|
||||
mock_config.prompts.template_sources = {}
|
||||
mock_config.set_dirty = AsyncMock()
|
||||
|
||||
with patch("talemate.server.prompts.get_config", return_value=mock_config):
|
||||
with patch(
|
||||
"talemate.server.prompts.get_template_content", return_value=None
|
||||
):
|
||||
await plugin.handle_set_template_source(
|
||||
{"uid": "narrator.test", "group": "nonexistent-group"}
|
||||
)
|
||||
|
||||
response = plugin.websocket_handler.messages[0]
|
||||
assert response["data"]["success"] is False
|
||||
assert "not found" in response["data"]["error"]
|
||||
Reference in New Issue
Block a user