feat: upgrade客服多店铺分流、批量报价与稳定性防护

2026-02-28 18:52:31 +08:00
parent c39840fe15
commit 46143be86c
16 changed files with 1329 additions and 37 deletions
--- a/tests/test_system_inquiry_rules.py
+++ b/tests/test_system_inquiry_rules.py
@@ -0,0 +1,70 @@
+import os
+import unittest
+from unittest.mock import AsyncMock
+
+from core.websocket_client import QingjianAPIClient
+
+
+class SystemInquiryRulesTest(unittest.IsolatedAsyncioTestCase):
+    def setUp(self):
+        self.rules = {
+            "enabled": True,
+            "default_action": "silent",
+            "default_reply": "已收到",
+            "sender_keywords": ["系统客服", "官方客服"],
+            "message_keywords": ["系统询单", "代客咨询"],
+            "shops": {
+                "shop_reply": {
+                    "enabled": True,
+                    "action": "reply",
+                    "reply": "店铺回复模板",
+                    "sender_keywords": ["机器人客服"],
+                    "message_keywords": ["询单"],
+                }
+            },
+        }
+        os.environ["SYSTEM_INQUIRY_ENABLED"] = "true"
+        os.environ["SYSTEM_INQUIRY_SHOPS"] = ""
+
+    async def test_detect_by_sender_keyword(self):
+        client = QingjianAPIClient(enable_agent=False)
+        client._system_inquiry_rules = self.rules
+        policy = client._resolve_system_inquiry_policy("shop_a")
+        data = {"acc_id": "shop_a", "from_name": "平台系统客服", "from_id": "kefu001", "msg": "你好"}
+        self.assertTrue(client._match_system_inquiry(data, policy))
+
+    async def test_shop_rule_reply_action(self):
+        client = QingjianAPIClient(enable_agent=False)
+        client._system_inquiry_rules = self.rules
+        client.send_reply = AsyncMock()
+        client.transfer_to_human = AsyncMock()
+
+        data = {
+            "acc_id": "shop_reply",
+            "from_name": "机器人客服A",
+            "from_id": "robot_01",
+            "msg": "有个询单请处理",
+            "acc_type": "AliWorkbench",
+        }
+        handled = await client._handle_system_inquiry(data)
+        self.assertTrue(handled)
+        client.send_reply.assert_awaited_once()
+        client.transfer_to_human.assert_not_awaited()
+
+    async def test_shop_whitelist_blocks_other_shops(self):
+        os.environ["SYSTEM_INQUIRY_SHOPS"] = "shop_only"
+        client = QingjianAPIClient(enable_agent=False)
+        client._system_inquiry_rules = self.rules
+        client.send_reply = AsyncMock()
+        data = {"acc_id": "shop_other", "from_name": "系统客服", "from_id": "sys_1", "msg": "系统询单"}
+        handled = await client._handle_system_inquiry(data)
+        self.assertFalse(handled)
+        client.send_reply.assert_not_awaited()
+
+    def tearDown(self):
+        for k in ("SYSTEM_INQUIRY_ENABLED", "SYSTEM_INQUIRY_SHOPS"):
+            os.environ.pop(k, None)
+
+
+if __name__ == "__main__":
+    unittest.main(verbosity=2)