fix(import): read export files through mediaexport

The command still required the old v2 format.
2026-11-12 20:15:10 +00:00 · 2026-09-23 13:30:35 -06:00
parent 783c8ae1c2
commit 8a942e7795
3 changed files with 101 additions and 15 deletions
@@ -0,0 +1,93 @@
+"""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 == []