feat: message-api for mission via wecomm agent

2026-05-14 09:30:38 +08:00
parent 9359013b4c
commit f409f2e6ee
38 changed files with 2085 additions and 49 deletions
--- a/notifier/tests.py
+++ b/notifier/tests.py
@@ -1,4 +1,5 @@
-from unittest.mock import patch
+import json
+from unittest.mock import MagicMock, patch

 from django.test import TestCase
 from django.utils import timezone
@@ -52,6 +53,103 @@ class NotifierServiceTestCase(TestCase):
        self.assertIn("任务ID：12", content)
        self.assertIn("创建人：张三", content)

+    def test_render_notification_content_with_structured_description_template(self):
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="结构化描述通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_structured_description_text",
+            config={"agent_id": 1000007},
+        )
+
+        content = render_notification_content(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "creator_name": "张三",
+                "category_name": "通用",
+                "description": "原始描述",
+                "parsed_description_title": "标题一\n标题二",
+                "parsed_description_body": "正文一\n正文二",
+                "parsed_description_url": "https://example.com/detail",
+                "parsed_description_image_url": "https://images.yuwen.cloud/abc.jpg",
+            },
+        )
+
+        rendered_payload = json.loads(content)
+        self.assertEqual(rendered_payload["msgtype"], "text")
+        self.assertIn("标题：标题一\n标题二", rendered_payload["content"])
+        self.assertIn("正文：正文一\n正文二", rendered_payload["content"])
+        self.assertIn("链接：https://example.com/detail", rendered_payload["content"])
+
+    def test_render_notification_content_with_structured_description_news_template(self):
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="结构化描述图文通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_structured_description_news",
+            config={
+                "agent_ids": [1000007],
+                "image_url": "https://cdn.example.com/covers/mission-news.png",
+                "url": "https://erp.example.com/missions/fallback",
+            },
+        )
+
+        content = render_notification_content(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "creator_name": "张三",
+                "category_name": "通用",
+                "description": "原始描述",
+                "parsed_description_title": "标题一\n标题二",
+                "parsed_description_body": "正文一\n正文二",
+                "parsed_description_url": "https://example.com/detail",
+                "parsed_description_image_url": "https://images.yuwen.cloud/abc.jpg",
+            },
+        )
+
+        rendered_payload = json.loads(content)
+        self.assertEqual(rendered_payload["msgtype"], "news")
+        self.assertEqual(rendered_payload["title"], "标题一\n标题二")
+        self.assertEqual(rendered_payload["description"], "正文一\n正文二")
+        self.assertEqual(rendered_payload["url"], "https://example.com/detail")
+        self.assertEqual(rendered_payload["image_url"], "https://images.yuwen.cloud/abc.jpg")
+
+    def test_render_notification_content_with_structured_description_news_template_falls_back_to_default_image(self):
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="结构化描述图文通知-默认图",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_structured_description_news",
+            config={
+                "agent_ids": [1000007],
+                "url": "https://erp.example.com/missions/fallback",
+            },
+        )
+
+        with self.settings(MESSAGE_API_DEFAULT_NEWS_IMAGE_URL="https://via.placeholder.com/640x360.png?text=No+Image"):
+            content = render_notification_content(
+                notifier=notifier,
+                payload={
+                    "mission_id": 12,
+                    "creator_name": "张三",
+                    "category_name": "通用",
+                    "description": "原始描述",
+                    "parsed_description_title": "标题一\n标题二",
+                    "parsed_description_body": "正文一\n正文二",
+                    "parsed_description_url": "",
+                    "parsed_description_image_url": "",
+                },
+            )
+
+        rendered_payload = json.loads(content)
+        self.assertEqual(rendered_payload["url"], "https://erp.example.com/missions/fallback")
+        self.assertEqual(
+            rendered_payload["image_url"],
+            "https://via.placeholder.com/640x360.png?text=No+Image",
+        )
+
    @patch("notifier.backends.send_wecom_webhook_message")
    def test_send_notification_with_notifier_uses_wecom_backend(self, mock_send):
        mock_send.return_value.ok = True
@@ -252,3 +350,262 @@ class NotifierServiceTestCase(TestCase):

        self.assertEqual(result["sent_count"], 0)
        mock_dispatch.assert_not_called()
