"""Pins per-service decryption-tool and CDM resolution, including key casing.

A per-service `cdm:`/`decryption:` entry must resolve regardless of the key's case vs the
service tag, falling back to the configured default otherwise. Guards the silent
"falls back to default on a case mismatch" foot-gun.
"""

from __future__ import annotations

import pytest

from unshackle.core.config import Config, resolve_cdm_name, resolve_decryption
from unshackle.core.utils.collections import ci_get

pytestmark = pytest.mark.unit


def test_ci_get_exact_and_case_insensitive():
    m = {"NF": "a", "Default": "b"}
    assert ci_get(m, "NF") == "a"
    assert ci_get(m, "nf") == "a"
    assert ci_get(m, "default") == "b"
    assert ci_get(m, "missing") is None
    assert ci_get(m, "missing", "fallback") == "fallback"
    assert ci_get({}, "NF", "fallback") == "fallback"


def test_decryption_map_keys_upper_cased_on_build():
    c = Config(decryption={"cr": "mp4decrypt", "default": "shaka"})
    assert c.decryption_map == {"CR": "mp4decrypt", "DEFAULT": "shaka"}
    assert c.decryption == "shaka"


def test_resolve_decryption_per_service_and_default():
    c = Config(decryption={"CR": "mp4decrypt", "default": "shaka"})
    assert resolve_decryption(c.decryption_map, c.decryption, "CR") == "mp4decrypt"
    # service not mapped -> default
    assert resolve_decryption(c.decryption_map, c.decryption, "NF") == "shaka"
    # case-insensitive both ways (lowercase yaml key, mixed-case lookup)
    c2 = Config(decryption={"cr": "mp4decrypt", "default": "shaka"})
    assert resolve_decryption(c2.decryption_map, c2.decryption, "Cr") == "mp4decrypt"


def test_decryption_scalar_form():
    c = Config(decryption="mp4decrypt")
    assert c.decryption_map == {}
    assert resolve_decryption(c.decryption_map, c.decryption, "ANY") == "mp4decrypt"


def test_cdm_keys_preserved_on_build():
    c = Config(cdm={"nf": "dev_a", "default": "dev_b"})
    assert c.cdm == {"nf": "dev_a", "default": "dev_b"}


def test_resolve_cdm_name_case_insensitive():
    # lowercase yaml key must resolve for an uppercase service tag (the bug being fixed)
    c = Config(cdm={"nf": "dev_a", "default": "dev_b"})
    assert resolve_cdm_name(c.cdm, "NF") == "dev_a"
    # unmapped service -> default
    assert resolve_cdm_name(c.cdm, "ATV") == "dev_b"
    # uppercase yaml key, lowercase lookup
    c2 = Config(cdm={"NF": "dev_a", "default": "dev_b"})
    assert resolve_cdm_name(c2.cdm, "nf") == "dev_a"


def test_resolve_cdm_name_override_wins():
    c = Config(cdm={"NF": "dev_a", "default": "dev_b"})
    assert resolve_cdm_name(c.cdm, "NF", override="dev_override") == "dev_override"


def test_resolve_cdm_name_no_match_no_default():
    c = Config(cdm={"NF": "dev_a"})
    assert resolve_cdm_name(c.cdm, "ATV") is None