"""Tests for the data-training-tier selection guard."""

from hermes_cli.model_data_policy_guard import (
    DataTrainingWarning,
    data_training_warning,
)


def test_fires_on_meta_contributor():
    w = data_training_warning("muse-spark-1.2-contributor", provider="meta-ai")
    assert isinstance(w, DataTrainingWarning)
    assert w.model == "muse-spark-1.2-contributor"
    assert w.message


def test_silent_on_non_contributor_muse():
    assert data_training_warning("muse-spark-1.2", provider="meta-ai") is None
    assert data_training_warning("muse-spark-1.1", provider="meta-ai") is None


def test_silent_on_unrelated_models():
    for m in ("anthropic/claude-opus-4.8", "gpt-5.6-sol", "deepseek-v4-pro", ""):
        assert data_training_warning(m, provider="anthropic") is None


def test_fires_regardless_of_provider_string():
    # id-keyed: must fire even if selected via custom/gateway (no meta-ai provider)
    assert data_training_warning("muse-spark-1.2-contributor", provider="custom") is not None
    assert data_training_warning("muse-spark-1.2-contributor") is not None


def test_case_insensitive():
    assert data_training_warning("MUSE-SPARK-1.2-CONTRIBUTOR", provider="meta-ai") is not None