mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-11 19:45:06 +00:00
74 lines
2.8 KiB
Python
74 lines
2.8 KiB
Python
"""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 = {"DEMO": "a", "Default": "b"}
|
|
assert ci_get(m, "DEMO") == "a"
|
|
assert ci_get(m, "demo") == "a"
|
|
assert ci_get(m, "default") == "b"
|
|
assert ci_get(m, "missing") is None
|
|
assert ci_get(m, "missing", "fallback") == "fallback"
|
|
assert ci_get({}, "DEMO", "fallback") == "fallback"
|
|
|
|
|
|
def test_decryption_map_keys_upper_cased_on_build():
|
|
c = Config(decryption={"example": "mp4decrypt", "default": "shaka"})
|
|
assert c.decryption_map == {"EXAMPLE": "mp4decrypt", "DEFAULT": "shaka"}
|
|
assert c.decryption == "shaka"
|
|
|
|
|
|
def test_resolve_decryption_per_service_and_default():
|
|
c = Config(decryption={"EXAMPLE": "mp4decrypt", "default": "shaka"})
|
|
assert resolve_decryption(c.decryption_map, c.decryption, "EXAMPLE") == "mp4decrypt"
|
|
# service not mapped -> default
|
|
assert resolve_decryption(c.decryption_map, c.decryption, "DEMO") == "shaka"
|
|
# case-insensitive both ways (lowercase yaml key, mixed-case lookup)
|
|
c2 = Config(decryption={"example": "mp4decrypt", "default": "shaka"})
|
|
assert resolve_decryption(c2.decryption_map, c2.decryption, "Example") == "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={"demo": "dev_a", "default": "dev_b"})
|
|
assert c.cdm == {"demo": "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={"demo": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c.cdm, "DEMO") == "dev_a"
|
|
# unmapped service -> default
|
|
assert resolve_cdm_name(c.cdm, "EXAMPLE") == "dev_b"
|
|
# uppercase yaml key, lowercase lookup
|
|
c2 = Config(cdm={"DEMO": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c2.cdm, "demo") == "dev_a"
|
|
|
|
|
|
def test_resolve_cdm_name_override_wins():
|
|
c = Config(cdm={"DEMO": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c.cdm, "DEMO", override="dev_override") == "dev_override"
|
|
|
|
|
|
def test_resolve_cdm_name_no_match_no_default():
|
|
c = Config(cdm={"DEMO": "dev_a"})
|
|
assert resolve_cdm_name(c.cdm, "EXAMPLE") is None
|