mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
cc4bf19706
Judge a vault KID only on a file whose KID map holds it, tell remote vaults of a bad pair only once another key replaces it, and let the server CDM unflag a pair it returns; the server never flags a CDM-licensed pair. Log FFmpeg stderr on a failed check and name the missing CDM when none is set.
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""A failed vault push is reported as a failure, and a server does not store the same pair twice."""
|
|
|
|
from typing import Any
|
|
from uuid import UUID
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.api import handlers
|
|
from unshackle.core.vaults import Vaults
|
|
from unshackle.vaults.HTTP import HTTP
|
|
|
|
KID = UUID("11111111-2222-3333-4444-555555555555")
|
|
KEY = "a" * 32
|
|
|
|
|
|
def test_an_unreachable_json_vault_fails_the_push(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
vault = HTTP("json", "https://vault.invalid/rpc", "token", api_mode="json")
|
|
|
|
def down(method: str, params: dict) -> dict:
|
|
raise ConnectionError("vault is down")
|
|
|
|
monkeypatch.setattr(vault, "request", down)
|
|
vaults = Vaults("SVC")
|
|
vaults.vaults = [vault]
|
|
assert vaults.add_keys({KID: KEY}) == 0
|
|
|
|
|
|
class FakeVault:
|
|
no_push = False
|
|
|
|
def __init__(self, ok: bool) -> None:
|
|
self.ok = ok
|
|
|
|
|
|
def fake_server_vaults(pushes: list, vault_list: list[FakeVault]) -> Any:
|
|
class FakeVaults:
|
|
vaults = vault_list
|
|
|
|
def __len__(self) -> int:
|
|
return len(vault_list)
|
|
|
|
def clear_licensed_flags(self, content_keys: dict) -> list:
|
|
return []
|
|
|
|
def add_keys(self, key_map: dict) -> int:
|
|
pushes.append(key_map)
|
|
return sum(vault.ok for vault in vault_list)
|
|
|
|
return FakeVaults()
|
|
|
|
|
|
def test_server_does_not_push_a_pair_it_already_stored(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
pushes: list = []
|
|
monkeypatch.setattr(handlers, "load_server_vaults", lambda name: fake_server_vaults(pushes, [FakeVault(True)]))
|
|
monkeypatch.setattr(handlers, "CACHED_PAIRS", set())
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC")
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC")
|
|
assert pushes == [{KID: KEY}]
|
|
|
|
|
|
def test_server_pushes_again_after_one_vault_failed(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
pushes: list = []
|
|
vault_list = [FakeVault(True), FakeVault(False)]
|
|
monkeypatch.setattr(handlers, "load_server_vaults", lambda name: fake_server_vaults(pushes, vault_list))
|
|
monkeypatch.setattr(handlers, "CACHED_PAIRS", set())
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC")
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC")
|
|
assert len(pushes) == 2
|