fix(search): include plugin resources without indexers (#6399)

tests/test_search_ai_recommend.py

@@ -55,6 +55,12 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
         chain.save_cache = lambda _cache, _filename: None
         chain.remove_cache = lambda _filename: None
         chain.get_search_page_size = IndexerModule.get_search_page_size
+        chain.search_plugin_torrents = lambda **_kwargs: []
+
+        async def no_plugin_results(**_kwargs):
+            return []
+
+        chain.async_search_plugin_torrents = no_plugin_results
         return chain
 
     async def test_start_recommend_task_restores_original_indices(self):
@@ -185,7 +191,7 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
                 for index in range(count)
             ]
 
-        chain.search_torrents = search_torrents
+        chain.search_site_torrents = search_torrents
 
         with (
             patch.object(settings, "SEARCH_RESOURCE_PAGES", 4, create=True),
@@ -231,7 +237,7 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
                 for index in range(count)
             ]
 
-        chain.search_torrents = search_torrents
+        chain.search_site_torrents = search_torrents
 
         with (
             patch.object(settings, "SEARCH_RESOURCE_PAGES", 3, create=True),
@@ -277,7 +283,7 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
                 for index in range(count)
             ]
 
-        chain.search_torrents = search_torrents
+        chain.search_site_torrents = search_torrents
 
         with (
             patch.object(settings, "SEARCH_RESOURCE_PAGES", 3, create=True),
@@ -342,7 +348,7 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
                 for index in range(count)
             ]
 
-        chain.async_search_torrents = async_search_torrents
+        chain.async_search_site_torrents = async_search_torrents
 
         with (
             patch.object(settings, "SEARCH_RESOURCE_PAGES", 4, create=True),
@@ -388,7 +394,7 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase):
                 for index in range(count)
             ]
 
-        chain.async_search_torrents = async_search_torrents
+        chain.async_search_site_torrents = async_search_torrents
 
         with (
             patch.object(settings, "SEARCH_RESOURCE_PAGES", 3, create=True),

tests/test_search_plugin_sources.py

@@ -0,0 +1,124 @@
+import asyncio
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, patch
+
+from app.chain.search import SearchChain
+from app.modules.indexer import IndexerModule
+from app.runtime.config import settings
+
+
+def make_chain() -> SearchChain:
+    """构造不触发完整启动流程的搜索链。"""
+    chain = object.__new__(SearchChain)
+    chain.get_search_page_size = IndexerModule.get_search_page_size
+    return chain
+
+
+def test_search_returns_plugin_results_without_indexer_sites():
+    """未配置 PT 站点时，插件资源仍应进入原生资源搜索。"""
+    chain = make_chain()
+    plugin_item = SimpleNamespace(title="Plugin Result", description="")
+    calls = []
+    chain.search_plugin_torrents = lambda **kwargs: calls.append(kwargs) or [plugin_item]
+
+    with (
+        patch("app.chain.search.get_configured_system_config") as system_config_oper,
+        patch("app.chain.search.SitesHelper") as sites_helper,
+    ):
+        system_config_oper.return_value.get.return_value = []
+        sites_helper.return_value.get_indexers.return_value = []
+        results = chain._SearchChain__search_all_sites(keyword="keyword")
+
+    assert results == [plugin_item]
+    assert len(calls) == 1
+    assert calls[0]["keyword"] == "keyword"
+
+
+def test_search_invokes_plugin_once_with_multiple_indexers():
+    """多个 PT 站点不应导致插件资源源被重复搜索。"""
+    chain = make_chain()
+    plugin_calls = []
+    site_calls = []
+    chain.search_plugin_torrents = lambda **kwargs: plugin_calls.append(kwargs) or [
+        SimpleNamespace(title="Plugin Result", description="")
+    ]
+    chain.search_site_torrents = lambda **kwargs: site_calls.append(kwargs) or [
+        SimpleNamespace(title=f"Site {kwargs['site']['id']}", description="")
+    ]
+
+    with (
+        patch.object(settings, "SEARCH_RESOURCE_PAGES", 1, create=True),
+        patch("app.chain.search.get_configured_system_config") as system_config_oper,
+        patch("app.chain.search.SitesHelper") as sites_helper,
+        patch("app.chain.search.ProgressHelper") as progress_helper,
+    ):
+        system_config_oper.return_value.get.return_value = [1, 2]
+        sites_helper.return_value.get_indexers.return_value = [
+            {"id": 1, "name": "站点一"},
+            {"id": 2, "name": "站点二"},
+        ]
+        progress_helper.return_value = SimpleNamespace(
+            start=lambda: None, update=lambda **_kwargs: None, end=lambda: None
+        )
+        results = chain._SearchChain__search_all_sites(keyword="keyword")
+
+    assert len(plugin_calls) == 1
+    assert sorted(call["site"]["id"] for call in site_calls) == [1, 2]
+    assert len(results) == 3
+
+
+def test_async_search_returns_plugin_results_without_indexers():
+    """异步搜索应支持只有插件资源源的部署方式。"""
+    chain = make_chain()
+    plugin_item = SimpleNamespace(title="Plugin Result", description="")
+    calls = []
+
+    async def plugin_search(**kwargs):
+        calls.append(kwargs)
+        return [plugin_item]
+
+    chain.async_search_plugin_torrents = plugin_search
+    async def run_search():
+        with (
+            patch("app.chain.search.get_configured_system_config") as system_config_oper,
+            patch("app.chain.search.SitesHelper") as sites_helper,
+        ):
+            system_config_oper.return_value.get.return_value = []
+            sites_helper.return_value.async_get_indexers = AsyncMock(return_value=[])
+            return await chain._SearchChain__async_search_all_sites(keyword="keyword")
+
+    results = asyncio.run(run_search())
+
+    assert results == [plugin_item]
+    assert len(calls) == 1
+
+
+def test_async_search_stream_emits_plugin_results_once_without_indexers():
+    """流式搜索完成事件不应重复发送插件资源。"""
+    chain = make_chain()
+    plugin_item = SimpleNamespace(title="Plugin Result", description="")
+
+    async def plugin_search(**_kwargs):
+        return [plugin_item]
+
+    chain.async_search_plugin_torrents = plugin_search
+    async def collect_events():
+        with (
+            patch("app.chain.search.get_configured_system_config") as system_config_oper,
+            patch("app.chain.search.SitesHelper") as sites_helper,
+        ):
+            system_config_oper.return_value.get.return_value = []
+            sites_helper.return_value.async_get_indexers = AsyncMock(return_value=[])
+            return [
+                event
+                async for event in chain._SearchChain__async_search_all_sites_stream(
+                    keyword="keyword"
+                )
+            ]
+
+    events = asyncio.run(collect_events())
+
+    assert [event["type"] for event in events] == ["append", "done"]
+    assert events[0]["items"] == [plugin_item]
+    assert events[1]["items"] == []
+    assert events[1]["total_items"] == 1