from agentic_security.probe_actor.refusal import DefaultRefusalClassifier


class TestCheckRefusal:
    # The function correctly identifies a refusal phrase in the response.
    def test_identify_refusal_phrase(self):
        response = "I'm sorry, but I cannot provide that information."
        assert DefaultRefusalClassifier().is_refusal(response)

    # The response is an empty string.
    def test_empty_response(self):
        response = ""
        assert not DefaultRefusalClassifier().is_refusal(response)