初始化提交

backend/tests/application/chat/test_daoti_injection.py

@@ -0,0 +1,59 @@
+"""道体注入与完整性校验测试。"""
+
+from __future__ import annotations
+
+import pytest
+
+from app.application.chat.send_message import _contract_system_messages
+from app.modules.runtime.contract import RuntimeContract
+from app.modules.runtime.integrity import validate_daoti_integrity
+
+
+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_system_prompt_injected() -> None:
+    msgs = _contract_system_messages(_contract(system_prompt="特殊系统指令"))
+    assert "特殊系统指令" in msgs
+
+
+def test_mission_injected() -> None:
+    msgs = _contract_system_messages(_contract(mission="守护用户"))
+    assert any("守护用户" in m for m in msgs)
+
+
+def test_safety_guidelines_precede_safety_constraints() -> None:
+    c = _contract(safety_guidelines="道体边界", safety_constraints=["司命约束"])
+    msgs = _contract_system_messages(c)
+    assert msgs.index("行为边界：\n道体边界") < msgs.index("安全约束：\n- 司命约束")
+
+
+def test_integrity_pass_when_consistent() -> None:
+    c = _contract(mission="M", safety_guidelines="S")
+    validate_daoti_integrity(c, mission="M", safety_guidelines="S")  # 不抛即通过
+
+
+def test_integrity_raises_on_mission_tamper() -> None:
+    c = _contract(mission="M")
+    with pytest.raises(ValueError):
+        validate_daoti_integrity(c, mission="被篡改")
+
+
+def test_integrity_raises_on_safety_tamper() -> None:
+    c = _contract(safety_guidelines="S")
+    with pytest.raises(ValueError):
+        validate_daoti_integrity(c, safety_guidelines="被篡改")
+
+
+def test_integrity_skips_none_fields() -> None:
+    # 道体字段未设（None）时跳过校验
+    validate_daoti_integrity(_contract(), mission=None, safety_guidelines=None)