mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
688 lines
25 KiB
Python
688 lines
25 KiB
Python
# -*- coding: utf-8 -*-
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from app.domain.context import MediaInfo
|
|
from app.domain.metainfo import MetaInfo, MetaInfoPath, find_metainfo
|
|
from app.domain.meta.metabase import MetaBase
|
|
from app.domain.meta.metamusic import MetaMusic
|
|
from app.domain.meta.metaanime import MetaAnime
|
|
from app.application.torrent import TorrentHelper
|
|
from app.schemas.types import MediaSource, MediaType
|
|
from tests.cases.meta import meta_cases
|
|
|
|
|
|
def test_metainfo():
|
|
"""测试常见标题元数据识别结果。"""
|
|
for info in meta_cases:
|
|
if info.get("path"):
|
|
meta_info = MetaInfoPath(path=Path(info.get("path")))
|
|
else:
|
|
meta_info = MetaInfo(
|
|
title=info.get("title"),
|
|
subtitle=info.get("subtitle"),
|
|
custom_words=["#"],
|
|
)
|
|
target = {
|
|
"type": meta_info.type.value,
|
|
"cn_name": meta_info.cn_name or "",
|
|
"en_name": meta_info.en_name or "",
|
|
"year": meta_info.year or "",
|
|
"part": meta_info.part or "",
|
|
"season": meta_info.season,
|
|
"episode": meta_info.episode,
|
|
"restype": meta_info.edition,
|
|
"pix": meta_info.resource_pix or "",
|
|
"video_codec": meta_info.video_encode or "",
|
|
"audio_codec": meta_info.audio_encode or "",
|
|
"fps": meta_info.fps or None,
|
|
}
|
|
|
|
if info.get("target").get("media_source"):
|
|
target["media_source"] = str(meta_info.media_source)
|
|
target["media_id"] = meta_info.media_id
|
|
|
|
expected = info.get("target")
|
|
if "fps" not in expected:
|
|
target.pop("fps", None)
|
|
assert target == expected
|
|
|
|
|
|
def test_emby_format_ids():
|
|
"""测试 Emby 格式 ID 识别。"""
|
|
test_paths = [
|
|
(
|
|
"/movies/The Vampire Diaries (2009) [tmdbid=18165]/The.Vampire.Diaries.S01E01.1080p.mkv",
|
|
18165,
|
|
),
|
|
("/movies/Inception (2010) [tmdbid-27205]/Inception.2010.1080p.mkv", 27205),
|
|
(
|
|
"/movies/Breaking Bad (2008) [tmdb=1396]/Season 1/Breaking.Bad.S01E01.1080p.mkv",
|
|
1396,
|
|
),
|
|
(
|
|
"/tv/Game of Thrones (2011) {tmdb=1399}/Season 1/Game.of.Thrones.S01E01.1080p.mkv",
|
|
1399,
|
|
),
|
|
("/movies/Avatar (2009) {tmdb-19995}/Avatar.2009.1080p.mkv", 19995),
|
|
]
|
|
|
|
for path_str, expected_media_id in test_paths:
|
|
meta = MetaInfoPath(Path(path_str))
|
|
assert meta.media_source == MediaSource.TMDB
|
|
assert meta.media_id == str(expected_media_id)
|
|
assert not hasattr(meta, "tmdbid")
|
|
|
|
|
|
def test_metainfopath_with_custom_words():
|
|
"""测试 MetaInfoPath 使用自定义识别词。"""
|
|
custom_words = ["测试替换 => "]
|
|
path = Path("/movies/电影测试替换名称 (2024)/movie.mkv")
|
|
meta = MetaInfoPath(path, custom_words=custom_words)
|
|
if meta.cn_name:
|
|
assert "测试替换" not in meta.cn_name
|
|
|
|
|
|
def test_metainfopath_without_custom_words():
|
|
"""测试 MetaInfoPath 不传入自定义识别词。"""
|
|
path = Path("/movies/Normal Movie (2024)/movie.mkv")
|
|
meta = MetaInfoPath(path)
|
|
assert meta is not None
|
|
|
|
|
|
def test_metainfopath_with_empty_custom_words():
|
|
"""测试 MetaInfoPath 传入空的自定义识别词。"""
|
|
path = Path("/movies/Test Movie (2024)/movie.mkv")
|
|
meta = MetaInfoPath(path, custom_words=[])
|
|
assert meta is not None
|
|
|
|
|
|
def test_custom_words_apply_words_recording():
|
|
"""测试 apply_words 记录功能。"""
|
|
custom_words = ["替换词 => 新词"]
|
|
title = "电影替换词.2024.mkv"
|
|
meta = MetaInfo(title=title, custom_words=custom_words)
|
|
assert hasattr(meta, "apply_words")
|
|
if meta.apply_words:
|
|
assert "替换词 => 新词" in meta.apply_words
|
|
|
|
|
|
def test_metainfo_preserves_original_name_when_custom_words_applied():
|
|
"""测试应用识别词后仍保留未应用识别词时识别出的名称。"""
|
|
custom_words = ["测试替换 => "]
|
|
meta = MetaInfo(title="电影测试替换名称 (2024)", custom_words=custom_words)
|
|
assert meta.name == "电影名称"
|
|
assert meta.original_name == "电影测试替换名称"
|
|
|
|
|
|
def test_torrent_title_match_ignores_question_mark_variants():
|
|
"""问号差异不应导致番剧罗马字标题匹配失败。"""
|
|
mediainfo = SimpleNamespace(
|
|
title="哪里有温柔对待阿宅的辣妹！？",
|
|
original_title="オタクに優しいギャルはいない!?",
|
|
names=["Otaku ni Yasashii Gal wa Inai!?"],
|
|
type=MediaType.TV,
|
|
year=None,
|
|
tmdb_id=None,
|
|
douban_id=None,
|
|
imdb_id=None,
|
|
season_years={},
|
|
)
|
|
torrent_meta = SimpleNamespace(
|
|
cn_name=None,
|
|
en_name="Otaku ni Yasashii Gal wa Inai",
|
|
type=MediaType.TV,
|
|
year=None,
|
|
org_string=None,
|
|
)
|
|
torrent = SimpleNamespace(
|
|
site_name="MiKan",
|
|
title="[今晚月色真美][Otaku ni Yasashii Gal wa Inai!?][12][1080P]",
|
|
category=MediaType.TV.value,
|
|
description=None,
|
|
)
|
|
|
|
assert TorrentHelper.match_torrent(
|
|
mediainfo=mediainfo,
|
|
torrent_meta=torrent_meta,
|
|
torrent=torrent,
|
|
)
|
|
|
|
mediainfo.names = []
|
|
torrent_meta.cn_name = "哪里有温柔对待阿宅的辣妹"
|
|
torrent_meta.en_name = None
|
|
assert TorrentHelper.match_torrent(
|
|
mediainfo=mediainfo,
|
|
torrent_meta=torrent_meta,
|
|
torrent=torrent,
|
|
)
|
|
|
|
|
|
def test_torrent_title_match_rejects_season_absent_from_target_series():
|
|
"""无年份同名剧的资源季超出目标剧季范围时应拒绝。"""
|
|
mediainfo = MediaInfo(
|
|
title="家族计划",
|
|
original_title="가족계획",
|
|
names=["Family Matters"],
|
|
type=MediaType.TV,
|
|
year="2024",
|
|
number_of_seasons=1,
|
|
seasons={1: list(range(1, 7))},
|
|
season_years={1: "2024"},
|
|
)
|
|
torrent_meta = MetaInfo("Family Matters S02 1080p WEBRip DD2.0 x264-TrollHD")
|
|
torrent = SimpleNamespace(
|
|
site_name="测试站点",
|
|
title=torrent_meta.org_string,
|
|
category=MediaType.TV.value,
|
|
description=None,
|
|
)
|
|
|
|
assert not TorrentHelper.match_torrent(mediainfo, torrent_meta, torrent)
|
|
|
|
|
|
def test_python_metainfo_fallback_preserves_xxx_movie_title():
|
|
"""Python 兜底解析不应删除合法 xXx 片名。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo("xXx 2002 1080p AMZN WEB-DL H.264 DDP 5.1-FROGWeb")
|
|
|
|
assert meta.en_name == "Xxx"
|
|
assert meta.year == "2002"
|
|
assert meta.resource_pix == "1080p"
|
|
assert meta.edition == "WEB-DL"
|
|
assert meta.audio_encode == "DDP 5.1"
|
|
|
|
|
|
def test_python_metainfo_fallback_recognizes_eac3_audio_codec():
|
|
"""Python 兜底解析应识别 EAC3 及其声道信息。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo("Test.Movie.2026.1080p.BluRay.x264.EAC3.5.1-GROUP")
|
|
|
|
assert meta.resource_pix == "1080p"
|
|
assert meta.video_encode == "x264"
|
|
assert meta.audio_encode == "EAC3 5.1"
|
|
|
|
|
|
RESOURCE_TYPE_CASES = [
|
|
(
|
|
"They.Will.Kill.You.2026.2160p.UHD.BluRay.Remux."
|
|
"HEVC.DV.TrueHD.7.1.Atmos.mkv",
|
|
"UHD BluRay REMUX",
|
|
),
|
|
(
|
|
"Movie.2026.2160p.UHD.Blu-ray.Remux.BDRip.HEVC.mkv",
|
|
"UHD BluRay REMUX BDRIP",
|
|
),
|
|
(
|
|
"Movie.2026.2160p.UHD.BluRay.UHD.Remux.Remux.HEVC.mkv",
|
|
"UHD BluRay REMUX",
|
|
),
|
|
(
|
|
"Movie.2026.1080p.WEB-DL.WEBRip.Remux.H264.mkv",
|
|
"WEB-DL WEBRip REMUX",
|
|
),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize(("title", "expected"), RESOURCE_TYPE_CASES)
|
|
def test_metainfo_preserves_all_resource_types(title, expected):
|
|
"""默认解析入口应按顺序保留并去重所有受支持的资源类型。"""
|
|
meta = MetaInfo(title)
|
|
|
|
assert meta.resource_type == expected
|
|
assert meta.edition.startswith(expected)
|
|
assert expected in meta.resource_term
|
|
|
|
|
|
@pytest.mark.parametrize(("title", "expected"), RESOURCE_TYPE_CASES)
|
|
def test_python_metainfo_preserves_all_resource_types(title, expected):
|
|
"""Python 兜底解析应按顺序保留并去重所有受支持的资源类型。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title)
|
|
|
|
assert meta.resource_type == expected
|
|
assert meta.edition.startswith(expected)
|
|
assert expected in meta.resource_term
|
|
|
|
|
|
def test_metainfo_routes_audio_filename_to_music():
|
|
"""音频文件名应直接走音乐分支并完成艺术家/曲名拆分，不再进入影视季集解析。"""
|
|
meta = MetaInfo("周杰伦 - 晴天.flac")
|
|
|
|
assert isinstance(meta, MetaMusic)
|
|
assert isinstance(meta, MetaBase)
|
|
assert meta.type == MediaType.MUSIC
|
|
assert meta.org_string == "周杰伦 - 晴天.flac"
|
|
assert meta.title == "晴天"
|
|
assert meta.artists == ["周杰伦"]
|
|
assert meta.audio_format == "FLAC"
|
|
# 音乐没有季集信息，兼容通用访问
|
|
assert meta.season is None
|
|
assert meta.episode is None
|
|
assert meta.apply_words == []
|
|
|
|
|
|
def test_metainfo_routes_audio_path_to_music_without_parent_merge():
|
|
"""音频路径应直接构造音乐元数据，不参与影视季集合并，并拆分歌手与曲名。"""
|
|
meta = MetaInfoPath(Path("/music/叶惠美/周杰伦 - 晴天.flac"))
|
|
|
|
assert isinstance(meta, MetaMusic)
|
|
assert meta.type == MediaType.MUSIC
|
|
assert meta.org_string == "周杰伦 - 晴天.flac"
|
|
# 文件名中的歌手与曲名应拆分，便于无标签音频搜索识别
|
|
assert meta.title == "晴天"
|
|
assert meta.artists == ["周杰伦"]
|
|
assert meta.audio_format == "FLAC"
|
|
|
|
|
|
def test_metainfo_keeps_video_path_for_non_audio_files():
|
|
"""非音频文件应继续走影视识别链，不受音频路由影响。"""
|
|
meta = MetaInfoPath(Path("/movies/Inception (2010)/Inception.2010.1080p.mkv"))
|
|
|
|
assert not isinstance(meta, MetaMusic)
|
|
assert meta.type != MediaType.MUSIC
|
|
|
|
|
|
def test_metainfo_music_round_trip_preserves_fields():
|
|
"""音频解析结果字典往返后应保留音乐字段。"""
|
|
meta = MetaInfoPath(Path("/music/周杰伦 - 晴天.flac"))
|
|
payload = meta.to_dict()
|
|
restored = MetaMusic.from_dict(payload)
|
|
|
|
assert restored.type == MediaType.MUSIC
|
|
assert restored.title == "晴天"
|
|
assert restored.artists == ["周杰伦"]
|
|
assert restored.audio_format == "FLAC"
|
|
assert payload["type"] == "音乐"
|
|
|
|
|
|
def test_python_subtitle_episode_range_fin_with_chinese_season():
|
|
"""Python 兜底解析应识别副标题中 [01-26Fin] 格式的集数范围（#6103）。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="JoJos Bizarre Adventure S01 2012 1080i BluRay x264 FLAC 2.0-AnimeF@ADE",
|
|
subtitle="JOJO的奇妙冒险 第一季 / JoJo's Bizarre Adventure [01-26Fin] [简繁字幕]",
|
|
)
|
|
|
|
assert meta.type == MediaType.TV
|
|
assert meta.begin_season == 1
|
|
assert meta.begin_episode == 1
|
|
assert meta.end_episode == 26
|
|
assert meta.total_episode == 26
|
|
|
|
|
|
def test_subtitle_episode_range_fin_with_default_parser():
|
|
"""默认解析路径应识别副标题中 [01-26Fin] 格式的集数范围（#6103）。"""
|
|
meta = MetaInfo(
|
|
title="JoJos Bizarre Adventure S01 2012 1080i BluRay x264 FLAC 2.0-AnimeF@ADE",
|
|
subtitle="JOJO的奇妙冒险 第一季 / JoJo's Bizarre Adventure [01-26Fin] [简繁字幕]",
|
|
)
|
|
|
|
assert meta.type == MediaType.TV
|
|
assert meta.begin_season == 1
|
|
assert meta.begin_episode == 1
|
|
assert meta.end_episode == 26
|
|
assert meta.total_episode == 26
|
|
|
|
|
|
def test_python_subtitle_episode_range_fin_without_chinese_marker():
|
|
"""副标题无中文季集标记时也应识别 [01-38Fin] 集数范围。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="Some Show S01 2022 1080p WEB-DL H264-GRP",
|
|
subtitle="Some Show [01-38Fin]",
|
|
)
|
|
|
|
assert meta.begin_episode == 1
|
|
assert meta.end_episode == 38
|
|
assert meta.total_episode == 38
|
|
|
|
|
|
def test_python_subtitle_episode_range_end_variant():
|
|
"""END/完结 等完结标记变体同样应识别为集数范围。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta_end = MetaInfo(
|
|
title="Some Show S01 2022 1080p WEB-DL H264-GRP",
|
|
subtitle="Some Show 01-24 END",
|
|
)
|
|
meta_cn = MetaInfo(
|
|
title="Some Show S01 2022 1080p WEB-DL H264-GRP",
|
|
subtitle="某剧 01-12完结",
|
|
)
|
|
|
|
assert meta_end.begin_episode == 1
|
|
assert meta_end.end_episode == 24
|
|
assert meta_cn.begin_episode == 1
|
|
assert meta_cn.end_episode == 12
|
|
|
|
|
|
def test_python_subtitle_year_range_not_treated_as_episodes():
|
|
"""年份范围（如 2019-2020）不得误识别为集数。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="Some Collection 2020 1080p WEB-DL H264-GRP",
|
|
subtitle="A Collection [2019-2020Fin]",
|
|
)
|
|
|
|
assert meta.begin_episode is None
|
|
assert meta.total_episode == 0
|
|
|
|
|
|
def test_python_subtitle_episode_range_fin_rejects_numeric_suffix():
|
|
"""Python 兜底解析不得把带数字后缀的完结范围截断识别为集数。"""
|
|
for subtitle in ("Some Show [01-26Fin]2", "Some Show 01-26Fin 2"):
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="Some Show S01 2022 1080p WEB-DL H264-GRP",
|
|
subtitle=subtitle,
|
|
)
|
|
|
|
assert meta.begin_episode is None
|
|
assert meta.total_episode == 0
|
|
|
|
|
|
def test_custom_words_replace_then_episode_offset():
|
|
"""测试复杂识别词仍按先替换、后集数偏移的顺序处理。"""
|
|
custom_words = ["旧名 => 新名 && 第 <> 集 >> EP+1"]
|
|
meta = MetaInfo(title="旧名 第03集", custom_words=custom_words)
|
|
assert meta.name == "新名"
|
|
assert meta.episode == "E04"
|
|
assert meta.apply_words == custom_words
|
|
|
|
|
|
def test_get_torrent_episodes_applies_custom_words():
|
|
"""种子文件集数解析应使用订阅识别词完成跨季集数映射。"""
|
|
custom_words = [
|
|
"A.Will.Eternal.S04 => 一念永恒{[tmdbid=107371;type=tv]}S01 "
|
|
"&& S01 <> 2160p >> EP+165"
|
|
]
|
|
|
|
episodes = TorrentHelper.get_torrent_episodes(
|
|
["A.Will.Eternal.S04E05.2026.2160p.WEB-DL.mkv"],
|
|
custom_words=custom_words,
|
|
)
|
|
|
|
assert episodes == [170]
|
|
|
|
|
|
def test_custom_words_episode_offset_supports_multiplication_expression():
|
|
"""测试集数偏移表达式支持乘法和连续运算。"""
|
|
custom_words = [
|
|
r"Ha.Ha.Ha.Ha.Ha.2026.S06E([0-1][0-9]).Part1 => 哈哈哈哈哈 (2020){[tmdbid=112732;type=tv]} S06E\1.Part1 && S06 <> .Part1 >> 2*EP-1"
|
|
]
|
|
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="Ha.Ha.Ha.Ha.Ha.2026.S06E03.Part1",
|
|
custom_words=custom_words,
|
|
)
|
|
|
|
assert meta.name == "哈哈哈哈哈"
|
|
assert meta.media_source == MediaSource.TMDB
|
|
assert meta.media_id == "112732"
|
|
assert meta.begin_season == 6
|
|
assert meta.episode == "E05"
|
|
assert meta.apply_words == custom_words
|
|
|
|
|
|
def test_custom_words_episode_offset_supports_repeated_ep_expression():
|
|
"""测试集数偏移表达式支持重复使用 EP 占位符。"""
|
|
custom_words = ["旧名 => 新名 && 第 <> 集 >> EP+EP-1"]
|
|
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title="旧名 第03集", custom_words=custom_words)
|
|
|
|
assert meta.name == "新名"
|
|
assert meta.episode == "E05"
|
|
assert meta.apply_words == custom_words
|
|
|
|
|
|
def test_custom_words_episode_offset_rejects_implicit_ep_expression():
|
|
"""测试集数偏移表达式不把 2EP 当作隐式乘法或字符串拼接。"""
|
|
custom_words = ["旧名 => 新名 && 第 <> 集 >> 2EP"]
|
|
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title="旧名 第03集", custom_words=custom_words)
|
|
|
|
assert meta.name == "新名"
|
|
assert meta.episode == "E03"
|
|
assert meta.apply_words == []
|
|
|
|
|
|
def test_custom_words_support_episode_group_parameter():
|
|
"""测试自定义识别词替换结果中的 g 参数会写入剧集组。"""
|
|
group_id = "5ad0ec240e0a26303f00d84d"
|
|
custom_words = [
|
|
f"Bakemonogatari => 物语系列 {{[tmdbid=46195;type=tv;g={group_id};s=1]}}"
|
|
]
|
|
meta = MetaInfo(title="Bakemonogatari 01", custom_words=custom_words)
|
|
assert meta.media_source == MediaSource.TMDB
|
|
assert meta.media_id == "46195"
|
|
assert meta.type.value == "电视剧"
|
|
assert meta.begin_season == 1
|
|
assert meta.episode_group == group_id
|
|
assert meta.apply_words == custom_words
|
|
|
|
|
|
def test_custom_words_support_special_season_zero_parameter():
|
|
"""显式媒体标签中的 s=0 应作为特别季写入元数据。"""
|
|
custom_words = [
|
|
"Test Show => 测试剧 {[tmdbid=12345;type=tv;s=0]}"
|
|
]
|
|
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title="Test Show 01", custom_words=custom_words)
|
|
|
|
assert meta.media_source == MediaSource.TMDB
|
|
assert meta.media_id == "12345"
|
|
assert meta.type.value == "电视剧"
|
|
assert meta.begin_season == 0
|
|
|
|
|
|
def test_find_metainfo_supports_episode_group_parameter():
|
|
"""测试显式媒体标签支持 g 剧集组参数。"""
|
|
group_id = "5ad0ec240e0a26303f00d84d"
|
|
title, metainfo = find_metainfo(f"物语系列 {{[tmdbid=46195;type=tv;g={group_id};s=1]}}")
|
|
assert metainfo["episode_group"] == group_id
|
|
assert "g=" not in title
|
|
|
|
|
|
def test_find_metainfo_does_not_support_episode_group_alias():
|
|
"""测试 e_group 不会被当作剧集组参数识别。"""
|
|
group_id = "5ad0ec240e0a26303f00d84d"
|
|
with patch("app.adapters.system.rust.find_metainfo", return_value=None):
|
|
_, metainfo = find_metainfo(f"物语系列 {{[tmdbid=46195;type=tv;e_group={group_id};s=1]}}")
|
|
assert metainfo["episode_group"] is None
|
|
|
|
|
|
def test_video_bit_extracted_for_video_title():
|
|
"""测试普通影视标题中的视频位深可单独识别。"""
|
|
meta = MetaInfo(title="The 355 2022 BluRay 1080p DTS-HD MA5.1 X265.10bit-BeiTai")
|
|
assert meta.video_encode == "x265 10bit"
|
|
assert meta.video_bit == "10bit"
|
|
|
|
|
|
def test_special_season_zero_enables_whole_season_resource_parsing():
|
|
"""只有 S00、没有集号的整季标题仍应识别后续编码信息。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title="Demo Show S00 X265 AAC")
|
|
|
|
assert meta.begin_season == 0
|
|
assert meta.video_encode == "x265"
|
|
assert meta.audio_encode == "AAC"
|
|
|
|
|
|
def test_anime_parser_preserves_numeric_special_season_zero():
|
|
"""第三方动漫解析器返回整数 0 时也应保留特别季。"""
|
|
parsed = {
|
|
"anime_title": "Demo Anime",
|
|
"anime_season": 0,
|
|
"episode_number": "1",
|
|
}
|
|
with patch("app.domain.meta.metaanime.anitopy.parse", return_value=parsed):
|
|
meta = MetaAnime(title="Demo Anime S00E01")
|
|
|
|
assert meta.begin_season == 0
|
|
assert meta.begin_episode == 1
|
|
assert meta.type == MediaType.TV
|
|
|
|
parsed["anime_season"] = [0, "1"]
|
|
with patch("app.domain.meta.metaanime.anitopy.parse", return_value=parsed):
|
|
ranged_meta = MetaAnime(title="Demo Anime S00-S01")
|
|
|
|
assert ranged_meta.begin_season == 0
|
|
assert ranged_meta.end_season == 1
|
|
|
|
|
|
def test_anime_parser_ignores_empty_and_invalid_season_values():
|
|
"""第三方动漫季号的空值和非法列表项应按未指定处理且不得抛错。"""
|
|
empty = {
|
|
"anime_title": "Demo Anime",
|
|
"anime_season": "",
|
|
"episode_number": "1",
|
|
}
|
|
invalid_list = {
|
|
"anime_title": "Demo Anime",
|
|
"anime_season": ["", "invalid"],
|
|
"episode_number": "1",
|
|
}
|
|
|
|
with patch("app.domain.meta.metaanime.anitopy.parse", return_value=empty):
|
|
empty_meta = MetaAnime(title="Demo Anime E01")
|
|
with patch("app.domain.meta.metaanime.anitopy.parse", return_value=invalid_list):
|
|
invalid_meta = MetaAnime(title="Demo Anime E01")
|
|
|
|
assert empty_meta.begin_season is None
|
|
assert invalid_meta.begin_season is None
|
|
|
|
|
|
def test_hdr_vivid_effect_extracted_for_video_title():
|
|
"""测试合并写法 HDRVivid 可识别为资源效果。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(
|
|
title="Never-Ending Summer 2026 S01E18-S01E19 2160p WEB-DL 50Fps "
|
|
"HDRVivid H265 10bit AAC-XXWEB"
|
|
)
|
|
|
|
assert meta.resource_type == "WEB-DL"
|
|
assert meta.resource_effect == "HDRVivid"
|
|
assert meta.fps == 50
|
|
|
|
|
|
def test_video_bit_extracted_for_anime_title():
|
|
"""测试动漫标题中的视频位深可单独识别。"""
|
|
meta = MetaInfo(
|
|
title="[云歌字幕组][7月新番][欢迎来到实力至上主义的教室 第二季][01]"
|
|
"[X264 10bit][1080p][简体中文].mp4"
|
|
)
|
|
assert meta.video_encode == "X264"
|
|
assert meta.video_bit == "10bit"
|
|
|
|
|
|
def test_streaming_platform_word_kept_in_movie_title():
|
|
"""测试正式片名中的流媒体平台词不会被预置清理规则移除。"""
|
|
with patch("app.adapters.system.rust.parse_metainfo", return_value=None):
|
|
meta = MetaInfo(title="Amazon Forever 2004 1080p WEB-DL")
|
|
assert meta.name == "Amazon Forever"
|
|
assert meta.year == "2004"
|
|
|
|
|
|
def test_emby_tmdbid_overrides_braced_metainfo_tmdbid():
|
|
"""测试 Emby [tmdbid] 标签保持历史优先级。"""
|
|
title, metainfo = find_metainfo("Movie {[tmdbid=111;type=movies]} [tmdbid=222]")
|
|
assert metainfo["media_source"] == MediaSource.TMDB
|
|
assert metainfo["media_id"] == "222"
|
|
assert "tmdbid" not in metainfo
|
|
assert "[tmdbid=222]" not in title
|
|
|
|
|
|
def test_custom_identifier_uses_source_specific_id_and_returns_unified_identity():
|
|
"""自定义识别词使用专用ID语法，解析结果仍转换为统一身份。"""
|
|
title, metainfo = find_metainfo(
|
|
"Movie {[tmdbid=550;type=movies]}"
|
|
)
|
|
|
|
assert title.strip() == "Movie"
|
|
assert metainfo["media_source"] == MediaSource.TMDB
|
|
assert metainfo["media_id"] == "550"
|
|
assert {"tmdbid", "doubanid", "bangumiid", "anilistid"}.isdisjoint(metainfo)
|
|
|
|
|
|
def test_generic_media_identity_is_not_custom_identifier_syntax():
|
|
"""通用身份字段不得因 Rust 扩展版本差异被自定义识别词解析器接收。"""
|
|
with patch(
|
|
"app.adapters.system.rust.find_metainfo",
|
|
side_effect=AssertionError("通用身份标签必须绕过 Rust 解析器"),
|
|
):
|
|
_, metainfo = find_metainfo(
|
|
"Movie {[media_source=themoviedb;media_id=550;type=movies]}"
|
|
)
|
|
|
|
assert metainfo["media_source"] is None
|
|
assert metainfo["media_id"] is None
|
|
|
|
|
|
def test_metainfopath_auxiliary_chinese_stem_uses_parent_title():
|
|
"""测试辅助文件名合并父目录标题与年份。"""
|
|
path = Path(
|
|
"/Marty Supreme 2025 2160p DoVi HDR Atmos TrueHD 7.1 x265-PbK/简英双语特效.mp4"
|
|
)
|
|
meta = MetaInfoPath(path)
|
|
assert meta.en_name == "Marty Supreme"
|
|
assert meta.year == "2025"
|
|
assert meta.original_name == "Marty Supreme"
|
|
|
|
|
|
def test_metainfopath_chinese_parent_not_replaced_by_auxiliary_rule():
|
|
"""测试纯中文父目录不触发辅助文件名规则。"""
|
|
path = Path("/movies/流浪地球 (2023)/简体中字.mkv")
|
|
meta = MetaInfoPath(path)
|
|
assert meta.cn_name
|
|
assert "简体" in meta.cn_name
|
|
|
|
|
|
def test_metainfopath_cn_title_containing_keyword_not_cleared():
|
|
"""测试中文片名包含辅助关键词子串时不应被清空。"""
|
|
path = Path("/Some Movie 2024/粤语残片.mkv")
|
|
meta = MetaInfoPath(path)
|
|
assert "粤语残片" in meta.cn_name
|
|
|
|
|
|
def test_metainfopath_movie_collection_parent_does_not_override_file_title():
|
|
"""电影合集父目录不应覆盖文件名中更具体的片名与年份。"""
|
|
collection = (
|
|
"/Unraid/Media/MoviePilot/电影/"
|
|
"The.Hunger.Games.Complete.4-Film.Collection.2160p.UHD.Blu-ray."
|
|
"DV.Atmos.TrueHD.7.1.x265-HDH"
|
|
)
|
|
cases = [
|
|
(
|
|
"The.Hunger.Games.2012.2160p.UHD.Blu-ray.DV.Atmos.TrueHD.7.1.x265-HDH.mkv",
|
|
"The Hunger Games",
|
|
"2012",
|
|
),
|
|
(
|
|
"The.Hunger.Games.Catching.Fire.2013.2160p.UHD.Blu-ray.DV.Atmos.TrueHD.7.1.x265-HDH.mkv",
|
|
"The Hunger Games Catching Fire",
|
|
"2013",
|
|
),
|
|
(
|
|
"The.Hunger.Games.Mockingjay.Part.1.2014.2160p.UHD.Blu-ray.DV.Atmos.TrueHD.7.1.x265-HDH.mkv",
|
|
"The Hunger Games Mockingjay Part 1",
|
|
"2014",
|
|
),
|
|
(
|
|
"The.Hunger.Games.Mockingjay.Part.2.2015.2160p.UHD.Blu-ray.DV.Atmos.TrueHD.7.1.x265-HDH.mkv",
|
|
"The Hunger Games Mockingjay Part 2",
|
|
"2015",
|
|
),
|
|
]
|
|
|
|
for file_name, expected_name, expected_year in cases:
|
|
meta = MetaInfoPath(Path(f"{collection}/{file_name}"))
|
|
assert meta.name == expected_name
|
|
assert meta.year == expected_year
|