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: replace real service tags and brands with generic placeholders

Browse Source
This commit is contained in:
imSp4rky
2026-10-18 12:17:08 -06:00
parent 768e35d905
commit 6fdfeb9cae
12 changed files with 48 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/remote/unit/test_dashboard.py
+2 -2
View File
@@ -136,12 +136,12 @@ def test_ring_masks_host_paths_and_secrets_in_records() -> None:
logger = logging.getLogger("test.ring.redact")
logger.setLevel(logging.INFO)
logger.addHandler(ring)
logger.error(f'File "{root}/unshackle/services/HMAX/__init__.py" 404: https://api.x/y?token=abc')
logger.error(f'File "{root}/unshackle/services/EXAMPLE/__init__.py" 404: https://api.x/y?token=abc')
logger.removeHandler(ring)
msg = ring.records[-1]["msg"]
assert root not in msg
assert "<unshackle>/unshackle/services/HMAX" in msg
assert "<unshackle>/unshackle/services/EXAMPLE" in msg
assert "token=abc" not in msg
assert "https://api.x/y" in msg
assert ring.since(0, logger="test") and not ring.since(0, logger="other")
tests/remote/unit/test_remote_relay.py
+21 -21
View File
@@ -11,8 +11,8 @@ from unshackle.core.config import config
SERVICES = [
{
"tag": "AMZN",
"aliases": ["amazon"],
"tag": "EXAMPLE",
"aliases": ["sample"],
"cli_params": [{"kind": "option", "name": "extras"}, {"kind": "argument", "name": "title"}],
}
]
@@ -20,7 +20,7 @@ INFO = SERVICES[0]
class FakeRemote:
service_tag = "AMZN"
service_tag = "EXAMPLE"
def __init__(self) -> None:
self.closed = False
@@ -47,7 +47,7 @@ def lease(data, **extra):
return handlers.lease_remote_service({**data, **extra}, "srv", INFO, None, "show")
DATA = {"service": "AMZN", "title_id": "show", "remote": "srv"}
DATA = {"service": "EXAMPLE", "title_id": "show", "remote": "srv"}
async def test_concurrent_leases_share_one_session(pool):
@@ -89,17 +89,17 @@ def remote_config(monkeypatch):
def test_remote_server_resolves_case_and_aliases(remote_config):
assert handlers.remote_server({"remote": "srv", "service": "amzn"}) == ("srv", INFO)
assert handlers.remote_server({"remote": "srv", "service": "Amazon"}) == ("srv", INFO)
assert handlers.remote_server({"service": "AMZN"}) is None
assert handlers.remote_server({"remote": "srv", "service": "example"}) == ("srv", INFO)
assert handlers.remote_server({"remote": "srv", "service": "Sample"}) == ("srv", INFO)
assert handlers.remote_server({"service": "EXAMPLE"}) is None
@pytest.mark.parametrize(
"data, code",
[
({"remote": "nope", "service": "AMZN"}, APIErrorCode.INVALID_PARAMETERS),
({"remote": ["srv"], "service": "AMZN"}, APIErrorCode.INVALID_PARAMETERS),
({"remote": "srv", "service": "NF"}, APIErrorCode.INVALID_SERVICE),
({"remote": "nope", "service": "EXAMPLE"}, APIErrorCode.INVALID_PARAMETERS),
({"remote": ["srv"], "service": "EXAMPLE"}, APIErrorCode.INVALID_PARAMETERS),
({"remote": "srv", "service": "OTHER"}, APIErrorCode.INVALID_SERVICE),
],
)
def test_remote_server_rejects(remote_config, data, code):
@@ -109,31 +109,31 @@ def test_remote_server_rejects(remote_config, data, code):
def test_remote_server_applies_the_allowlist_and_reports_an_unreachable_server(remote_config, monkeypatch):
monkeypatch.setattr(handlers, "get_allowed_services", lambda request=None: ["NF"])
monkeypatch.setattr(handlers, "get_allowed_services", lambda request=None: ["OTHER"])
with pytest.raises(APIError) as e:
handlers.remote_server({"remote": "srv", "service": "AMZN"})
handlers.remote_server({"remote": "srv", "service": "EXAMPLE"})
assert e.value.error_code == APIErrorCode.INVALID_SERVICE
monkeypatch.setattr(remote_service, "remote_service_list", lambda name: None)
with pytest.raises(APIError) as e:
handlers.remote_server({"remote": "srv", "service": "AMZN"})
handlers.remote_server({"remote": "srv", "service": "EXAMPLE"})
assert e.value.error_code == APIErrorCode.SERVICE_UNAVAILABLE
def test_make_remote_service_passes_only_cli_options(remote_config, monkeypatch):
built = {}
monkeypatch.setattr(remote_service, "resolve_server", lambda name: ("https://srv", "k", {"AMZN": {"x": 1}}))
monkeypatch.setattr(remote_service, "resolve_server", lambda name: ("https://srv", "k", {"EXAMPLE": {"x": 1}}))
monkeypatch.setattr(remote_service, "RemoteService", lambda *a, **kw: built.update(args=a, **kw))
remote_service.make_remote_service(
None, "srv", "AMZN", "id", {"service_params": {"extras": "y", "service": "DSNP", "proxy": "x"}}, False
None, "srv", "EXAMPLE", "id", {"service_params": {"extras": "y", "service": "OTHER", "proxy": "x"}}, False
)
assert built["service_params"] == {"extras": "y"}
assert "AMZN" not in built["args"][5] # apply_overrides=False keeps the remote's overrides out
assert "EXAMPLE" not in built["args"][5] # apply_overrides=False keeps the remote's overrides out
remote_service.make_remote_service(None, "srv", "AMZN", "id", {"extras": "z", "service": "DSNP", "quality": 1})
remote_service.make_remote_service(None, "srv", "EXAMPLE", "id", {"extras": "z", "service": "OTHER", "quality": 1})
assert built["service_params"] == {"extras": "z"}
assert built["args"][5]["AMZN"] == {"x": 1}
assert built["args"][5]["EXAMPLE"] == {"x": 1}
def test_remote_error_carries_its_reason_and_exits_a_cli_run():
@@ -145,8 +145,8 @@ def test_remote_error_carries_its_reason_and_exits_a_cli_run():
def test_a_geofenced_remote_service_gets_its_first_allowed_region_as_proxy(monkeypatch):
monkeypatch.setattr(handlers, "server_proxy_allowed", lambda request=None: True)
fenced = {"tag": "PCOK", "geofence": ["US", "ca"], "geoblock": ["us"]}
assert handlers.geofence_proxy({}, {"tag": "PCOK", "geofence": ["US", "ca"]}) == {"proxy": "us"}
fenced = {"tag": "EXAMPLE", "geofence": ["US", "ca"], "geoblock": ["us"]}
assert handlers.geofence_proxy({}, {"tag": "EXAMPLE", "geofence": ["US", "ca"]}) == {"proxy": "us"}
assert handlers.geofence_proxy({}, fenced) == {"proxy": "ca"}
assert handlers.geofence_proxy({"proxy": "gb"}, fenced) == {}
assert handlers.geofence_proxy({"no_proxy": True}, fenced) == {}
@@ -194,7 +194,7 @@ async def test_a_remote_login_prompt_waits_for_its_owners_answer(monkeypatch):
class Remote:
auth_notice = None
login = handlers.RemoteLogin("owner", "srv", "DSNP", "show", Remote())
login = handlers.RemoteLogin("owner", "srv", "OTHER", "show", Remote())
monkeypatch.setattr(handlers, "_remote_logins", {login.id: login})
monkeypatch.setattr(handlers, "is_admin", lambda request: False)
caller = {"key": "owner"}
tests/remote/unit/test_remote_service_helpers.py
+1 -1
View File
@@ -320,7 +320,7 @@ def test_load_cache_files_wiring(tmp_path, monkeypatch) -> None:
def test_load_cache_files_forwards_nested_keys(tmp_path, monkeypatch) -> None:
"""A service that keeps its cache in subfolders (NF: session_web/<sha1>.json) must have
"""A service that keeps its cache in subfolders (session_web/<sha1>.json) must have
those files forwarded under the Cacher key, and the credential filter must still apply
to the basename of a nested file."""
import base64
tests/remote/unit/test_server_cdm_type_override.py
+12 -7
View File
@@ -69,14 +69,14 @@ def _wv_drm():
def _session(track):
# handle_single_server_cdm looks config.cdm up by the service class name, so the fake needs the tag as its name
service = type("PCOK", (SimpleNamespace,), {})(
service = type("EXAMPLE", (SimpleNamespace,), {})(
get_widevine_service_certificate=lambda **k: None,
get_widevine_license=lambda **k: b"",
get_playready_license=lambda **k: b"",
)
return SimpleNamespace(
lock=asyncio.Lock(),
service_tag="PCOK",
service_tag="EXAMPLE",
service_instance=service,
served_keys={},
tracks=SimpleNamespace(get={"vid": track}.get),
@@ -170,7 +170,10 @@ async def test_quality_keyed_mapping_resolves_to_the_device_it_loads(env, monkey
monkeypatch.setattr(
handlers.config,
"cdm",
{"PCOK": {"<=1080": "wv_tier", ">1080": {"widevine": "wv_tier", "playready": "pr_tier"}}, "default": "pr_dev"},
{
"EXAMPLE": {"<=1080": "wv_tier", ">1080": {"widevine": "wv_tier", "playready": "pr_tier"}},
"default": "pr_dev",
},
raising=False,
)
@@ -183,7 +186,9 @@ async def test_quality_keyed_mapping_resolves_to_the_device_it_loads(env, monkey
async def test_both_systems_mapped_follow_the_client(env, monkeypatch):
"""A mapping with a device per system settles nothing, so the client's choice goes first."""
monkeypatch.setattr(handlers.config, "cdm", {"PCOK": {"widevine": "wv_dev", "playready": "pr_dev"}}, raising=False)
monkeypatch.setattr(
handlers.config, "cdm", {"EXAMPLE": {"widevine": "wv_dev", "playready": "pr_dev"}}, raising=False
)
track = SimpleNamespace(id="vid", drm=[_pr_drm(), _wv_drm()])
body = await _license(monkeypatch, track, {"drm_type": "widevine"})
@@ -192,7 +197,7 @@ async def test_both_systems_mapped_follow_the_client(env, monkeypatch):
async def test_no_candidate_header_is_a_client_error(env, monkeypatch):
monkeypatch.setattr(handlers.config, "cdm", {"PCOK": "pr_dev"}, raising=False)
monkeypatch.setattr(handlers.config, "cdm", {"EXAMPLE": "pr_dev"}, raising=False)
track = SimpleNamespace(id="vid", drm=[_wv_drm()])
with pytest.raises(APIError, match="No PSSH on track"):
@@ -201,7 +206,7 @@ async def test_no_candidate_header_is_a_client_error(env, monkeypatch):
async def test_track_preference_goes_first_when_the_server_has_that_device(env, monkeypatch):
"""A track that prefers PlayReady licenses with PlayReady even on a Widevine-mapped server."""
monkeypatch.setattr(handlers.config, "cdm", {"PCOK": {"widevine": "wv_dev", "default": "pr_dev"}}, raising=False)
monkeypatch.setattr(handlers.config, "cdm", {"EXAMPLE": {"widevine": "wv_dev", "default": "pr_dev"}}, raising=False)
track = SimpleNamespace(id="vid", drm=[_pr_drm(), _wv_drm()], drm_preference="pr")
body = await _license(monkeypatch, track, {"drm_type": "widevine"})
@@ -212,7 +217,7 @@ async def test_track_preference_goes_first_when_the_server_has_that_device(env,
async def test_track_preference_falls_back_with_a_warning(env, monkeypatch, caplog):
"""With no PlayReady device the preference cannot be met: Widevine licenses and the client hears why."""
monkeypatch.setattr(handlers.config, "cdm", {"PCOK": "wv_dev"}, raising=False)
monkeypatch.setattr(handlers.config, "cdm", {"EXAMPLE": "wv_dev"}, raising=False)
track = SimpleNamespace(id="vid", drm=[_pr_drm(), _wv_drm()], drm_preference="pr")
with caplog.at_level("WARNING", logger=handlers.log.name):