mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
"""services.<TAG>.dl defaults reach REST API download jobs, below the values the request sends."""
|
|
|
|
import asyncio
|
|
from datetime import datetime
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.api import download_manager as dm
|
|
from unshackle.core.api import handlers
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
class FakeManager:
|
|
def __init__(self):
|
|
self.params = None
|
|
|
|
async def start_workers(self):
|
|
pass
|
|
|
|
def create_job(self, service, title_id, owner_key=None, **params):
|
|
self.params = params
|
|
return SimpleNamespace(job_id="j", status=SimpleNamespace(value="queued"), created_time=datetime.now())
|
|
|
|
|
|
@pytest.fixture
|
|
def env(monkeypatch):
|
|
manager = FakeManager()
|
|
gate_input = {}
|
|
monkeypatch.setattr(handlers, "validate_service", lambda service, request=None: "EXAMPLE1")
|
|
monkeypatch.setattr(handlers.Services, "load", lambda service: SimpleNamespace())
|
|
monkeypatch.setattr(dm, "get_download_manager", lambda: manager)
|
|
monkeypatch.setattr(handlers, "enforce_download_gates", lambda params, request: gate_input.update(params))
|
|
monkeypatch.setattr(handlers, "caller_key", lambda request: None)
|
|
monkeypatch.setattr(handlers.config, "serve", {}, raising=False)
|
|
monkeypatch.setattr(handlers.config, "services", {}, raising=False)
|
|
|
|
def run(body=None, dl=None, serve=None):
|
|
if dl is not None:
|
|
monkeypatch.setattr(handlers.config, "services", {"EXAMPLE1": {"dl": dl}})
|
|
if serve is not None:
|
|
monkeypatch.setattr(handlers.config, "serve", serve)
|
|
asyncio.run(handlers.download_handler({"service": "EXAMPLE1", "title_id": "t1", **(body or {})}, None))
|
|
return manager.params
|
|
|
|
run.gate = gate_input
|
|
return run
|
|
|
|
|
|
def test_service_dl_sets_defaults(env):
|
|
params = env(dl={"workers": 7, "range": ["HDR10"]})
|
|
assert params["workers"] == 7
|
|
assert params["range"] == ["HDR10"]
|
|
|
|
|
|
def test_request_value_beats_service_dl(env):
|
|
assert env(body={"workers": 3}, dl={"workers": 7})["workers"] == 3
|
|
|
|
|
|
def test_service_dl_beats_serve(env):
|
|
assert env(dl={"workers": 7}, serve={"workers": 5})["workers"] == 7
|
|
|
|
|
|
def test_routing_and_unknown_keys_are_ignored(env):
|
|
baseline = env()
|
|
params = env(dl={"remote": True, "server": "x", "speed_limit": "1M"})
|
|
assert params.get("remote") == baseline.get("remote")
|
|
assert params.get("server") == baseline.get("server")
|
|
assert "speed_limit" not in params
|
|
|
|
|
|
def test_click_spelling_of_range_lands_as_range(env):
|
|
assert env(dl={"range_": ["HLG"]})["range"] == ["HLG"]
|
|
|
|
|
|
def test_gate_sees_service_dl(env):
|
|
env(dl={"quality": [2160]})
|
|
assert env.gate["quality"] == [2160]
|
|
|
|
|
|
def test_client_keys_are_the_body_keys(env):
|
|
params = env(body={"workers": 3, "client_keys": ["forged"], "quality": [1080]}, dl={"workers": 7})
|
|
assert params["client_keys"] == ["client_keys", "quality", "workers"]
|
|
|
|
|
|
def test_relayed_job_skips_service_dl(env, monkeypatch):
|
|
async def relay(data, request):
|
|
return "remote-server", {"tag": "EXAMPLE1"}, data
|
|
|
|
monkeypatch.setattr(handlers, "relay_target", relay)
|
|
baseline_workers = handlers.DEFAULT_DOWNLOAD_PARAMS["workers"]
|
|
params = env(dl={"workers": 7})
|
|
assert params["workers"] == baseline_workers
|
|
assert set(params["client_keys"]) >= set(handlers.DEFAULT_DOWNLOAD_PARAMS)
|
|
|
|
|
|
def test_client_sent_keys_uses_the_stamp():
|
|
params = {"workers": 1, "quality": [1080], "client_keys": ["workers"]}
|
|
assert dm.client_sent_keys(params) == {"workers"}
|
|
assert "client_keys" not in params
|
|
|
|
|
|
def test_client_sent_keys_without_stamp_counts_all_keys():
|
|
assert dm.client_sent_keys({"workers": 1, "quality": None}) == {"workers", "quality"}
|