mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
bb081046e2
Cache proxy provider catalogues (10 min), reuse vaults and CDMs across one batch licence call, and cache the health update check, /api/services entries and parsed service config.yaml.
86 lines
2.8 KiB
Python
86 lines
2.8 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
|
|
|
|
|
|
def test_one_licence_request_builds_the_server_vaults_once(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
builds: list = []
|
|
pushes: list = []
|
|
|
|
def build(name: str) -> Any:
|
|
builds.append(name)
|
|
return fake_server_vaults(pushes, [FakeVault(False)])
|
|
|
|
monkeypatch.setattr(handlers, "load_server_vaults", build)
|
|
monkeypatch.setattr(handlers, "CACHED_PAIRS", set())
|
|
shared: dict = {}
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC", shared=shared)
|
|
handlers.cache_to_vaults({KID.hex: KEY}, "SVC", shared=shared)
|
|
assert len(builds) == 1
|
|
assert len(pushes) == 2
|