From b10d282980950ba70c8e1d5171d7cf4dd04b0b63 Mon Sep 17 00:00:00 2001
From: hub-gif <2487812171@qq.com>
Date: Thu, 16 Apr 2026 10:52:04 +0800
Subject: [PATCH] =?UTF-8?q?test(pipeline):=20=E5=8D=95=E7=BB=86=E7=B1=BB?=
 =?UTF-8?q?=E8=AF=84=E8=AE=BA=E8=A6=81=E7=82=B9=E5=BD=92=E7=BA=B3=20LLM=20?=
 =?UTF-8?q?=E7=83=9F=E6=B5=8B=EF=BC=88mock=20=E7=BD=91=E5=85=B3=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

构造仅「饼干」矩阵细类的 build_comment_groups_llm_payload，断言 generate_comment_group_summaries_llm 与 chunked 单组时各调用一次网关，且 user JSON 含 keyword 与 group。

Made-with: Cursor
---
 .../tests/test_comment_group_summaries_llm.py | 96 +++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 backend/pipeline/tests/test_comment_group_summaries_llm.py

diff --git a/backend/pipeline/tests/test_comment_group_summaries_llm.py b/backend/pipeline/tests/test_comment_group_summaries_llm.py
new file mode 100644
index 0000000..f37f4c5
--- /dev/null
+++ b/backend/pipeline/tests/test_comment_group_summaries_llm.py
@@ -0,0 +1,96 @@
+"""§8 末「细类评论与关注词要点归纳」大模型：单细类烟测（mock 网关，不调真实 API）。"""
+from __future__ import annotations
+
+import json
+import sys
+from pathlib import Path
+from unittest.mock import patch
+
+from django.conf import settings
+from django.test import SimpleTestCase
+
+from pipeline.llm.generate import (
+    generate_comment_group_summaries_llm,
+    generate_comment_group_summaries_llm_chunked,
+)
+
+
+def _load_jcr():
+    root = Path(settings.CRAWLER_JD_ROOT).resolve()
+    if str(root) not in sys.path:
+        sys.path.insert(0, str(root))
+    import jd_competitor_report as jcr  # noqa: WPS433
+
+    return jcr
+
+
+def _payload_single_category_biscuit() -> tuple[list[dict], str, str]:
+    """合并表 + 评价仅对应一个矩阵细类「饼干」。"""
+    jcr = _load_jcr()
+    sku_h = "SKU(skuId)"
+    title_h = "标题(wareName)"
+    merged = [
+        {
+            sku_h: "111",
+            "detail_category_path": "食品饮料 > 休闲食品 > 饼干 > 粗粮饼干",
+            title_h: "测试饼干",
+            "detail_shop_name": "测试店",
+        },
+    ]
+    comments = [
+        {
+            "sku": "111",
+            "tagCommentContent": "口感不错，低GI很适合控糖",
+        },
+    ]
+    fb = jcr._consumer_feedback_by_matrix_group(
+        merged_rows=merged,
+        comment_rows=comments,
+        sku_header=sku_h,
+    )
+    pl = jcr.build_comment_groups_llm_payload(
+        feedback_groups=fb,
+        focus_words=jcr.COMMENT_FOCUS_WORDS,
+        merged_rows=merged,
+        sku_header=sku_h,
+        title_h=title_h,
+    )
+    return pl, sku_h, title_h
+
+
+class CommentGroupSummariesLlmTests(SimpleTestCase):
+    def test_single_matrix_group_payload_and_llm_smoke(self) -> None:
+        pl, _, _ = _payload_single_category_biscuit()
+        self.assertEqual(len(pl), 1)
+        self.assertEqual(pl[0].get("group"), "饼干")
+        self.assertIn("sample_text_snippets", pl[0])
+        self.assertTrue(any("饼干" in s for s in pl[0]["sample_text_snippets"]))
+
+        with patch(
+            "pipeline.llm.generate._call_llm",
+            return_value="#### 饼干\n单测归纳段落。",
+        ) as mock_llm:
+            out = generate_comment_group_summaries_llm(pl, keyword="低GI测试")
+
+        mock_llm.assert_called_once()
+        _sys, user = mock_llm.call_args[0]
+        self.assertIn("细类评论与关注词要点归纳", user)
+        self.assertIn("低GI测试", user)
+        raw = user.split("正文（Markdown）。\n\n", 1)[-1]
+        data = json.loads(raw.strip())
+        self.assertEqual(data["keyword"], "低GI测试")
+        self.assertEqual(len(data["groups"]), 1)
+        self.assertEqual(data["groups"][0]["group"], "饼干")
+        self.assertIn("饼干", out)
+
+    def test_chunked_single_category_one_gateway_call(self) -> None:
+        """与生产「按细类拆分」一致：仅一个细类时只请求一次。"""
+        pl, _, _ = _payload_single_category_biscuit()
+        with patch(
+            "pipeline.llm.generate._call_llm",
+            return_value="#### 饼干\nchunked。",
+        ) as mock_llm:
+            out = generate_comment_group_summaries_llm_chunked(pl, keyword="低GI测试")
+
+        self.assertEqual(mock_llm.call_count, 1)
+        self.assertIn("饼干", out)