kenzuya/unshackle
1
0
Fork 2
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-11-12 20:15:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(vaults): fix the vault-write fakes is_flagged and log.error

Browse Source
This commit is contained in:
imSp4rky
2026-09-19 22:42:26 -06:00
parent cd6a38aec8
commit 025499efa7
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/remote/unit/test_server_cdm_vault_write.py
+5 -2
View File
@@ -36,6 +36,9 @@ class FakeVaults:
def get_key(self, kid):
return self.held.get(kid), self.vault_used
def is_flagged(self, kid, key):
return False
def add_keys(self, kid_keys):
self.added.update(kid_keys)
self.pushes.append(dict(kid_keys))
@@ -54,7 +57,7 @@ class FakeWidevine:
def make_cmd(vaults):
cmd = dl.__new__(dl)
cmd.log = SimpleNamespace(warning=print, info=print, debug=print)
cmd.log = SimpleNamespace(warning=print, info=print, debug=print, error=print)
cmd.service = "TEST"
cmd.vaults = vaults
cmd.vault_cache_tally = None
@@ -107,7 +110,7 @@ def test_batch_resolved_keys_are_cached_before_download():
"""The batch licence fills track.drm before prepare_drm, so dl caches it at the call site."""
vaults = FakeVaults()
cmd = dl.__new__(dl)
cmd.log = SimpleNamespace(warning=print, info=print, debug=print)
cmd.log = SimpleNamespace(warning=print, info=print, debug=print, error=print)
cmd.service = "TEST"
cmd.vaults = vaults
cmd.vault_cache_tally = None