from desklock_gateway.tatlock import strip_reasoning


def test_strip_reasoning_removes_think_block() -> None:
    raw = "<think>\nDELEGATE: none\nREASON: simple\n</think>\nGood evening. I am Tatlock."
    assert strip_reasoning(raw) == "Good evening. I am Tatlock."


def test_strip_reasoning_passthrough_without_think() -> None:
    assert strip_reasoning("Just an answer.") == "Just an answer."