backend/tests/test_jimaku_matching.py
Rama master · réplica de lectura del repositorio original.
"""Jimaku: only the queried series, season and episode (names from the live API, 2026-11)."""
from unittest.mock import MagicMock
import pytest
from providers.base import VideoQuery
from providers.jimaku import JimakuProvider
from providers.jimaku_matching import (
entry_season,
file_matches_episode,
is_later_part,
search_names,
select_entries,
)
MUSHOKU = [
{
"id": 1927,
"name": "Mushoku Tensei II: Isekai Ittara Honki Dasu",
"english_name": "Mushoku Tensei: Jobless Reincarnation Season 2",
},
{
"id": 2699,
"name": "Mushoku Tensei II: Isekai Ittara Honki Dasu Part 2",
"english_name": "Mushoku Tensei: Jobless Reincarnation Season 2 Part 2",
},
{
"id": 12216,
"name": "Mushoku Tensei III: Isekai Ittara Honki Dasu",
"english_name": "Mushoku Tensei: Jobless Reincarnation Season 3",
},
{
"id": 1426,
"name": "Mushoku Tensei: Isekai Ittara Honki Dasu",
"english_name": "Mushoku Tensei: Jobless Reincarnation",
},
{
"id": 3547,
"name": "Mushoku Tensei: Isekai Ittara Honki Dasu Part 2",
"english_name": "Mushoku Tensei: Jobless Reincarnation Cour 2",
},
]
FRIEREN = [
{"id": 729, "name": "Sousou no Frieren", "english_name": "Frieren: Beyond Journey’s End"},
{
"id": 11446,
"name": "Sousou no Frieren 2nd Season",
"english_name": "Frieren: Beyond Journey’s End Season 2",
"japanese_name": "葬送のフリーレン 第2期",
},
{
"id": 7350,
"name": "Dragon Ball: Ossu! Kaette Kita Son Goku to Nakama-tachi!!",
"english_name": "Dragon Ball: Yo! Son Goku and His Friends Return!!",
},
]
def _q(title="Mushoku Tensei: Jobless Reincarnation", season=1, episode=5, **kw):
return VideoQuery(series_title=title, season=season, episode=episode, languages=["ja"], **kw)
def _ids(entries):
return [e["id"] for e in entries]
@pytest.mark.parametrize(
("entry", "season"),
[(MUSHOKU[0], 2), (MUSHOKU[2], 3), (MUSHOKU[3], None), (MUSHOKU[4], None), (FRIEREN[1], 2)],
)
def test_entry_season(entry, season):
assert entry_season(entry) == season
def test_later_part():
assert is_later_part(MUSHOKU[4]) and is_later_part(MUSHOKU[1])
assert not is_later_part(MUSHOKU[3])
def test_season_one_keeps_the_unmarked_entries_first_part_first():
assert _ids(select_entries(MUSHOKU, _q(season=1))) == [1426, 3547]
def test_a_later_season_keeps_its_own_entries():
assert _ids(select_entries(MUSHOKU, _q(season=2))) == [1927, 2699]
assert _ids(select_entries(MUSHOKU, _q(season=3))) == [12216]
def test_other_series_are_dropped():
assert _ids(select_entries(FRIEREN, _q("Frieren: Beyond Journey's End", 1, 3))) == [729]
def test_no_title_match_keeps_the_old_behaviour():
assert _ids(select_entries(MUSHOKU, _q("Arbeitslos wiedergeboren"))) == _ids(MUSHOKU)
@pytest.mark.parametrize(
("name", "fits"),
[
("[SubsPlease] Mushoku Tensei - 05v2 (1080p) [DBC034A9].ja.srt", True),
("無職転生.～異世界行ったら本気だす～.S01E05.お嬢様と暴力.WEBRip.Netflix.ja[cc].srt", True),
(
"[Nekomoe kissaten&VCB-Studio] Mushoku Tensei ~Isekai Ittara Honki Dasu~ [05][Ma10p_1080p].ass",
True,
),
("[Netflix] Mushoku Tensei Jobless Reincarnation 05 (non SDH).srt", True),
("[SubsPlease] Mushoku Tensei - 06 (1080p) [DBC034A9].ja.srt", False),
("無職転生.～異世界行ったら本気だす～.S02E05.WEBRip.Netflix.ja[cc].srt", False),
("[NanakoRaws] Sousou no Frieren - 01-04 (1080p).ass", False),
("Mushoku Tensei OP.ass", False),
],
)
def test_file_matches_episode(name, fits):
assert file_matches_episode(name, _q(), 5) is fits
def test_a_multi_episode_video_takes_its_range():
assert file_matches_episode(
"[NanakoRaws] Sousou no Frieren - 01-04 (1080p).ass",
_q(episode=3, episodes=[1, 2, 3, 4]),
3,
)
def test_search_names():
q = _q("Frieren: Beyond Journey's End", series_aliases=[("Sousou no Frieren", -1, None)])
assert search_names(q) == ["Frieren: Beyond Journey's End", "Sousou no Frieren", "Frieren"]
def test_search_offers_only_the_episode_with_its_matches():
provider = JimakuProvider(api_key="k")
provider.session = MagicMock()
provider._search_by_name = MagicMock(return_value=MUSHOKU)
files = {
3: "[SubsPlease] Mushoku Tensei S3 - 05 (1080p).ja.srt",
1426: "[SubsPlease] Mushoku Tensei - 05v2 (1080p) [DBC034A9].ja.srt",
1427: "[SubsPlease] Mushoku Tensei - 06 (1080p) [DBC034A9].ja.srt",
}
provider._get_entry_files = MagicMock(
side_effect=lambda entry_id, _q: (
[
(files[1426], "u1", False),
(files[1427], "u2", False),
("Mushoku Tensei.zip", "u3", True),
]
if entry_id == 1426
else [(files[3], "u4", False)]
)
)
results = provider.search(_q())
names = {r.filename: r.matches for r in results}
assert set(names) == {files[1426], "Mushoku Tensei.zip"}
assert {"episode", "season"} <= names[files[1426]]
assert "episode" not in names["Mushoku Tensei.zip"]
assert [c.args[0] for c in provider._get_entry_files.call_args_list] == [1426, 3547]
def test_name_search_falls_back_to_other_names():
provider = JimakuProvider(api_key="k")
provider.session = MagicMock()
provider._search_by_name = MagicMock(side_effect=[[], FRIEREN])
provider._get_entry_files = MagicMock(return_value=[])
provider.search(_q("Frieren: Beyond Journey's End", 1, 3))
assert [c.args[0] for c in provider._search_by_name.call_args_list] == [
"Frieren: Beyond Journey's End",
"Frieren",
]
@pytest.mark.parametrize(
("name", "lang"),
[
("[Nekomoe kissaten] Akiba Maid Sensou - 04 [WebRip][JPN].ass", "ja"),
("[SubsPlease] Mushoku Tensei - 05v2 (1080p) [DBC034A9].ja.srt", "ja"),
("Pokemon Horizon.S01E01.Amazon.ja[sdh].srt", "ja"),
("Show - 01 [ENG].ass", "en"),
("Show.S01E01.en.srt", "en"),
# Two languages in one file: a clean subtitle in neither.
("[Nekomoe kissaten] Akiba Maid Sensou [04][BDRip][CHS, JPN].ass", None),
("[SubsPlease] Sousou no Frieren - 03 (1080p) [7EF3F175]_ja-en.ass", None),
("[NanakoRaws] Otonari no Tenshi-sama S01E03 (AT-X 1080p HEVC AAC).ass", None),
("Show [1080p][Web].ass", None),
],
)
def test_language_from_tags(name, lang):
from providers.jimaku_matching import language_from_tags
assert language_from_tags(name) == lang