From bc358fc6d216477ea445cbbf446f35bde022882a Mon Sep 17 00:00:00 2001 From: InfinityPacer <160988576+InfinityPacer@users.noreply.github.com> Date: Tue, 2 Jun 2026 16:32:11 +0800 Subject: [PATCH] =?UTF-8?q?test:=20=E5=A4=84=E7=90=86=20#5877=20review=20?= =?UTF-8?q?=E5=8F=8D=E9=A6=88=20+=20=E6=8F=90=20PR=20=E5=89=8D=E8=B7=91?= =?UTF-8?q?=E5=85=A8=E9=87=8F=E9=97=A8=E7=A6=81=E7=BA=A6=E5=AE=9A=20(#5880?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- AGENTS.md | 2 +- docs/testing.md | 6 +- tests/test_telegram.py | 379 +++++++++++++++++------------------ tests/test_tmdb_recognize.py | 10 +- 4 files changed, 198 insertions(+), 199 deletions(-) diff --git a/AGENTS.md b/AGENTS.md index 014f855e..58f65413 100644 --- a/AGENTS.md +++ b/AGENTS.md @@ -43,7 +43,7 @@ Before executing any task, identify the domain and load the corresponding docume ### Testing * **Primary Reference:** `docs/testing.md` -* **Required Constraints:** pytest is the only runner; `tests/conftest.py` isolates each run to a temporary `CONFIG_DIR`. Tests must not touch the real database, network, or external services (TMDB, LLM catalogs, downloaders, media servers, MP server) — mock at the boundary or replay recorded responses; the bar is zero real outbound traffic. Tests must restore any process-level state they stub (`sys.modules`, singletons, caches, settings). New tests must be pytest-native (function + `assert` + fixtures); do not add new `unittest.TestCase`. Convert existing `TestCase` files to pytest-native opportunistically when you modify them. +* **Required Constraints:** pytest is the only runner; `tests/conftest.py` isolates each run to a temporary `CONFIG_DIR`. Tests must not touch the real database, network, or external services (TMDB, LLM catalogs, downloaders, media servers, MP server) — mock at the boundary or replay recorded responses; the bar is zero real outbound traffic. Tests must restore any process-level state they stub (`sys.modules`, singletons, caches, settings). New tests must be pytest-native (function + `assert` + fixtures); do not add new `unittest.TestCase`. Convert existing `TestCase` files to pytest-native opportunistically when you modify them. Before opening a PR to `v2`, run the full suite locally (`python tests/run.py`) and confirm it is green with zero real network calls; the `.github/workflows/test.yml` gate runs the same suite on every PR/push to `v2`. ### Commands and Development Workflow * **Primary Reference:** `docs/rules/03-commands.md` diff --git a/docs/testing.md b/docs/testing.md index c781e7f9..6682fac3 100644 --- a/docs/testing.md +++ b/docs/testing.md @@ -129,6 +129,8 @@ def test_recognize_prefers_explicit_id(sample_meta, monkeypatch): - **测试间污染（测不准）**：定位被改而未还原的进程级状态（单例 / `lru_cache` / `sys.modules` / 环境变量 / `settings`），按「自隔离」补还原。 - **怀疑用例空过**：用变异验证——临时打断对应生产逻辑（让它返回错误值），跑该用例应**失败**；若仍通过，说明断言没真正覆盖该逻辑。 -## CI +## CI 与 PR -CI 以 pytest 运行 `tests`。建议 CI 与本地复现都用仅安装 `requirements.in` 的干净环境，保证可选扩展、动态模块的存在性与 CI 一致。 +- **门禁**：`.github/workflows/test.yml` 在指向 `v2` 的 `pull_request` / `push` 及手动触发时，用 `python tests/run.py` 跑全量单测。 +- **PR**：`python tests/run.py` 确认全绿、且 socket 探针零真实出站，避免把红的改动推上去空耗门禁。 +- 复现 CI 用仅安装 `requirements.in` 的干净环境（含 pytest 与可选扩展），保证可选扩展、动态模块的存在性与 CI 一致。 diff --git a/tests/test_telegram.py b/tests/test_telegram.py index ac7b9b99..018a75bc 100644 --- a/tests/test_telegram.py +++ b/tests/test_telegram.py @@ -1,8 +1,8 @@ # -*- coding: utf-8 -*- """ -Telegram模块单元测试 +Telegram 模块单元测试（pytest 原生）。 """ -import unittest +import pytest from unittest.mock import MagicMock, patch from app.core.context import MediaInfo, Context, TorrentInfo @@ -11,237 +11,230 @@ from app.modules.telegram.telegram import Telegram from app.schemas.types import MediaType -class TestTelegram(unittest.TestCase): +@pytest.fixture +def telegram(): + """构造 TeleBot 与 ImageHelper 均已打桩的 Telegram 实例。 - def setUp(self): - """测试前准备。 - - 模拟 telebot.TeleBot 以避免真实 API 调用：空 token 会让 Telegram.__init__ 提前返回、 - 属性未初始化导致 send_* 抛错；这里用假 bot 让初始化完整且消息发送走内存桩。 - """ - # 用 addCleanup 注册停桩：即使下方 Telegram(...) 初始化抛错、setUp 未跑完， - # 已启动的 patcher 也会被清理，杜绝 patch 泄漏污染后续用例（tearDown 在 setUp 失败时不执行）。 - telebot_patcher = patch("app.modules.telegram.telegram.TeleBot") - mock_telebot_cls = telebot_patcher.start() - self.addCleanup(telebot_patcher.stop) - self.mock_bot_instance = MagicMock() + 空 token 会让 Telegram.__init__ 提前返回、致 send_* 抛错，故用假 bot 让初始化完整、 + 消息发送走内存桩；ImageHelper 打桩避免 send_medias/send_msg 按 poster_path 真实下载海报 + （否则对 raw.githubusercontent.com 等外链发起真实 HTTP，外部 IO 不可接受且拖慢用例）。 + with 上下文在 fixture 结束时自动停桩，即使实例化失败也不泄漏 patch。 + """ + with patch("app.modules.telegram.telegram.TeleBot") as mock_telebot_cls, \ + patch("app.modules.telegram.telegram.ImageHelper") as mock_image_cls: + bot_instance = MagicMock() # get_me 用于初始化 bot 用户名，需返回带 username 的对象 - self.mock_bot_instance.get_me.return_value = MagicMock(username="test_bot") - mock_telebot_cls.return_value = self.mock_bot_instance - - # send_medias/send_msg 发图时会经 ImageHelper().fetch_image 按 poster_path 真实下载海报， - # 单测必须打桩，否则对 raw.githubusercontent.com 等外链发起真实 HTTP（外部 IO 不可接受且拖慢用例）。 - image_patcher = patch("app.modules.telegram.telegram.ImageHelper") - mock_image_cls = image_patcher.start() - self.addCleanup(image_patcher.stop) + bot_instance.get_me.return_value = MagicMock(username="test_bot") + mock_telebot_cls.return_value = bot_instance mock_image_cls.return_value.fetch_image.return_value = b"fake-image-bytes" - - self.telegram = Telegram(TELEGRAM_TOKEN="fake_token", TELEGRAM_CHAT_ID="fake_chat_id") - - def test_send_msg_success(self): - """测试发送普通消息成功""" - # 调用send_msg方法 - result = self.telegram.send_msg( - title="📥 开始下载\n唐朝诡事录 (2022)S03E31-E32", - text="\n🕒 时间： 2025-11-21 18:14:51\n🎭 类别： 国产剧\n🌐 站点： 天空\n🌟 质量： WEB-DL 2160p\n💾 大小： 1.68G\n⚡️ 促销： 未知\n🚨 H&R： 否\n📛 名称： \nStrange Tales of Tang Dynasty S03E31-E32 2025 2160p WEB-DL DDP5.1 H265-Pure@HDSWEB [唐朝诡事录之长安3 / 唐朝诡事录3 / 唐朝诡事录 第三部 / 唐朝诡事录·长安 / 唐诡3 / Horror Stories of Tang Dynasty Ⅲ / Strange Legend of Tang Dynasty Ⅲ 第3季 第31-32集 | 主演: 杨旭文 杨志刚 郜思雯 [内封简繁英多国软字幕] 【去头尾广告纯享版】[非伪去头] *发现未去净的广告或片头片尾，奖励魔力1W]" - ) - - # 验证返回值 - self.assertTrue(result) - - def test_send_msg_with_longtext(self): - """测试发送长消息""" - result = self.telegram.send_msg( - title="MoviePilot助手", - text="好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？", - ) + yield Telegram(TELEGRAM_TOKEN="fake_token", TELEGRAM_CHAT_ID="fake_chat_id") - def test_send_medias_msg_success(self): - """测试发送媒体列表消息成功""" - # 创建模拟的媒体信息列表 - media1 = MediaInfo() - media1.type = MediaType.MOVIE - media1.title = "测试电影1" - media1.year = "2023" - media1.vote_average = 8.5 - media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - media1.tmdb_id=123123 +def test_send_msg_success(telegram): + """测试发送普通消息成功""" + # 调用send_msg方法 + result = telegram.send_msg( + title="📥 开始下载\n唐朝诡事录 (2022)S03E31-E32", + text="\n🕒 时间： 2025-11-21 18:14:51\n🎭 类别： 国产剧\n🌐 站点： 天空\n🌟 质量： WEB-DL 2160p\n💾 大小： 1.68G\n⚡️ 促销： 未知\n🚨 H&R： 否\n📛 名称： \nStrange Tales of Tang Dynasty S03E31-E32 2025 2160p WEB-DL DDP5.1 H265-Pure@HDSWEB [唐朝诡事录之长安3 / 唐朝诡事录3 / 唐朝诡事录 第三部 / 唐朝诡事录·长安 / 唐诡3 / Horror Stories of Tang Dynasty Ⅲ / Strange Legend of Tang Dynasty Ⅲ 第3季 第31-32集 | 主演: 杨旭文 杨志刚 郜思雯 [内封简繁英多国软字幕] 【去头尾广告纯享版】[非伪去头] *发现未去净的广告或片头片尾，奖励魔力1W]" + ) - media2 = MediaInfo() - media2.type = MediaType.TV - media2.title = "测试电视剧1" - media2.year = "2023" - media2.vote_average = 9.0 - media2.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" + # 验证返回值：send_msg 失败时返回 {"success": False}（非空字典，仅 truthy 检查会漏判），故显式断言 success + assert result and result.get("success") - medias = [media1, media2] +def test_send_msg_with_longtext(telegram): + """测试发送长消息""" + result = telegram.send_msg( + title="MoviePilot助手", + text="好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？好的，为您推荐一些近期热门的电视剧：\n\n* *怪奇物语 (Stranger Things)* - 2016年，TMDB评分8.6\n* *小丑回魂：欢迎来到德里镇* - 2025年，TMDB评分8.0\n* *维京传奇* - 2013年，TMDB评分8.1\n* *地狱客栈* - 2024年，TMDB评分8.7\n* *超人回来了* - 2013年，TMDB评分7.7\n\n还有一些经典剧集也一直很受欢迎：\n\n* *法律与秩序：特殊受害者* - 1999年，TMDB评分7.9\n* *实习医生格蕾* - 2005年，TMDB评分8.2\n* *邪恶力量* - 2005年，TMDB评分8.3\n* *菜鸟老警* - 2018年，TMDB评分8.5\n* *猎魔人* - 2019年，TMDB评分8.0\n* *海军罪案调查处* - 2003年，TMDB评分7.6\n* *塔尔萨之王* - 2022年，TMDB评分8.3\n* *武士生死斗* - 2025年，TMDB评分8.1\n* *嗜血法医* - 2006年，TMDB评分8.2\n* *辛普森一家* - 1989年，TMDB评分8.0\n* *无耻之徒* - 2011年，TMDB评分8.2\n* *绝命毒师* - 2008年，TMDB评分8.9\n* *法律与秩序* - 1990年，TMDB评分7.4\n* *权力的游戏* - 2011年，TMDB评分8.5\n\n您对哪部剧比较感兴趣，或者想了解更多信息呢？", + ) + assert result and result.get("success") - result = self.telegram.send_medias_msg( - medias=medias, - title="推荐媒体列表" - ) - self.assertTrue(result) +def test_send_medias_msg_success(telegram): + """测试发送媒体列表消息成功""" + # 创建模拟的媒体信息列表 + media1 = MediaInfo() + media1.type = MediaType.MOVIE + media1.title = "测试电影1" + media1.year = "2023" + media1.vote_average = 8.5 + media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" + media1.tmdb_id=123123 - def test_send_medias_msg_without_vote_average(self): - """测试发送无评分的媒体列表消息""" - # 创建模拟的媒体信息列表（无评分） - media1 = MediaInfo() - media1.type = MediaType.MOVIE - media1.title = "测试电影1" - media1.year = "2023" - media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - media1.tmdb_id=123123 - medias = [media1] + media2 = MediaInfo() + media2.type = MediaType.TV + media2.title = "测试电视剧1" + media2.year = "2023" + media2.vote_average = 9.0 + media2.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - result = self.telegram.send_medias_msg( - medias=medias, - title="推荐媒体列表" - ) + medias = [media1, media2] - self.assertTrue(result) + result = telegram.send_medias_msg( + medias=medias, + title="推荐媒体列表" + ) - def test_send_medias_msg_with_link_and_buttons(self): - """测试发送带链接和按钮的媒体列表消息""" - media1 = MediaInfo() - media1.type = MediaType.MOVIE - media1.title = r"测试*-|\.电影1" - media1.year = "2023" - media1.vote_average = 8.5 - media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - media1.tmdb_id=123123 + assert result - medias = [media1] +def test_send_medias_msg_without_vote_average(telegram): + """测试发送无评分的媒体列表消息""" + # 创建模拟的媒体信息列表（无评分） + media1 = MediaInfo() + media1.type = MediaType.MOVIE + media1.title = "测试电影1" + media1.year = "2023" + media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" + media1.tmdb_id=123123 + medias = [media1] - buttons = [[ - {"text": "测试按钮", "callback_data": "test_callback"} - ]] + result = telegram.send_medias_msg( + medias=medias, + title="推荐媒体列表" + ) - result = self.telegram.send_medias_msg( - medias=medias, - title="推荐媒体列表", - link="http://example.com", - buttons=buttons - ) + assert result - self.assertTrue(result) +def test_send_medias_msg_with_link_and_buttons(telegram): + """测试发送带链接和按钮的媒体列表消息""" + media1 = MediaInfo() + media1.type = MediaType.MOVIE + media1.title = r"测试*-|\.电影1" + media1.year = "2023" + media1.vote_average = 8.5 + media1.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" + media1.tmdb_id=123123 + + medias = [media1] + + buttons = [[ + {"text": "测试按钮", "callback_data": "test_callback"} + ]] + + result = telegram.send_medias_msg( + medias=medias, + title="推荐媒体列表", + link="http://example.com", + buttons=buttons + ) + + assert result - def test_send_torrents_msg_success(self): - """测试发送种子列表消息成功""" - # 创建模拟的种子信息 - media_info = MediaInfo() - media_info.type = MediaType.TV - media_info.title = "唐朝诡事录" - media_info.year = "2025" - media_info.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" +def test_send_torrents_msg_success(telegram): + """测试发送种子列表消息成功""" + # 创建模拟的种子信息 + media_info = MediaInfo() + media_info.type = MediaType.TV + media_info.title = "唐朝诡事录" + media_info.year = "2025" + media_info.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - torrent_info = TorrentInfo() - torrent_info.site_name = r"测试*-|\.站点" - torrent_info.title = "唐朝诡事录" - torrent_info.description = "唐朝诡事录之长安3 / 唐朝诡事录3 / 唐朝诡事录 第三部 / 唐朝诡事录·长安 / 唐诡3 / Horror Stories of Tang Dynasty Ⅲ / Strange Legend of Tang Dynasty Ⅲ 第3季 第31-32集 | 主演: 杨旭文 杨志刚 郜思雯 [内封简繁英多国软字幕] 【去头尾广告纯享版】[非伪去头] *发现未去净的广告或片头片尾，奖励魔力1W" - torrent_info.page_url = "http://example.com/torrent" - torrent_info.size = 1024 * 1024 * 1024 # 1GB - torrent_info.seeders = 10 - torrent_info.uploadvolumefactor = 1.0 - torrent_info.downloadvolumefactor = 0.0 + torrent_info = TorrentInfo() + torrent_info.site_name = r"测试*-|\.站点" + torrent_info.title = "唐朝诡事录" + torrent_info.description = "唐朝诡事录之长安3 / 唐朝诡事录3 / 唐朝诡事录 第三部 / 唐朝诡事录·长安 / 唐诡3 / Horror Stories of Tang Dynasty Ⅲ / Strange Legend of Tang Dynasty Ⅲ 第3季 第31-32集 | 主演: 杨旭文 杨志刚 郜思雯 [内封简繁英多国软字幕] 【去头尾广告纯享版】[非伪去头] *发现未去净的广告或片头片尾，奖励魔力1W" + torrent_info.page_url = "http://example.com/torrent" + torrent_info.size = 1024 * 1024 * 1024 # 1GB + torrent_info.seeders = 10 + torrent_info.uploadvolumefactor = 1.0 + torrent_info.downloadvolumefactor = 0.0 - meta_info = MetaInfo(title="唐朝诡事录") + meta_info = MetaInfo(title="唐朝诡事录") - context = Context() - context.media_info = media_info - context.torrent_info = torrent_info - context.meta_info = meta_info + context = Context() + context.media_info = media_info + context.torrent_info = torrent_info + context.meta_info = meta_info - torrents = [context] + torrents = [context] - result = self.telegram.send_torrents_msg( - torrents=torrents, - title="种子列表" - ) + result = telegram.send_torrents_msg( + torrents=torrents, + title="种子列表" + ) - self.assertTrue(result) + assert result - def test_send_torrents_msg_with_link_and_buttons(self): - """测试发送带链接和按钮的种子列表消息""" - media_info = MediaInfo() - media_info.type = MediaType.MOVIE - media_info.title = "^测试电影~_测试_" - media_info.year = "2023" - media_info.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" +def test_send_torrents_msg_with_link_and_buttons(telegram): + """测试发送带链接和按钮的种子列表消息""" + media_info = MediaInfo() + media_info.type = MediaType.MOVIE + media_info.title = "^测试电影~_测试_" + media_info.year = "2023" + media_info.poster_path = "https://raw.githubusercontent.com/jxxghp/MoviePilot-Frontend/refs/heads/v2/public/logo.png" - torrent_info = TorrentInfo() - torrent_info.site_name = "^测试~站点_测试_" - torrent_info.title = "测试种子标题" - torrent_info.description = "测试种子描述" - torrent_info.page_url = "http://example.com/torrent" - torrent_info.size = 1024 * 1024 * 1024 # 1GB - torrent_info.seeders = 10 - torrent_info.uploadvolumefactor = 1.0 - torrent_info.downloadvolumefactor = 0.0 + torrent_info = TorrentInfo() + torrent_info.site_name = "^测试~站点_测试_" + torrent_info.title = "测试种子标题" + torrent_info.description = "测试种子描述" + torrent_info.page_url = "http://example.com/torrent" + torrent_info.size = 1024 * 1024 * 1024 # 1GB + torrent_info.seeders = 10 + torrent_info.uploadvolumefactor = 1.0 + torrent_info.downloadvolumefactor = 0.0 - meta_info = MetaInfo(title="测试种子标题") + meta_info = MetaInfo(title="测试种子标题") - context = Context() - context.media_info = media_info - context.torrent_info = torrent_info - context.meta_info = meta_info + context = Context() + context.media_info = media_info + context.torrent_info = torrent_info + context.meta_info = meta_info - torrents = [context] + torrents = [context] - buttons = [[ - {"text": "测试按钮", "callback_data": "test_callback"} - ]] + buttons = [[ + {"text": "测试按钮", "callback_data": "test_callback"} + ]] - result = self.telegram.send_torrents_msg( - torrents=torrents, - title="种子列表", - link="http://example.com", - buttons=buttons - ) + result = telegram.send_torrents_msg( + torrents=torrents, + title="种子列表", + link="http://example.com", + buttons=buttons + ) - self.assertTrue(result) + assert result - def test_send_msg_with_buttons_and_link(self): - """测试发送带按钮和链接的消息""" - buttons = [[ - {"text": "测试按钮", "callback_data": "test_callback"} - ]] +def test_send_msg_with_buttons_and_link(telegram): + """测试发送带按钮和链接的消息""" + buttons = [[ + {"text": "测试按钮", "callback_data": "test_callback"} + ]] - result = self.telegram.send_msg( - title="测试标题", - text="*测试内容*", - link="http://example.com", - buttons=buttons - ) + result = telegram.send_msg( + title="测试标题", + text="*测试内容*", + link="http://example.com", + buttons=buttons + ) - # 验证返回值 - self.assertTrue(result) + # 验证返回值：send_msg 失败时返回 {"success": False}（非空字典），故显式断言 success + assert result and result.get("success") - def test_send_msg_with_url_buttons(self): - """测试发送带URL按钮的消息""" - buttons = [[ - {"text": "URL按钮", "url": "http://example.com"} - ]] +def test_send_msg_with_url_buttons(telegram): + """测试发送带URL按钮的消息""" + buttons = [[ + {"text": "URL按钮", "url": "http://example.com"} + ]] - result = self.telegram.send_msg( - title="测试标题", - text="测试内容", - buttons=buttons - ) + result = telegram.send_msg( + title="测试标题", + text="测试内容", + buttons=buttons + ) - # 验证返回值 - self.assertTrue(result) + # 验证返回值：send_msg 失败时返回 {"success": False}（非空字典），故显式断言 success + assert result and result.get("success") - def test_send_msg_markdown_escaping(self): - """测试Markdown特殊字符转义""" - result = self.telegram.send_msg( - title="测试标题", - text="_测试_||内容||" - ) +def test_send_msg_markdown_escaping(telegram): + """测试Markdown特殊字符转义""" + result = telegram.send_msg( + title="测试标题", + text="_测试_||内容||" + ) - # 验证返回值 - self.assertTrue(result) + # 验证返回值：send_msg 失败时返回 {"success": False}（非空字典），故显式断言 success + assert result and result.get("success") diff --git a/tests/test_tmdb_recognize.py b/tests/test_tmdb_recognize.py index 6a5c3563..6bf6a668 100644 --- a/tests/test_tmdb_recognize.py +++ b/tests/test_tmdb_recognize.py @@ -75,13 +75,17 @@ def setUpModule(): patch.object(MoviePilotServerHelper, "report_recognize_share", new=MagicMock(return_value=None)), patch.object(MoviePilotServerHelper, "query_recognize_share", new=MagicMock(return_value=None)), ]) + started = [] try: for patcher in _PATCHERS: patcher.start() + started.append(patcher) except Exception: - # 任一 patcher.start() 失败（如重构致类/方法改名）时回滚已启动的桩并清空， - # 避免半启动状态泄漏到其它测试模块。 - tearDownModule() + # 仅回滚已成功启动的桩：对未启动的 patcher 调用 stop() 会抛 RuntimeError， + # 既掩盖原始启动异常又中断清理；记录 started 可精确回滚、避免半启动状态泄漏。 + for patcher in started: + patcher.stop() + _PATCHERS.clear() raise