From 025499efa77873a9e9cb144dca6e0c878ac4629c Mon Sep 17 00:00:00 2001 From: imSp4rky Date: Sat, 19 Sep 2026 22:42:26 -0600 Subject: [PATCH] test(vaults): fix the vault-write fakes is_flagged and log.error --- tests/remote/unit/test_server_cdm_vault_write.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/remote/unit/test_server_cdm_vault_write.py b/tests/remote/unit/test_server_cdm_vault_write.py index 739de725..386df90d 100644 --- a/tests/remote/unit/test_server_cdm_vault_write.py +++ b/tests/remote/unit/test_server_cdm_vault_write.py @@ -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