feat: 事件异常超时编码与点名对象

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-21 18:15:22 +08:00
parent ef0424a28f
commit 2527901efd
2 changed files with 299 additions and 0 deletions
@@ -0,0 +1,106 @@
+"""
+事件异常纯规则：超时编码、点名对象、卡片接收人（不打外网）。
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+import unittest
+
+ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if ROOT not in sys.path:
+    sys.path.insert(0, ROOT)
+
+from agent.policy.event_exception import (
+    WAIT_ADOPT,
+    WAIT_AIR_QUOTE,
+    WAIT_DEAL,
+    WAIT_PRODUCT_QUOTE,
+    conversation_of,
+    format_timeout_hours,
+    pick_app_targets,
+    pick_mentions,
+    timeout_code,
+)
+
+
+class TestTimeoutCode(unittest.TestCase):
+    def test_sales_unified(self) -> None:
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_ADOPT, business_line="SEA", transport_class="急件"),
+            "sea_sales",
+        )
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_DEAL, business_line="AIR", transport_class=""),
+            "air_sales",
+        )
+        self.assertEqual(
+            timeout_code(wait_kind="lost_reason", business_line="LAND", transport_class="急件"),
+            "land_sales",
+        )
+
+    def test_air_route(self) -> None:
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_AIR_QUOTE, business_line="AIR", transport_class=""),
+            "air_route",
+        )
+
+    def test_sea_and_land_product_class(self) -> None:
+        self.assertEqual(
+            timeout_code(
+                wait_kind=WAIT_PRODUCT_QUOTE, business_line="SEA", transport_class="Port to Port"
+            ),
+            "sea_product_ptp",
+        )
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_PRODUCT_QUOTE, business_line="SEA", transport_class="其他"),
+            "sea_product_other",
+        )
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_PRODUCT_QUOTE, business_line="LAND", transport_class="急件"),
+            "land_hk_express",
+        )
+        self.assertEqual(
+            timeout_code(wait_kind=WAIT_PRODUCT_QUOTE, business_line="SEA", transport_class=""),
+            "",
+        )
+
+    def test_hours_format(self) -> None:
+        self.assertEqual(format_timeout_hours(0.25), "0.25")
+        self.assertEqual(format_timeout_hours(48), "48")
+
+
+class TestMentions(unittest.TestCase):
+    def test_product_all_unanswered(self) -> None:
+        sales = {
+            "wecomId": "s1",
+            "name": "销售甲",
+            "roleCode": "sales",
+            "notifyEventException": 1,
+        }
+        members = [
+            {"wecomId": "p1", "name": "张三", "roleCode": "sea", "notifyEventException": 1},
+            {"wecomId": "p2", "name": "李四", "roleCode": "sea", "notifyEventException": 0},
+            {"wecomId": "s1", "name": "销售甲", "roleCode": "sales", "notifyEventException": 1},
+        ]
+        hit = pick_mentions(
+            wait_kind=WAIT_PRODUCT_QUOTE,
+            sales=sales,
+            members=members,
+            business_line="SEA",
+        )
+        self.assertEqual([p["name"] for p in hit], ["张三", "李四"])
+        apps = pick_app_targets(hit)
+        self.assertEqual([p["wecomId"] for p in apps], ["p1"])
+
+    def test_adopt_is_private(self) -> None:
+        self.assertEqual(conversation_of(wait_kind=WAIT_ADOPT, collab_chat_id="wrxxx"), "private")
+
+    def test_deal_uses_group_when_present(self) -> None:
+        self.assertEqual(conversation_of(wait_kind=WAIT_DEAL, collab_chat_id="wrxxx"), "group")
+        self.assertEqual(conversation_of(wait_kind=WAIT_DEAL, collab_chat_id=""), "private")
+
+
+if __name__ == "__main__":
+    unittest.main()