feat: support torrent description filtering (#6172)

tests/test_agent_get_search_results_tool.py

@@ -0,0 +1,100 @@
+import asyncio
+import json
+from unittest.mock import AsyncMock, MagicMock, patch
+
+from app.agent.tools.impl._torrent_search_utils import simplify_search_result
+from app.agent.tools.impl.get_search_results import GetSearchResultsTool
+from app.core.context import Context, TorrentInfo
+
+
+def _build_context(
+    title: str,
+    *,
+    description: str = None,
+    labels: list = None,
+    index: int = 1,
+) -> Context:
+    """构造种子搜索结果上下文。"""
+    return Context(
+        torrent_info=TorrentInfo(
+            title=title,
+            description=description,
+            labels=labels or [],
+            enclosure=f"https://example.com/download/{index}",
+            size=1024,
+            seeders=index,
+            site_name="测试站点",
+        )
+    )
+
+
+def _run_tool(items: list[Context], **kwargs) -> str:
+    """使用指定缓存结果运行搜索结果工具。"""
+    search_chain = MagicMock()
+    search_chain.async_last_search_results = AsyncMock(return_value=items)
+    with patch(
+        "app.agent.tools.impl.get_search_results.SearchChain",
+        return_value=search_chain,
+    ):
+        return asyncio.run(
+            GetSearchResultsTool(session_id="session-1", user_id="10001").run(
+                **kwargs
+            )
+        )
+
+
+def test_simplify_search_result_only_includes_description_when_requested():
+    """精简结果应按参数控制简介输出，避免默认增加上下文长度。"""
+    context = _build_context("Movie.2026.1080p", description="简繁特效字幕")
+
+    default_result = simplify_search_result(context, 1)
+    detailed_result = simplify_search_result(context, 1, include_description=True)
+
+    assert "description" not in default_result["torrent_info"]
+    assert detailed_result["torrent_info"]["description"] == "简繁特效字幕"
+
+
+def test_content_pattern_matches_title_description_and_labels():
+    """内容正则应联合匹配标题、简介和标签，并可返回命中的简介。"""
+    items = [
+        _build_context("Movie.Special.Effect.2026", description="普通字幕", index=1),
+        _build_context("Movie.2026.1080p", description="简繁特效字幕", index=2),
+        _build_context("Movie.2026.2160p", description="国语音轨", labels=["官译"], index=3),
+        _build_context("Movie.2026.WEB-DL", description="英文字幕", index=4),
+    ]
+
+    result = _run_tool(
+        items,
+        content_pattern="Special.Effect|特效字幕|官译",
+        include_description=True,
+    )
+    payload = json.loads(result)
+
+    assert payload["total_count"] == 3
+    assert [item["torrent_info"]["description"] for item in payload["results"]] == [
+        "普通字幕",
+        "简繁特效字幕",
+        "国语音轨",
+    ]
+
+
+def test_title_pattern_keeps_title_only_matching_semantics():
+    """标题正则不应因新增内容筛选而匹配简介或标签。"""
+    items = [
+        _build_context("Movie.特效.2026", description="普通字幕", index=1),
+        _build_context("Movie.2026.1080p", description="简繁特效字幕", index=2),
+        _build_context("Movie.2026.2160p", labels=["特效"], index=3),
+    ]
+
+    result = _run_tool(items, title_pattern="特效", include_description=True)
+    payload = json.loads(result)
+
+    assert payload["total_count"] == 1
+    assert payload["results"][0]["torrent_info"]["title"] == "Movie.特效.2026"
+
+
+def test_invalid_content_pattern_returns_validation_message():
+    """非法内容正则应返回明确错误，不进入搜索结果筛选。"""
+    result = _run_tool([_build_context("Movie.2026")], content_pattern="[")
+
+    assert result.startswith("正则表达式格式错误:")