"""자살사고 단계 절대 상한의 출력 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()