mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
"""Tests for the ``unshackle import`` command: it reads every export format and forwards to ``dl``."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from unshackle.commands.dl import dl
|
|
|
|
KID = "00000000000000000000000000000001"
|
|
KEY = "aa" * 16
|
|
|
|
import_cli = importlib.import_module("unshackle.commands.import").ImportCommand.cli
|
|
|
|
MEDIAEXPORT = {
|
|
"kind": "mediaexport",
|
|
"version": 1,
|
|
"service": {"tag": "EXAMPLE", "name": "Example"},
|
|
"titles": [
|
|
{
|
|
"id": "movie-1",
|
|
"kind": "movie",
|
|
"title": "Example Movie",
|
|
"manifests": [{"url": "https://example.test/m.mpd", "type": "dash", "role": "primary"}],
|
|
"keys": {KID: KEY},
|
|
}
|
|
],
|
|
}
|
|
|
|
UNSHACKLE_V2 = {
|
|
"version": 2,
|
|
"service": "EXAMPLE",
|
|
"titles": {
|
|
"movie-1": {
|
|
"meta": {"type": "movie", "name": "Example Movie"},
|
|
"manifest_url": "https://example.test/m.mpd",
|
|
"manifest_type": "DASH",
|
|
"tracks": {"v1": {"id": "v1", "url": "https://example.test/m.mpd", "keys": {KID: KEY}}},
|
|
}
|
|
},
|
|
}
|
|
|
|
UNIDL_V1 = {
|
|
"kind": "unidl-export",
|
|
"version": 1,
|
|
"service": "EXAMPLE",
|
|
"titles": [
|
|
{
|
|
"title": {"id": "movie-1", "kind": "movie", "name": "Example Movie"},
|
|
"manifest_url": "https://example.test/m.mpd",
|
|
"keys": [f"{KID}:{KEY}"],
|
|
}
|
|
],
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def forwarded(monkeypatch: pytest.MonkeyPatch) -> list[list[str]]:
|
|
"""The argument lists the import command sends to ``dl``."""
|
|
calls: list[list[str]] = []
|
|
|
|
def fake_main(*, args: list[str], **_: Any) -> None:
|
|
calls.append(list(args))
|
|
|
|
monkeypatch.setattr(dl.cli, "main", fake_main)
|
|
return calls
|
|
|
|
|
|
@pytest.mark.parametrize("raw", [MEDIAEXPORT, UNSHACKLE_V2, UNIDL_V1], ids=["mediaexport", "unshackle-v2", "unidl-v1"])
|
|
def test_import_forwards_service_tag(tmp_path: Path, forwarded: list[list[str]], raw: dict[str, Any]) -> None:
|
|
export = tmp_path / "export.json"
|
|
export.write_text(json.dumps(raw), encoding="utf-8")
|
|
|
|
result = CliRunner().invoke(import_cli, [str(export), "-r", "HDR10"])
|
|
|
|
assert result.exit_code == 0, result.output
|
|
assert forwarded == [["-r", "HDR10", "--import", str(export), "EXAMPLE"]]
|
|
|
|
|
|
def test_import_rejects_a_file_that_is_not_an_export(tmp_path: Path, forwarded: list[list[str]]) -> None:
|
|
export = tmp_path / "export.json"
|
|
export.write_text(json.dumps({"hello": "world"}), encoding="utf-8")
|
|
|
|
result = CliRunner().invoke(import_cli, [str(export)])
|
|
|
|
assert result.exit_code != 0
|
|
assert "is not a usable export file" in result.output
|
|
assert forwarded == []
|