85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
"""企微机器人显示名：后台配置 + 新旧都认。"""
|
|
|
|
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)
|
|
|
|
os.environ.setdefault("LEDGER_BACKEND", "memory")
|
|
|
|
from agent.policy.bot_name import (
|
|
DEFAULT_BOT_DISPLAY_NAME,
|
|
REPLY_CODE,
|
|
at_bot,
|
|
bot_display_name,
|
|
bot_mention_names,
|
|
strip_bot_mention,
|
|
text_mentions_bot,
|
|
)
|
|
from agent.policy.bot_reply import override_catalog, render
|
|
from agent.policy import inquiry_copy as copy
|
|
|
|
|
|
class BotDisplayNameTests(unittest.TestCase):
|
|
def tearDown(self) -> None:
|
|
override_catalog(None)
|
|
|
|
def test_default_when_empty(self) -> None:
|
|
override_catalog({})
|
|
self.assertEqual(bot_display_name(), DEFAULT_BOT_DISPLAY_NAME)
|
|
self.assertEqual(at_bot(), f"@{DEFAULT_BOT_DISPLAY_NAME}")
|
|
self.assertIn("询价小助手", bot_mention_names())
|
|
self.assertIn(DEFAULT_BOT_DISPLAY_NAME, bot_mention_names())
|
|
|
|
def test_override_from_admin(self) -> None:
|
|
override_catalog({REPLY_CODE: "测试助手名"})
|
|
self.assertEqual(bot_display_name(), "测试助手名")
|
|
self.assertTrue(text_mentions_bot("@测试助手名 WO1"))
|
|
self.assertTrue(text_mentions_bot("@询价小助手 WO1"))
|
|
self.assertEqual(strip_bot_mention("@测试助手名 2"), "2")
|
|
self.assertEqual(strip_bot_mention("@询价小助手 2"), "2")
|
|
|
|
def test_collab_hint_uses_display_name(self) -> None:
|
|
override_catalog({REPLY_CODE: "智报通AIQ"})
|
|
hint = copy.collab_hint("WO202610060030")
|
|
self.assertIn("WO202610060030@智报通AIQ", hint)
|
|
self.assertNotIn("询价小助手", hint)
|
|
|
|
def test_template_bot_name_placeholder(self) -> None:
|
|
override_catalog(
|
|
{
|
|
REPLY_CODE: "智报通AIQ",
|
|
"inquiring.air_collab_hint": "请发送 {{ticket_no}}@{{bot_name}}",
|
|
}
|
|
)
|
|
self.assertEqual(
|
|
render("inquiring.air_collab_hint", "缺省", ticket_no="WO1"),
|
|
"请发送 WO1@智报通AIQ",
|
|
)
|
|
|
|
def test_legacy_and_new_strip(self) -> None:
|
|
override_catalog({REPLY_CODE: "智报通AIQ"})
|
|
rest_old = strip_bot_mention("工单编号WO202610070026@询价小助手")
|
|
rest_new = strip_bot_mention("@智报通AIQ WO202610070026")
|
|
self.assertNotIn("询价小助手", rest_old)
|
|
self.assertNotIn("智报通AIQ", rest_new)
|
|
self.assertIn("WO202610070026", rest_old)
|
|
self.assertIn("WO202610070026", rest_new)
|
|
|
|
def test_strip_wecom_lock_instruction(self) -> None:
|
|
"""锁舱 instructionText 不得把 @机器人 原样带给 TMS。"""
|
|
override_catalog({REPLY_CODE: "智价通AIQ"})
|
|
cleaned = strip_bot_mention("@智价通AIQ AIR-OPT01, TH, TH641")
|
|
self.assertEqual(cleaned, "AIR-OPT01, TH, TH641")
|
|
self.assertNotIn("@", cleaned)
|
|
self.assertNotIn("智价通AIQ", cleaned)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|