"""Unit tests for the rule_confidence lookup."""

from src.logic import rule_confidence as rc


def test_lookup_only_contains_valid_levels() -> None:
    """Every entry maps to HIGH, MEDIUM, or LOW."""
    valid = set(rc.CONFIDENCE_WEIGHTS.keys())
    assert set(rc.get_lookup().values()) <= valid


def test_get_confidence_returns_level_for_known_rule() -> None:
    """Lookup returns the configured level for a known rule."""
    assert rc.get_confidence("1.5") == "HIGH"
    assert rc.get_confidence("1.2") == "MEDIUM"
    assert rc.get_confidence("1.1") == "LOW"


def test_get_confidence_returns_none_for_unknown_rule() -> None:
    """Unknown rule IDs return None rather than raising."""
    assert rc.get_confidence("999.99") is None


def test_get_expected_rule_ids_matches_lookup_keys() -> None:
    """Expected IDs are exactly the lookup keys."""
    assert rc.get_expected_rule_ids() == set(rc.get_lookup().keys())