fix(search): disambiguate yearless same-title resources

2026-09-05 07:27:15 +08:00 · 2026-08-22 11:35:01 +08:00
parent 5d98f397cb
commit 2d982f08fa
6 changed files with 435 additions and 12 deletions
@@ -0,0 +1,104 @@
+from unittest.mock import Mock
+
+from app.chain import search as search_module
+from app.chain.search import SearchChain
+from app.domain.context import MediaInfo, TorrentInfo
+from app.schemas.types import MediaSource, MediaType
+
+
+def test_exact_search_rejects_no_year_alias_recognized_as_different_work(monkeypatch):
+    """精确搜索中的无年份别名候选识别为不同作品时应拒绝。"""
+    target = MediaInfo(
+        media_source=MediaSource.TMDB,
+        media_id="236356",
+        tmdb_id=236356,
+        title="家族计划",
+        original_title="가족계획",
+        names=["Family Matters"],
+        type=MediaType.TV,
+        year="2024",
+        original_language="ko",
+        season_years={1: "2024"},
+    )
+    candidate = MediaInfo(
+        media_source=MediaSource.TMDB,
+        media_id="30161",
+        tmdb_id=30161,
+        title="Family Matters",
+        original_title="Family Matters",
+        type=MediaType.TV,
+        year="2008",
+        original_language="en",
+    )
+    torrent = TorrentInfo(
+        title="Family Matters S01 1080p WEBRip DD2.0 x264-TrollHD",
+        site_name="测试站点",
+        category=MediaType.TV.value,
+    )
+    media_chain = Mock()
+    media_chain.recognize_by_meta.return_value = candidate
+    monkeypatch.setattr(search_module, "MediaChain", Mock(return_value=media_chain))
+    monkeypatch.setattr(
+        search_module.TorrentHelper,
+        "sort_torrents",
+        staticmethod(lambda contexts: contexts),
+    )
+    chain = object.__new__(SearchChain)
+
+    contexts = chain._SearchChain__parse_result(
+        torrents=[torrent],
+        mediainfo=target,
+        rule_groups=[],
+    )
+
+    assert contexts == []
+    media_chain.recognize_by_meta.assert_called_once()
+
+
+def test_exact_search_reuses_disambiguation_for_same_parsed_title(monkeypatch):
+    """同一解析标题的多条资源应复用一次候选识别，避免重复外部查询。"""
+    target = MediaInfo(
+        media_source=MediaSource.TMDB,
+        media_id="236356",
+        tmdb_id=236356,
+        title="家族计划",
+        original_title="가족계획",
+        names=["Family Matters"],
+        type=MediaType.TV,
+        year="2024",
+        season_years={1: "2024"},
+    )
+    candidate = MediaInfo(
+        media_source=MediaSource.TMDB,
+        media_id="236356",
+        tmdb_id=236356,
+        title="家族计划",
+        type=MediaType.TV,
+        year="2024",
+    )
+    torrents = [
+        TorrentInfo(
+            title=f"Family Matters S01 1080p WEB-DL {group}",
+            site_name=f"测试站点{index}",
+            category=MediaType.TV.value,
+        )
+        for index, group in enumerate(("GROUP-A", "GROUP-B"), start=1)
+    ]
+    media_chain = Mock()
+    media_chain.recognize_by_meta.return_value = candidate
+    monkeypatch.setattr(search_module, "MediaChain", Mock(return_value=media_chain))
+    monkeypatch.setattr(
+        search_module.TorrentHelper,
+        "sort_torrents",
+        staticmethod(lambda contexts: contexts),
+    )
+    chain = object.__new__(SearchChain)
+
+    contexts = chain._SearchChain__parse_result(
+        torrents=torrents,
+        mediainfo=target,
+        rule_groups=[],
+    )
+
+    assert len(contexts) == 2
+    media_chain.recognize_by_meta.assert_called_once()