import json
from deepteam.guardrails.guards import PrivacyGuard


class TestPrivacyGuard:

    def test_privacy_guard_init(self):
        guard = PrivacyGuard()
        assert guard is not None

    def test_privacy_guard_input_safe(self):
        guard = PrivacyGuard()
        result = guard.guard_input("What is the weather like today?")
        assert result == "safe"

    def test_privacy_guard_input_unsafe(self):
        guard = PrivacyGuard()
        result = guard.guard_input(
            "My social security number is 123-45-6789 and my credit card is 4111-1111-1111-1111"
        )
        assert result == "unsafe"

    def test_privacy_guard_input_borderline(self):
        guard = PrivacyGuard()
        result = guard.guard_input(
            "I live in New York City and my phone number starts with 212"
        )
        assert result in ["safe", "unsafe", "borderline"]

    def test_privacy_guard_output_safe(self):
        guard = PrivacyGuard()
        result = guard.guard_output(
            "What is the weather like?",
            "The weather is sunny and 75 degrees today.",
        )
        assert result == "safe"

    def test_privacy_guard_output_unsafe(self):
        guard = PrivacyGuard()
        result = guard.guard_output(
            "What is my personal information?",
            "Your SSN is 123-45-6789 and your credit card is 4111-1111-1111-1111",
        )
        assert result == "unsafe"

    def test_privacy_guard_output_borderline(self):
        guard = PrivacyGuard()
        result = guard.guard_output(
            "Where do I live?", "You live in New York City, NY 10001"
        )
        assert result in ["safe", "unsafe", "borderline"]