Initial commit

tests/unit/test_namer.py

@@ -0,0 +1,94 @@
+from pathlib import Path
+import pytest
+from media_sorter.analyzer import MediaMetadata
+from media_sorter.classifier import ClassificationResult
+from media_sorter.config import Settings
+from media_sorter.namer import MediaNamer, sanitize_filename_component
+from media_sorter.tokenizer import TokenizedFilename
+
+
+@pytest.fixture
+def settings():
+    return Settings()
+
+
+@pytest.fixture
+def namer(settings):
+    return MediaNamer(settings)
+
+
+def test_sanitize_filename_forbidden_chars():
+    messy = 'Movie: "The Final Chapter" <Director\'s Cut> | Part 1?.mkv'
+    cleaned = sanitize_filename_component(messy)
+    assert ":" not in cleaned
+    assert '"' not in cleaned
+    assert "<" not in cleaned
+    assert ">" not in cleaned
+    assert "|" not in cleaned
+    assert "?" not in cleaned
+    assert cleaned.endswith(".mkv")
+
+
+def test_sanitize_windows_reserved_names():
+    res = sanitize_filename_component("CON.mp4")
+    assert res == "_CON.mp4"
+    res2 = sanitize_filename_component("nul.txt")
+    assert res2 == "_nul.txt"
+
+
+def test_generate_movie_destination(namer):
+    tokens = TokenizedFilename(
+        raw_name="The.Matrix.1999.1080p.BluRay.x264.mkv",
+        title="The Matrix",
+        year=1999,
+        resolution="1080p",
+        video_codec="x264",
+    )
+    meta = MediaMetadata(path=Path("The.Matrix.1999.1080p.BluRay.x264.mkv"), mime_type="video/x-matroska", container="mkv")
+    cls_res = ClassificationResult(category="movie", confidence=0.95, tokens=tokens, metadata=meta)
+
+    dest = namer.generate_destination_path(cls_res)
+    assert "The Matrix (1999)" in str(dest)
+    assert dest.suffix == ".mkv"
+
+
+def test_generate_tv_destination(namer):
+    tokens = TokenizedFilename(
+        raw_name="Breaking Bad S01E01 Pilot.mkv",
+        title="Breaking Bad",
+        season=1,
+        episode=1,
+        episode_title="Pilot",
+    )
+    meta = MediaMetadata(path=Path("Breaking Bad S01E01 Pilot.mkv"), mime_type="video/x-matroska", container="mkv")
+    cls_res = ClassificationResult(category="tv", confidence=0.95, tokens=tokens, metadata=meta)
+
+    dest = namer.generate_destination_path(cls_res)
+    assert "Season 01" in str(dest)
+    assert "Breaking Bad" in str(dest)
+    assert "S01E01" in str(dest)
+
+
+def test_generate_quarantine_destination(namer):
+    meta = MediaMetadata(path=Path("weird_unknown_file.xyz"), mime_type="application/octet-stream", container="xyz")
+    cls_res = ClassificationResult(
+        category="unknown",
+        confidence=0.1,
+        metadata=meta,
+        needs_quarantine=True,
+        quarantine_reason="unrecognized_format",
+    )
+
+    dest = namer.generate_destination_path(cls_res)
+    assert "Quarantine" in str(dest)
+    assert "unrecognized-format" in str(dest) or "unrecognized_format" in str(dest)
+
+
+def test_sidecar_subtitle_matching(namer):
+    sub_meta = MediaMetadata(path=Path("movie.en.srt"), mime_type="text/plain", container="srt")
+    cls_res = ClassificationResult(category="subtitle", confidence=0.95, metadata=sub_meta)
+    primary_dst = Path("/organized/Movies/Inception (2010)/Inception (2010) [1080p].mkv")
+
+    sub_dst = namer.generate_destination_path(cls_res, primary_dst_path=primary_dst)
+    assert sub_dst.parent == primary_dst.parent
+    assert sub_dst.name == "Inception (2010) [1080p].en.srt"