oclaw/tests/test_prompt_loader.py
oliver dbbe3add6a 重构主控编排与运行时预热链路,统一工作区提示词/专家调度协议并补齐 wiki 记忆注入与写回闭环。
同时收敛启动与运维脚本默认行为(含 wiki worker)、更新 Admin 可观测性与相关测试,降低首轮时延并提高运行稳定性。

Made-with: Cursor
2026-04-26 08:34:33 +08:00

43 lines
1.4 KiB
Python

from __future__ import annotations
import pytest
from oclaw.prompts.loader import load_runtime_prompt_doc, load_prompt_doc, render_runtime_prompt, render_prompt
def test_load_prompt_doc_with_frontmatter() -> None:
doc = load_prompt_doc("runtime/default_system.zh.md")
assert doc.frontmatter.get("title") == "default_runtime_system_zh"
assert "你是一个通用 AI 助手" in doc.body
def test_render_prompt_strict_missing_variable_raises() -> None:
with pytest.raises(ValueError):
render_prompt("tools/tool_result_unpaired.md", variables={"tag": "x"}, strict=True)
def test_render_prompt_success() -> None:
out = render_prompt(
"tools/tool_result_unpaired.md",
variables={"tag": "tool_use_result:unpaired", "payload": "ok=true"},
strict=True,
)
assert "[tool_use_result:unpaired]" in out
assert "ok=true" in out
def test_load_runtime_prompt_doc_with_frontmatter() -> None:
doc = load_runtime_prompt_doc("runtime/default_system.zh.md")
assert doc.frontmatter.get("title") == "default_runtime_system_zh"
assert "你是一个通用 AI 助手" in doc.body
def test_render_runtime_prompt_success() -> None:
out = render_runtime_prompt(
"runtime/project_context_block.md",
variables={"project_context": "[AGENTS.md]\nfoo"},
strict=True,
)
assert "[project_context]" in out
assert "[AGENTS.md]" in out