59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
|
"""Tests for prompt registry."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from app.prompts.registry import get_prompt
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_prompt_returns_english_for_known_stage():
|
||
|
|
p = get_prompt("orchestrator_phase1", "en")
|
||
|
|
assert isinstance(p, str)
|
||
|
|
assert len(p) > 100
|
||
|
|
assert "Game Master" in p or "GM" in p
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_prompt_falls_back_to_english():
|
||
|
|
p = get_prompt("orchestrator_phase2", "ru")
|
||
|
|
# Russian bundle is empty — falls back to English
|
||
|
|
assert isinstance(p, str)
|
||
|
|
assert len(p) > 100
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_prompt_unknown_stage_raises():
|
||
|
|
with pytest.raises(KeyError):
|
||
|
|
get_prompt("does_not_exist", "en")
|
||
|
|
|
||
|
|
|
||
|
|
def test_all_stages_have_english_prompt():
|
||
|
|
stages = [
|
||
|
|
"world_builder_schema", "world_builder_env", "world_builder_entities",
|
||
|
|
"world_editor", "orchestrator_phase1", "orchestrator_phase2",
|
||
|
|
"orchestrator_phase3_summary", "orchestrator_phase3_suggest",
|
||
|
|
"intro_scene", "subagent", "summary",
|
||
|
|
]
|
||
|
|
for s in stages:
|
||
|
|
p = get_prompt(s, "en")
|
||
|
|
assert len(p) > 50, f"Stage {s} has empty prompt"
|
||
|
|
|
||
|
|
|
||
|
|
def test_orchestrator_phase1_prompt_has_required_placeholders():
|
||
|
|
p = get_prompt("orchestrator_phase1", "en")
|
||
|
|
for placeholder in [
|
||
|
|
"{world_name}", "{rules}", "{schemas_summary}", "{environment_json}",
|
||
|
|
"{plot_rails_json}", "{current_time}", "{max_substeps}",
|
||
|
|
"{player_action}", "{language}", "{recent_history}",
|
||
|
|
]:
|
||
|
|
assert placeholder in p, f"Missing placeholder {placeholder}"
|
||
|
|
|
||
|
|
|
||
|
|
def test_orchestrator_phase2_prompt_has_required_placeholders():
|
||
|
|
p = get_prompt("orchestrator_phase2", "en")
|
||
|
|
for placeholder in [
|
||
|
|
"{world_name}", "{language}", "{current_time}",
|
||
|
|
"{player_action}", "{plan}", "{summary_json}",
|
||
|
|
"{environment_json}", "{world_description}",
|
||
|
|
]:
|
||
|
|
assert placeholder in p, f"Missing placeholder {placeholder}"
|