from audita.protection import ProtectedVocabulary
from audita.schemas import parse_glossary_yaml


def _vocabulary():
    glossary = parse_glossary_yaml(
        """
        glossary:
          - name: "Hrank"
            aliases:
              - "Greenfield"
            category: pc
            summary: "Hrank Greenfield is a player character."
          - name: "Popov"
            category: npc
            summary: "Popov is an allied NPC."
        """
    )
    return ProtectedVocabulary.from_glossary(glossary)


def test_protection_blocks_replacing_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Hrank moves.", "Frank moves.") is not None


def test_protection_blocks_replacing_possessive_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Popov's exhausted.", "Pawpaw's exhausted.") is not None


def test_protection_blocks_lowercasing_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Hrank moves.", "hrank moves.") is not None


def test_protection_blocks_noncanonical_uppercase_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Popov moves.", "POPOV moves.") is not None


def test_protection_allows_canonical_capitalization():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("hrank moves.", "Hrank moves.") is None


def test_protection_allows_correction_toward_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Pawpaw moves.", "Popov moves.") is None


def test_protection_allows_possessive_correction_toward_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Pawpaw's exhausted.", "Popov's exhausted.") is None


def test_protection_allows_punctuation_around_protected_term():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Popov, moves.", "Popov. Moves.") is None


def test_protection_does_not_match_terms_inside_larger_words():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("The shrank spell worked.", "The shrank spell works.") is None


def test_protection_applies_to_aliases():
    vocabulary = _vocabulary()

    assert vocabulary.violation_reason("Greenfield waits.", "greenfield waits.") is not None