mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
jxxghp
7e851dbfa7
- ChainBase 拆分为 RecognitionMixin/MessageProcessingMixin/NotificationMixin - TransferChain 拆分为 7 个功能 mixin（_mixins.py），SubscribeChain 音乐订阅域拆出 _music.py - 斜杠命令交互四件套收敛为 InteractionChainMixin 委托，会话管理器移至 application 层，chain 层不再 re-export - 模块基础类收敛到 app/modules/_base（notification/mediaserver 语义重命名） - 清理 app/chain/__init__.py 24 个未使用导入，修正 49 处测试 patch 目标到实际命名空间 - 兼容层 legacy 符号不再并入 __all__，根治 schemas 初始化反向拉起 application.transfer 的循环导入 - 修复 bangumi 集数为字符串时 set_bangumi_info 抛 TypeError - 新增重复代码等架构门禁测试；capability 清单校验排除下划线内部目录
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import sys
|
|
from types import ModuleType
|
|
from unittest.mock import patch
|
|
|
|
sys.modules.setdefault("qbittorrentapi", ModuleType("qbittorrentapi"))
|
|
setattr(sys.modules["qbittorrentapi"], "TorrentFilesList", list)
|
|
sys.modules.setdefault("transmission_rpc", ModuleType("transmission_rpc"))
|
|
setattr(sys.modules["transmission_rpc"], "File", object)
|
|
sys.modules.setdefault("psutil", ModuleType("psutil"))
|
|
|
|
from app.chain import ChainBase
|
|
from app.domain.context import MediaInfo
|
|
from app.domain.meta.metabase import MetaBase
|
|
from app.schemas.types import MediaSource, MediaType
|
|
|
|
|
|
def test_recognize_media_uses_meta_episode_group():
|
|
"""
|
|
识别链未显式传 episode_group 时，应沿用元数据中识别出的剧集组。
|
|
"""
|
|
group_id = "5ad0ec240e0a26303f00d84d"
|
|
chain = ChainBase()
|
|
meta = MetaBase("测试剧集")
|
|
meta.name = "测试剧集"
|
|
meta.type = MediaType.TV
|
|
meta.episode_group = group_id
|
|
mediainfo = MediaInfo(
|
|
media_source=MediaSource.TMDB,
|
|
media_id="100",
|
|
title="测试剧集",
|
|
year="2024",
|
|
tmdb_id=100,
|
|
type=MediaType.TV,
|
|
)
|
|
|
|
with patch.object(chain, "run_module", return_value=mediainfo) as run_module, patch(
|
|
"app.chain._recognition.MoviePilotServerHelper.report_recognize_share",
|
|
return_value=True,
|
|
), patch("app.chain._recognition.MoviePilotServerHelper.query_recognize_share") as query_mock:
|
|
result = chain.recognize_media(meta=meta, cache=False)
|
|
|
|
assert result is mediainfo
|
|
assert run_module.call_args.kwargs["episode_group"] == group_id
|
|
query_mock.assert_not_called()
|