C-001 임상 검토 게이트 보강

apps/api/app/test_guardrail_ideation_cap.py

@@ -0,0 +1,43 @@
+"""자살사고 단계 절대 상한의 출력 fail-closed 회귀 테스트."""
+
+from __future__ import annotations
+
+import unittest
+
+from .services import guardrail, state_machine
+
+
+class GuardrailIdeationCapTest(unittest.TestCase):
+    def test_over_cap_is_blocked_and_requires_regeneration(self) -> None:
+        result = guardrail.sanitize_client_reply(
+            "요즘 버티기 어렵다는 생각이 자주 들어요.",
+            ideation_stage=5,
+            turn_seq=4,
+        )
+
+        self.assertTrue(result.blocked)
+        self.assertTrue(result.needs_regeneration)
+        self.assertIn("ideation_over_cap:5>3", result.reasons)
+
+    def test_at_cap_without_other_violation_is_unchanged(self) -> None:
+        text = "요즘 버티기 어렵다는 생각이 자주 들어요."
+
+        result = guardrail.sanitize_client_reply(
+            text,
+            ideation_stage=state_machine.IDEATION_STAGE_CAP,
+            turn_seq=4,
+        )
+
+        self.assertFalse(result.blocked)
+        self.assertFalse(result.needs_regeneration)
+        self.assertEqual(result.text, text)
+        self.assertEqual(result.reasons, [])
+
+    def test_guardrail_reuses_state_machine_cap_and_clamp(self) -> None:
+        self.assertEqual(guardrail.IDEATION_STAGE_CAP, state_machine.IDEATION_STAGE_CAP)
+        self.assertEqual(guardrail.clamp_ideation(5), state_machine.IDEATION_STAGE_CAP)
+        self.assertEqual(guardrail.clamp_ideation(0), 1)
+
+
+if __name__ == "__main__":
+    unittest.main()