Initial portable pretranscode service

2026-08-26 09:53:42 +10:00
commit 093315b382
11 changed files with 1077 additions and 0 deletions
@@ -0,0 +1,69 @@
+import importlib.util
+import json
+import os
+import tempfile
+import unittest
+from pathlib import Path
+
+
+class CoreTests(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.temp = tempfile.TemporaryDirectory()
+        root = Path(cls.temp.name)
+        for name in ("source", "view", "cache", "state"):
+            (root / name).mkdir()
+        (root / "cache" / ".pretranscode-cache").touch()
+        os.environ.update({
+            "SOURCE_ROOT": str(root / "source"),
+            "VIEW_ROOT": str(root / "view"),
+            "CACHE_ROOT": str(root / "cache"),
+            "STATE_ROOT": str(root / "state"),
+            "JELLYFIN_LIBRARY_ROOT": "/library",
+            "JELLYFIN_ORIGINAL_ROOT": "/original",
+            "JELLYFIN_CACHE_ROOT": "/cached",
+        })
+        spec = importlib.util.spec_from_file_location("pretranscode", Path(__file__).parents[1] / "pretranscode.py")
+        cls.module = importlib.util.module_from_spec(spec)
+        spec.loader.exec_module(cls.module)
+        cls.root = root
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.temp.cleanup()
+
+    def test_paths_are_configurable(self):
+        m = self.module
+        self.assertEqual(m.relative_source("/library/tv/Show/episode.mkv"), "tv/Show/episode.mkv")
+        self.assertEqual(m.source_target("tv/Show/episode.mkv"), "/original/tv/Show/episode.mkv")
+        self.assertTrue(m.cache_target("tv/Show/episode.mkv").startswith("/cached/files/"))
+
+    def test_series_lock_key(self):
+        m = self.module
+        self.assertEqual(m.item_key("tv/Show/Season 1/Episode.mkv"), "series:Show")
+        self.assertEqual(m.item_key("movies/Movie/file.mkv"), "item:movies/Movie/file.mkv")
+
+    def test_fixed_codec_blacklist(self):
+        manager = self.module.Manager()
+        self.assertTrue(manager.incompatible({"codec": "hevc", "bit_depth": 10}))
+        self.assertTrue(manager.incompatible({"codec": "h264", "bit_depth": 10}))
+        self.assertFalse(manager.incompatible({"codec": "h264", "bit_depth": 8}))
+        self.assertFalse(manager.incompatible({"codec": "av1", "bit_depth": 10}))
+
+    def test_cache_requires_matching_source_fingerprint(self):
+        m = self.module
+        manager = m.Manager()
+        relative = "movies/Test/Test.mkv"
+        source = self.root / "source" / relative
+        source.parent.mkdir(parents=True, exist_ok=True)
+        source.write_bytes(b"original")
+        output, manifest = manager.cache_paths(relative)
+        output.write_bytes(b"cached")
+        manifest.write_text(json.dumps({"relative": relative, "source_fingerprint": m.fingerprint(source)}))
+        self.assertTrue(manager.cache_valid(relative))
+        source.write_bytes(b"changed")
+        self.assertFalse(manager.cache_valid(relative))
+
+
+if __name__ == "__main__":
+    unittest.main()