+
+    @patch("notifier.tasks.dispatch_notification_event")
+    def test_notify_unreplied_missions_task_applies_category_payload_processor(self, mock_dispatch):
+        mock_dispatch.return_value = [{"status": "sent"}]
+        self.general_category.payload_processor = "structured_description_v1"
+        self.general_category.save(update_fields=["payload_processor", "updated_at"])
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="未回复提醒通知器",
+            channel=NotifierChannelEnum.WECOM_WEBHOOK,
+            template_key="mission_unreplied",
+            config={"key": "unreplied-key"},
+        )
+        NotifierRoute.objects.create(
+            merchant=self.merchant,
+            notifier=notifier,
+            event_key=NotificationEventKeyEnum.MISSION_UNREPLIED,
+        )
+        mission = Mission.objects.create(
+            merchant=self.merchant,
+            category=self.general_category,
+            creator=self.creator,
+            description=(
+                "系统首行\n标题一\n款式图:https://images.yuwen.cloud/abc.jpg\n标题二\n\n正文一\n正文二\n"
+                "手机端链接: https://example.com/detail"
+            ),
+            notify_if_unreplied=True,
+            unreplied_notify_interval_minutes=1,
+        )
+        mission.created_at = timezone.now() - timezone.timedelta(minutes=3)
+        mission.save(update_fields=["created_at", "updated_at"])
+
+        result = notify_unreplied_missions_task.run(limit=10)
+
+        self.assertEqual(result["sent_count"], 1)
+        payload = mock_dispatch.call_args.kwargs["payload"]
+        self.assertEqual(payload["payload_processor"], "structured_description_v1")
+        self.assertEqual(payload["parsed_description_title"], "标题一\n标题二")
+        self.assertEqual(payload["parsed_description_body"], "正文一\n正文二")
+        self.assertEqual(payload["parsed_description_url"], "https://example.com/detail")
+        self.assertEqual(payload["parsed_description_image_url"], "https://images.yuwen.cloud/abc.jpg")
+
+    @patch("notifier.backends.send_message_api_text_message")
+    def test_send_notification_with_notifier_uses_message_api_text_backend(self, mock_send):
+        mock_send.return_value.errcode = 0
+        mock_send.return_value.errmsg = "ok"
+        mock_send.return_value.raw = {"errcode": 0, "errmsg": "ok"}
+
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="任务创建消息 API 通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_created",
+            config={"agent_id": 1000007},
+        )
+
+        result = send_notification_with_notifier(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "created_by_name": "张三",
+                "creator_name": "张三",
+                "category_name": "通用",
+                "is_urgent": True,
+                "participant_names_display": "无",
+                "description": "检查打印质量",
+            },
+        )
+
+        self.assertEqual(result["status"], "sent")
+        self.assertEqual(result["channel"], NotifierChannelEnum.MESSAGE_API)
+        self.assertEqual(result["msgtype"], "text")
+        self.assertEqual(mock_send.call_count, 1)
+        self.assertEqual(mock_send.call_args.kwargs["agent_id"], 1000007)
+        self.assertEqual(mock_send.call_args.kwargs["user_ids"], None)
+        self.assertEqual(mock_send.call_args.kwargs["timeout_seconds"], 10.0)
+        self.assertIn("任务已创建", mock_send.call_args.kwargs["content"])
+
+    @patch("notifier.backends.send_message_api_text_message")
+    def test_send_notification_with_notifier_uses_structured_description_template(self, mock_send):
+        mock_send.return_value.errcode = 0
+        mock_send.return_value.errmsg = "ok"
+        mock_send.return_value.raw = {"errcode": 0, "errmsg": "ok"}
+
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="结构化描述消息 API 通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_structured_description_text",
+            config={"agent_id": 1000007},
+        )
+
+        result = send_notification_with_notifier(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "creator_name": "张三",
+                "category_name": "通用",
+                "description": "原始描述",
+                "parsed_description_title": "标题一\n标题二",
+                "parsed_description_body": "正文一\n正文二",
+                "parsed_description_url": "https://example.com/detail",
+                "parsed_description_image_url": "https://images.yuwen.cloud/abc.jpg",
+            },
+        )
+
+        self.assertEqual(result["status"], "sent")
+        self.assertEqual(result["channel"], NotifierChannelEnum.MESSAGE_API)
+        self.assertEqual(result["msgtype"], "text")
+        self.assertEqual(mock_send.call_count, 1)
+        self.assertIn("标题：标题一\n标题二", mock_send.call_args.kwargs["content"])
+        self.assertIn("正文：正文一\n正文二", mock_send.call_args.kwargs["content"])
+        self.assertIn("链接：https://example.com/detail", mock_send.call_args.kwargs["content"])
+
+    @patch("notifier.backends.send_message_api_news_to_agents")
+    def test_send_notification_with_notifier_uses_structured_description_news_template(self, mock_send):
+        mock_send.return_value = [
+            {
+                "agent_id": 1000007,
+                "ok": True,
+                "response": {"errcode": 0, "errmsg": "ok"},
+                "error": None,
+            }
+        ]
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="结构化描述图文消息 API 通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="mission_structured_description_news",
+            config={
+                "agent_ids": [1000007],
+                "image_url": "https://cdn.example.com/covers/mission-news.png",
+            },
+        )
+
+        result = send_notification_with_notifier(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "description": "原始描述",
+                "parsed_description_title": "标题一\n标题二",
+                "parsed_description_body": "正文一\n正文二",
+                "parsed_description_url": "https://example.com/detail",
+                "parsed_description_image_url": "https://images.yuwen.cloud/abc.jpg",
+            },
+        )
+
+        self.assertEqual(result["status"], "sent")
+        self.assertEqual(result["msgtype"], "news")
+        mock_send.assert_called_once()
+        self.assertEqual(mock_send.call_args.kwargs["agent_ids"], [1000007])
+        self.assertEqual(mock_send.call_args.kwargs["title"], "标题一\n标题二")
+        self.assertEqual(mock_send.call_args.kwargs["description"], "正文一\n正文二")
+        self.assertEqual(mock_send.call_args.kwargs["url"], "https://example.com/detail")
+        self.assertEqual(
+            mock_send.call_args.kwargs["image_url"],
+            "https://images.yuwen.cloud/abc.jpg",
+        )
+
+    @patch("notifier.backends.send_message_api_news_to_agents")
+    def test_send_notification_with_notifier_uses_message_api_news_backend(self, mock_send):
+        mock_send.return_value = [
+            {
+                "agent_id": 1000007,
+                "ok": True,
+                "response": {"errcode": 0, "errmsg": "ok"},
+                "error": None,
+            }
+        ]
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="任务创建图文通知",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="test_message_news",
+            config={"agent_ids": [1000007]},
+        )
+
+        result = send_notification_with_notifier(
+            notifier=notifier,
+            payload={
+                "mission_id": 12,
+                "description": "检查打印质量",
+                "category_name": "通用",
+            },
+        )
+
+        self.assertEqual(result["status"], "sent")
+        self.assertEqual(result["msgtype"], "news")
+        self.assertEqual(result["ok_count"], 1)
+        mock_send.assert_called_once()
+
+    def test_send_notification_with_notifier_rejects_invalid_message_api_template(self):
+        notifier = Notifier.objects.create(
+            merchant=self.merchant,
+            name="非法消息模板",
+            channel=NotifierChannelEnum.MESSAGE_API,
+            template_key="test_message_invalid",
+            config={"agent_id": 1000007},
+        )
+
+        with self.assertRaisesMessage(Exception, "content"):
+            send_notification_with_notifier(
+                notifier=notifier,
+                payload={"mission_id": 12},
+            )
+class MessageAPIServiceTestCase(TestCase):
+    @patch("notifier.message_api.urlopen")
+    def test_send_message_api_text_message_calls_message_api_endpoint(self, mock_urlopen):
+        from notifier.message_api import send_message_api_text_message
+
+        response_obj = MagicMock()
+        response_obj.read.return_value = b'{"errcode":0,"errmsg":"ok"}'
+        response_obj.__enter__.return_value = response_obj
+        response_obj.__exit__.return_value = False
+
+        mock_urlopen.return_value = response_obj
+
+        with self.settings(
+            MESSAGE_API_BASE_URL="http://message-api.internal:8198",
+            MESSAGE_API_AUTHORIZATION="secret-1",
+        ):
+            response = send_message_api_text_message(agent_id=1000007, content="hello")
+
+        self.assertTrue(response.ok)
+        self.assertEqual(response.errcode, 0)
+        self.assertEqual(mock_urlopen.call_count, 1)
+        request = mock_urlopen.call_args.args[0]
+        self.assertEqual(request.full_url, "http://message-api.internal:8198/api/message/send")
+        self.assertEqual(request.get_method(), "POST")
+        self.assertEqual(request.get_header("Authorization"), "secret-1")
+
+    @patch("notifier.message_api.urlopen")
+    def test_send_message_api_news_to_agents_calls_message_api_endpoint(self, mock_urlopen):
+        from notifier.message_api import send_message_api_news_to_agents
+
+        response_obj = MagicMock()
+        response_obj.read.return_value = (
+            b'{"results":[{"agent_id":1000007,"ok":true,"response":{"errcode":0,"errmsg":"ok"},"error":null}]}'
+        )
+        response_obj.__enter__.return_value = response_obj
+        response_obj.__exit__.return_value = False
+        mock_urlopen.return_value = response_obj
+
+        with self.settings(
+            MESSAGE_API_BASE_URL="http://message-api.internal:8198",
+            MESSAGE_API_AUTHORIZATION="secret-1",
+        ):
+            results = send_message_api_news_to_agents(
+                agent_ids=[1000007],
+                title="销售日报",
+                description="点击查看今日各区域销售汇总",
+                url="https://example.com/reports/daily-sales",
+                image_url="https://example.com/static/daily-sales-cover.png",
+            )
+
+        self.assertEqual(len(results), 1)
+        request = mock_urlopen.call_args.args[0]
+        self.assertEqual(request.full_url, "http://message-api.internal:8198/api/message/send/news")
+        self.assertEqual(request.get_header("Authorization"), "secret-1")