初始化提交

backend/tests/application/chat/test_lorebook_injection.py

@@ -0,0 +1,41 @@
+"""世谱注入位置测试：before/after 相对 personality_summary。"""
+
+from __future__ import annotations
+
+from app.application.chat.send_message import _contract_system_messages
+from app.modules.runtime.contract import RuntimeContract
+
+
+def _contract(**kw: object) -> RuntimeContract:
+    base: dict[str, object] = {
+        "character_id": "c",
+        "session_id": "s",
+        "run_id": "r",
+        "model_hint": "m",
+        "trace_id": "t",
+    }
+    base.update(kw)
+    return RuntimeContract(**base)
+
+
+def test_lorebook_before_precedes_personality() -> None:
+    c = _contract(personality_summary="人格X", lorebook_before="世界设定前")
+    msgs = _contract_system_messages(c)
+    assert msgs.index("世界设定前") < msgs.index("当前人格：\n人格X")
+
+
+def test_lorebook_after_follows_personality() -> None:
+    c = _contract(personality_summary="人格X", lorebook_after="世界设定后")
+    msgs = _contract_system_messages(c)
+    assert msgs.index("当前人格：\n人格X") < msgs.index("世界设定后")
+
+
+def test_no_lorebook_when_none() -> None:
+    msgs = _contract_system_messages(_contract())
+    assert not any("世界设定" in m for m in msgs)
+
+
+def test_lorebook_without_personality_still_injects() -> None:
+    c = _contract(lorebook_before="前", lorebook_after="后")
+    msgs = _contract_system_messages(c)
+    assert "前" in msgs and "后" in msgs