jillion886andCursor
378bb892b3
停发填写样例图，同群只跟当前工单，重量体积统一为 KG/CBM。 Co-authored-by: Cursor <cursoragent@cursor.com>
1277 lines
51 KiB
Python
1277 lines
51 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)
|
|
|
|
from agent.channel.wecom.models import InboundMessage
|
|
from agent.handlers.air_group_collab import handle_air_group
|
|
from agent.ledger.memory_ledger import MemoryLedger
|
|
from agent.policy.air_text_flow import AirTextInquiryFlow
|
|
from agent.policy import inquiry_copy as copy
|
|
from agent.routing.air_group_intent import (
|
|
INTENT_LOCK,
|
|
INTENT_RELEASE,
|
|
classify_air_group_text,
|
|
)
|
|
from agent.routing.dispatch import dispatch_inbound
|
|
from agent.routing.decision import RouteDecision
|
|
|
|
|
|
class FakeBotGroup:
|
|
def __init__(self) -> None:
|
|
self.sent: list[dict] = []
|
|
|
|
def send_group(self, *, chat_id: str, content: str, mention_userids=None):
|
|
self.sent.append({"chat_id": chat_id, "content": content, "kind": "text"})
|
|
|
|
def send_group_file(self, *, chat_id: str, filename: str, filepath: str = "", file_b64: str = ""):
|
|
self.sent.append({"chat_id": chat_id, "filename": filename, "kind": "file"})
|
|
return {"ok": True}
|
|
|
|
|
|
class AirGroupCollabTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
from agent.redis_coord.air_inbound import reset_air_inbound_for_tests
|
|
|
|
reset_air_inbound_for_tests()
|
|
self.ledger = MemoryLedger()
|
|
self.groups = FakeBotGroup()
|
|
self.flow = AirTextInquiryFlow(ledger=self.ledger, group_client=self.groups)
|
|
self.ledger.upsert_staff(
|
|
{
|
|
"name": "WuJiLin",
|
|
"wecomId": "sales1",
|
|
"roleCode": "sales",
|
|
"status": "active",
|
|
}
|
|
)
|
|
self.ledger.upsert_staff(
|
|
{
|
|
"name": "HangXian",
|
|
"wecomId": "air1",
|
|
"roleCode": "air",
|
|
"status": "active",
|
|
}
|
|
)
|
|
self.ledger.set_quote_templates(
|
|
[
|
|
{
|
|
"templateId": "air-default",
|
|
"name": "空运报价",
|
|
"bizType": "空运",
|
|
"isDefault": True,
|
|
},
|
|
{
|
|
"templateId": "sea-default",
|
|
"name": "海运报价",
|
|
"bizType": "海运",
|
|
"isDefault": True,
|
|
},
|
|
]
|
|
)
|
|
|
|
def _create_air(self, *, sender="sales1", quote=None, status="已报价", quote_date="2026-09-16") -> str:
|
|
created = self.ledger.create_ticket(
|
|
sender_id=sender,
|
|
business_line="AIR",
|
|
facts={
|
|
"起运地": "上海",
|
|
"目的地": "洛杉矶",
|
|
"品名": "配件",
|
|
"件数": "3",
|
|
"毛重": "100",
|
|
"体积": "2",
|
|
"包装类型": "纸箱",
|
|
"报价日期": quote_date,
|
|
},
|
|
)
|
|
wo = created["work_order_no"]
|
|
if quote is None:
|
|
quote = {
|
|
"total": "USD 800",
|
|
"source": "TMS",
|
|
"source_label": "TMS 标准报价",
|
|
"quoteId": "QT-AIR-1",
|
|
"airOptions": [
|
|
{"optionNo": "AIR-OPT-01", "airlineCode": "CZ", "airlineName": "南航"}
|
|
],
|
|
}
|
|
if quote:
|
|
self.ledger.upsert_quote(work_order_no=wo, quote=quote, to_status=status)
|
|
return wo
|
|
|
|
def _msg(self, *, sender: str, text: str, chat: str, mid: str) -> InboundMessage:
|
|
return InboundMessage(
|
|
sender_id=sender,
|
|
message_id=mid,
|
|
content=text,
|
|
chat_id=chat,
|
|
chat_type="group",
|
|
raw={"source": "aibot"},
|
|
)
|
|
|
|
def _texts(self) -> list[str]:
|
|
return [x.get("content") or "" for x in self.groups.sent]
|
|
|
|
def test_attach_bot_client_keeps_req_id_for_quote(self) -> None:
|
|
"""这次 @ 的 req_id 要带给桥，群里才能引用原消息。"""
|
|
from agent.channel.aibot.reply import AibotReplyClient
|
|
from agent.handlers.air_group_collab import _attach_bot_client
|
|
|
|
engine = AirTextInquiryFlow(ledger=self.ledger, group_client=None)
|
|
msg = InboundMessage(
|
|
sender_id="air1",
|
|
message_id="m_quote",
|
|
content="锁舱",
|
|
chat_id="wr_air_1",
|
|
chat_type="group",
|
|
raw={"source": "aibot", "aibot_req_id": "cb_lock_1"},
|
|
)
|
|
_attach_bot_client(engine, msg)
|
|
client = engine.group_client()
|
|
self.assertIsInstance(client, AibotReplyClient)
|
|
self.assertEqual(client._callback_req_id, "cb_lock_1")
|
|
|
|
def test_intent_lock_is_not_hardcoded(self) -> None:
|
|
self.assertEqual(classify_air_group_text("帮我锁舱 WO202609160001", allow_network=False), INTENT_LOCK)
|
|
self.assertEqual(classify_air_group_text("WO202609160001释放舱位", allow_network=False), INTENT_RELEASE)
|
|
|
|
def test_air_group_brief_matches_sea_sections(self) -> None:
|
|
text = copy.air_group_brief(
|
|
work_order_no="WO202609160009",
|
|
facts={
|
|
"起运地": "ZUH",
|
|
"目的地": "广州",
|
|
"品名": "服务器",
|
|
"件数": "1",
|
|
"毛重": "10",
|
|
"体积": "1",
|
|
"包装类型": "纸箱",
|
|
"报价日期": "2026-09-16",
|
|
},
|
|
quote=None,
|
|
collab_facts={},
|
|
has_price=False,
|
|
)
|
|
self.assertIn("工单号：WO202609160009", text)
|
|
self.assertIn("询价字段：\n运输方式：空运", text)
|
|
self.assertIn("起运地：ZUH", text)
|
|
self.assertIn("\n\n协同补充字段：\n是否含电：-", text)
|
|
self.assertIn("是否含磁：-", text)
|
|
general = copy.air_group_brief(
|
|
work_order_no="WO202609230017",
|
|
facts={
|
|
"起运地": "上海",
|
|
"目的地": "洛杉矶",
|
|
"品名": "普货",
|
|
},
|
|
quote=None,
|
|
collab_facts={},
|
|
has_price=False,
|
|
)
|
|
self.assertIn("是否含电：否", general)
|
|
self.assertIn("是否含磁：否", general)
|
|
self.assertNotIn("是否含电：-", general)
|
|
self.assertIn("\n\nTMS报价：\nTMS 暂无匹配报价", text)
|
|
self.assertNotIn("当前工单：", text)
|
|
self.assertNotIn("贸易条款", text)
|
|
|
|
def test_air_group_brief_tms_total_only_once(self) -> None:
|
|
"""群摘要 TMS 段总销售价只一行，禁止重复。"""
|
|
tms = {
|
|
"source": "TMS",
|
|
"source_label": "TMS标准报价",
|
|
"total": "183370.0",
|
|
"currency": "CNY",
|
|
"eta": "-",
|
|
"valid_until": "2026-09-30",
|
|
"fee_lines": [
|
|
{"name": "运输费", "amount": "11.0", "currency": "CNY"},
|
|
],
|
|
}
|
|
text = copy.air_group_brief(
|
|
work_order_no="WO202609220018",
|
|
facts={"起运港": "珠海", "目的港": "克拉克"},
|
|
quote=tms,
|
|
collab_facts={},
|
|
has_price=True,
|
|
tms_quote=tms,
|
|
)
|
|
tms_block = text.split("TMS报价：", 1)[1]
|
|
self.assertEqual(tms_block.count("总销售价："), 1)
|
|
self.assertIn("总销售价：CNY 183370.0", tms_block)
|
|
self.assertNotIn("时效", tms_block)
|
|
self.assertIn("报价有效期：2026-09-30", tms_block)
|
|
|
|
def test_air_group_brief_uses_route_fees_not_air_freight_total(self) -> None:
|
|
"""
|
|
群摘要必须跟单聊同一套线路分项。
|
|
主账快照若只有「空运费=总价」，仍要列出该线路运输费/提送货费/人工费。
|
|
"""
|
|
tms = {
|
|
"source": "TMS",
|
|
"source_label": "TMS标准报价",
|
|
"total": "183370.0",
|
|
"freight": "183370.0",
|
|
"currency": "CNY",
|
|
"eta": "-",
|
|
"valid_until": "2026-09-30",
|
|
"fee_lines": [
|
|
{
|
|
"name": "空运费",
|
|
"amount": "183370.00",
|
|
"currency": "CNY",
|
|
}
|
|
],
|
|
"airOptions": [
|
|
{
|
|
"originAirportCode": "ZUH",
|
|
"transferAirportCode": "CHG",
|
|
"destinationAirportCode": "CRK",
|
|
"salePrice": 183370,
|
|
"currency": "CNY",
|
|
"departureCharges": [
|
|
{"chargeName": "运输费", "currency": "CNY", "unitPrice": 11.0},
|
|
{"chargeName": "提送货费", "currency": "CNY", "unitPrice": 6.0},
|
|
{"chargeName": "人工费", "currency": "CNY", "unitPrice": 123.0},
|
|
],
|
|
}
|
|
],
|
|
}
|
|
text = copy.air_group_brief(
|
|
work_order_no="WO202609220018",
|
|
facts={"起运港": "珠海", "目的港": "克拉克", "originCode": "ZUH"},
|
|
quote=tms,
|
|
collab_facts={},
|
|
has_price=True,
|
|
tms_quote=tms,
|
|
)
|
|
tms_block = text.split("TMS报价：", 1)[1]
|
|
self.assertIn("运输费：CNY 11.0", tms_block)
|
|
self.assertIn("提送货费：CNY 6.0", tms_block)
|
|
self.assertIn("人工费：CNY 123.0", tms_block)
|
|
self.assertIn("总销售价：CNY 183370", tms_block)
|
|
self.assertEqual(tms_block.count("总销售价："), 1)
|
|
self.assertNotIn("空运费", tms_block)
|
|
|
|
def test_air_group_brief_multi_route_does_not_borrow_other_air_freight(self) -> None:
|
|
"""多条线路：只出对上起运港那条的分项，禁止套整单空运费或另一条的运输费。"""
|
|
tms = {
|
|
"source": "TMS",
|
|
"total": "200040",
|
|
"freight": "183370.00",
|
|
"currency": "CNY",
|
|
"valid_until": "2026-09-30",
|
|
"fee_lines": [
|
|
{"name": "空运费", "amount": "183370.00", "currency": "CNY"},
|
|
],
|
|
"airOptions": [
|
|
{
|
|
"originAirportCode": "ZUH",
|
|
"transferAirportCode": "CHG",
|
|
"destinationAirportCode": "CRK",
|
|
"salePrice": 183370,
|
|
"currency": "CNY",
|
|
"departureCharges": [
|
|
{"chargeName": "运输费", "currency": "CNY", "unitPrice": 11.0},
|
|
],
|
|
},
|
|
{
|
|
"originAirportCode": "SZX",
|
|
"destinationAirportCode": "CRK",
|
|
"salePrice": 200040,
|
|
"currency": "CNY",
|
|
"departureCharges": [],
|
|
},
|
|
],
|
|
}
|
|
text = copy.air_group_brief(
|
|
work_order_no="WO202609220018",
|
|
facts={"起运港": "深圳", "目的港": "克拉克", "originCode": "SZX"},
|
|
quote=tms,
|
|
collab_facts={},
|
|
has_price=True,
|
|
tms_quote=tms,
|
|
)
|
|
tms_block = text.split("TMS报价：", 1)[1]
|
|
self.assertNotIn("空运费", tms_block)
|
|
self.assertNotIn("运输费", tms_block)
|
|
self.assertIn("总销售价：CNY 200040", tms_block)
|
|
self.assertEqual(tms_block.count("总销售价："), 1)
|
|
|
|
def test_work_order_only_is_activate_not_quote(self) -> None:
|
|
from agent.routing.air_group_intent import INTENT_ACTIVATE
|
|
|
|
self.assertEqual(
|
|
classify_air_group_text("WO202609160002", allow_network=False),
|
|
INTENT_ACTIVATE,
|
|
)
|
|
self.assertEqual(
|
|
classify_air_group_text("@询价小助手 WO202609160002", allow_network=False),
|
|
INTENT_ACTIVATE,
|
|
)
|
|
from unittest.mock import patch
|
|
|
|
with patch(
|
|
"agent.routing.air_group_intent._classify_with_model",
|
|
side_effect=AssertionError("纯工单号不应再打模型"),
|
|
):
|
|
self.assertEqual(
|
|
classify_air_group_text(
|
|
"@询价小助手 WO202609160002",
|
|
allow_network=True,
|
|
),
|
|
INTENT_ACTIVATE,
|
|
)
|
|
|
|
def test_dispatch_aibot_goes_air(self) -> None:
|
|
from agent.routing.dispatch import dispatch_inbound as _disp
|
|
|
|
wo = self._create_air()
|
|
msg = self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="d1")
|
|
# 单测注入 flow，避免打到进程单例
|
|
import agent.handlers.air_group_collab as air_mod
|
|
orig = air_mod.get_air_text_flow
|
|
|
|
def _fake():
|
|
return self.flow
|
|
|
|
air_mod.get_air_text_flow = _fake # type: ignore
|
|
try:
|
|
name = _disp(msg, RouteDecision(intent="ordinary_text_other"))
|
|
finally:
|
|
air_mod.get_air_text_flow = orig # type: ignore
|
|
self.assertEqual(name, "air_group_collab")
|
|
|
|
def test_activate_and_switch(self) -> None:
|
|
wo1 = self._create_air()
|
|
wo2 = self._create_air()
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=wo1, chat="wr_air_1", mid="a1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "activated")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo1)
|
|
self.assertEqual(ticket.event_wait_kind, "air_quote")
|
|
self.assertTrue(ticket.event_wait_started_at)
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn(f"工单号：{wo1}", joined)
|
|
self.assertIn("询价字段：", joined)
|
|
self.assertIn("运输方式：空运", joined)
|
|
self.assertIn("协同补充字段：", joined)
|
|
self.assertIn("TMS报价：", joined)
|
|
self.assertNotIn("当前工单：", joined)
|
|
self.assertNotIn("询价要点：", joined)
|
|
self.assertTrue(any("@WuJiLin" in t and "@HangXian" in t for t in self._texts()))
|
|
self.assertEqual(getattr(self.ledger, "last_render", {}).get("templateId"), "air-default")
|
|
kinds = [x.get("kind") for x in self.groups.sent]
|
|
self.assertLess(kinds.index("text"), kinds.index("file"))
|
|
self.assertEqual(sum(1 for x in self.groups.sent if x.get("kind") == "file"), 1)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=wo2, chat="wr_air_1", mid="a2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "activated")
|
|
self.assertEqual(self.ledger.find_by_collab_chat(chat_id="wr_air_1").work_order_no, wo2)
|
|
|
|
def test_activate_prefills_magnet(self) -> None:
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales1",
|
|
business_line="AIR",
|
|
facts={
|
|
"起运地": "上海",
|
|
"目的地": "洛杉矶",
|
|
"品名": "配件",
|
|
"件数": "3",
|
|
"毛重": "100",
|
|
"体积": "2",
|
|
"包装类型": "纸箱",
|
|
"报价日期": "2026-09-16",
|
|
"是否含磁": "是",
|
|
},
|
|
)
|
|
wo = created["work_order_no"]
|
|
self.ledger.upsert_quote(
|
|
work_order_no=wo,
|
|
quote={"total": "USD 800", "source": "TMS", "quoteId": "QT-AIR-1"},
|
|
to_status="已报价",
|
|
)
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="pre1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("是否含磁：是", joined)
|
|
self.assertIn("是否含电：-", joined)
|
|
|
|
def test_activate_general_cargo_defaults_battery_and_magnet(self) -> None:
|
|
"""品名是普货：群内激活后含电、含磁都是否，报价单也带上。"""
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales1",
|
|
business_line="AIR",
|
|
facts={
|
|
"起运地": "上海",
|
|
"目的地": "洛杉矶",
|
|
"品名": "普货",
|
|
"件数": "3",
|
|
"毛重": "100",
|
|
"体积": "2",
|
|
"包装类型": "纸箱",
|
|
"报价日期": "2026-09-16",
|
|
},
|
|
)
|
|
wo = created["work_order_no"]
|
|
self.ledger.upsert_quote(
|
|
work_order_no=wo,
|
|
quote={"total": "USD 800", "source": "TMS", "quoteId": "QT-AIR-1"},
|
|
to_status="已报价",
|
|
)
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="gc1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.collab_facts.get("是否含电"), "否")
|
|
self.assertEqual(ticket.collab_facts.get("是否含磁"), "否")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("是否含电：否", joined)
|
|
self.assertIn("是否含磁：否", joined)
|
|
fill = dict(getattr(self.ledger, "last_render", {}).get("facts") or {})
|
|
self.assertEqual(fill.get("是否含电") or fill.get("has_battery"), "否")
|
|
self.assertEqual(fill.get("是否含磁") or fill.get("has_magnet"), "否")
|
|
|
|
def test_activate_general_cargo_keeps_explicit_battery(self) -> None:
|
|
"""普货只补空着的项；私聊已经说含电，激活不能改成否。"""
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales1",
|
|
business_line="AIR",
|
|
facts={
|
|
"起运地": "上海",
|
|
"目的地": "洛杉矶",
|
|
"品名": "普货",
|
|
"件数": "3",
|
|
"毛重": "100",
|
|
"体积": "2",
|
|
"包装类型": "纸箱",
|
|
"报价日期": "2026-09-16",
|
|
"是否含电": "是",
|
|
},
|
|
)
|
|
wo = created["work_order_no"]
|
|
self.ledger.upsert_quote(
|
|
work_order_no=wo,
|
|
quote={"total": "USD 800", "source": "TMS", "quoteId": "QT-AIR-1"},
|
|
to_status="已报价",
|
|
)
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="gc2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.collab_facts.get("是否含电"), "是")
|
|
self.assertEqual(ticket.collab_facts.get("是否含磁"), "否")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("是否含电：是", joined)
|
|
self.assertIn("是否含磁：否", joined)
|
|
|
|
def test_second_group_occupied(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="o1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_2", mid="o2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "occupied")
|
|
self.assertTrue(any("另一个群" in t for t in self._texts()))
|
|
|
|
def test_need_work_order_when_empty(self) -> None:
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text="空运费100USD", chat="wr_air_1", mid="n1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="product_quote",
|
|
)
|
|
self.assertEqual(phase, "need_work_order")
|
|
|
|
def test_sales_can_fill_battery(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="f0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text="含电 不含磁", chat="wr_air_1", mid="f1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="collab_fields",
|
|
)
|
|
self.assertEqual(phase, "fields_saved")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.collab_facts.get("是否含电"), "是")
|
|
self.assertEqual(ticket.collab_facts.get("是否含磁"), "否")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("工单号：", joined)
|
|
self.assertIn("询价字段：", joined)
|
|
self.assertIn("是否含电：是", joined)
|
|
self.assertIn("是否含磁：否", joined)
|
|
self.assertNotIn("已记下", joined)
|
|
self.assertTrue(any("@WuJiLin" in t and "@HangXian" in t for t in self._texts()))
|
|
self.assertEqual(getattr(self.ledger, "last_render", {}).get("templateId"), "air-default")
|
|
fill = dict(getattr(self.ledger, "last_render", {}).get("facts") or {})
|
|
self.assertEqual(fill.get("是否含电") or fill.get("has_battery"), "是")
|
|
self.assertEqual(fill.get("是否含磁") or fill.get("has_magnet"), "否")
|
|
texts = [i for i, x in enumerate(self.groups.sent) if x.get("kind") == "text" and "工单号：" in (x.get("content") or "")]
|
|
files = [i for i, x in enumerate(self.groups.sent) if x.get("kind") == "file"]
|
|
self.assertTrue(texts and files)
|
|
self.assertLess(texts[-1], files[-1])
|
|
self.assertEqual(len(files), 2)
|
|
|
|
def test_collab_after_activate_does_not_use_old_land_ticket(self) -> None:
|
|
"""
|
|
测服 WO202609230023：空运进原群激活后补含电含磁，
|
|
不能写到同群更早的陆运单 WO202609230007，摘要也不能变成空字段。
|
|
"""
|
|
land = self.ledger.create_ticket(
|
|
sender_id="sales1",
|
|
business_line="LAND",
|
|
facts={
|
|
"运输类型": "国内运输整车",
|
|
"始发站": "深圳",
|
|
"目的站": "上海",
|
|
"品名": "配件",
|
|
},
|
|
)["work_order_no"]
|
|
self.ledger.bind_collab_group(
|
|
work_order_no=land,
|
|
chat_id="wr_air_1",
|
|
member_ids=["sales1"],
|
|
)
|
|
air = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=air, chat="wr_air_1", mid="mix0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(self.ledger.find_by_collab_chat(chat_id="wr_air_1").work_order_no, air)
|
|
self.assertEqual(
|
|
str(self.ledger.get_ticket(work_order_no=land).collab_chat_id or ""),
|
|
"",
|
|
)
|
|
self.groups.sent.clear()
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text="含电 不含磁", chat="wr_air_1", mid="mix1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="collab_fields",
|
|
)
|
|
self.assertEqual(phase, "fields_saved")
|
|
air_ticket = self.ledger.get_ticket(work_order_no=air)
|
|
land_ticket = self.ledger.get_ticket(work_order_no=land)
|
|
self.assertEqual(air_ticket.collab_facts.get("是否含电"), "是")
|
|
self.assertEqual(air_ticket.collab_facts.get("是否含磁"), "否")
|
|
self.assertFalse(land_ticket.collab_facts.get("是否含电"))
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn(f"工单号：{air}", joined)
|
|
self.assertNotIn(f"工单号：{land}", joined)
|
|
self.assertIn("起运地：上海", joined)
|
|
self.assertIn("重量（KG）：100", joined)
|
|
self.assertNotIn("起运地：-", joined)
|
|
|
|
def test_sales_quote_rejected(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="q0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text="空运费 100USD", chat="wr_air_1", mid="q1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="product_quote",
|
|
)
|
|
self.assertEqual(phase, "not_airline")
|
|
|
|
def test_airline_quote_asks_tms_choice(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="t0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="空运费改为900USD", chat="wr_air_1", mid="t1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="product_quote",
|
|
injected_quote={"total": "USD 900", "source_label": "航线报价"},
|
|
)
|
|
self.assertEqual(phase, "wait_tms_choice")
|
|
self.assertTrue(any("在 TMS 基础上改" in t or "在TMS基础上改" in t for t in self._texts()))
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.quote.get("total"), "USD 800")
|
|
self.assertNotEqual(ticket.quote.get("source_label"), "航线报价")
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="在TMS基础上改", chat="wr_air_1", mid="t2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="quote_keep_tms",
|
|
)
|
|
self.assertEqual(phase, "quoted")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.quote.get("total"), "USD 900")
|
|
self.assertEqual(ticket.quote.get("quoteId"), "QT-AIR-1")
|
|
self.assertEqual(ticket.quote.get("source_label"), "航线在TMS上改")
|
|
self.assertFalse(ticket.collab_facts.get("__pending_air_quote"))
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("工单号：", joined)
|
|
self.assertIn("询价字段：", joined)
|
|
self.assertIn("人工报价：", joined)
|
|
self.assertIn("报价单已发送，请确认成交状态", joined)
|
|
self.assertEqual(getattr(self.ledger, "last_render", {}).get("templateId"), "air-default")
|
|
self.assertTrue(any(x.get("kind") == "file" for x in self.groups.sent))
|
|
|
|
def test_airline_quote_without_tms_sends_brief_sheet_deal(self) -> None:
|
|
wo = self._create_air(quote={}, status="询价中")
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="nq0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.groups.sent.clear()
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="空运费 900USD", chat="wr_air_1", mid="nq1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="product_quote",
|
|
injected_quote={"total": "USD 900", "source_label": "航线报价"},
|
|
)
|
|
self.assertEqual(phase, "quoted")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.event_wait_kind, "deal")
|
|
self.assertTrue(ticket.event_wait_started_at)
|
|
self.assertEqual(ticket.status, "已报价")
|
|
self.assertEqual(ticket.quote.get("total"), "USD 900")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("工单号：", joined)
|
|
self.assertIn("询价字段：", joined)
|
|
self.assertIn("TMS 暂无匹配报价", joined)
|
|
self.assertIn("人工报价：", joined)
|
|
self.assertIn("报价单已发送，请确认成交状态", joined)
|
|
self.assertNotIn("航线报价已记下", joined)
|
|
self.assertEqual(getattr(self.ledger, "last_render", {}).get("templateId"), "air-default")
|
|
self.assertTrue(any(x.get("kind") == "file" for x in self.groups.sent))
|
|
|
|
def test_quote_same_sentence_saves_battery(self) -> None:
|
|
wo = self._create_air(quote={}, status="询价中")
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="qs0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="含电 空运费900USD", chat="wr_air_1", mid="qs1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="product_quote",
|
|
injected_quote={"total": "USD 900", "source_label": "航线报价"},
|
|
)
|
|
self.assertEqual(phase, "quoted")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.collab_facts.get("是否含电"), "是")
|
|
self.assertIn("是否含电：是", "\n".join(self._texts()))
|
|
|
|
def test_confirm_tms_saves_sheet_and_deal(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="ct0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.groups.sent.clear()
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="确认报价", chat="wr_air_1", mid="ct1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="quote_confirm_tms",
|
|
)
|
|
self.assertEqual(phase, "quote_confirmed")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.quote.get("source_label"), "航线确认TMS")
|
|
self.assertTrue(any(f"工单{wo}的1个附件已保存到后台工单" in t for t in self._texts()))
|
|
self.assertTrue(any("报价单已发送，请确认成交状态" in t for t in self._texts()))
|
|
|
|
def test_stranger_cannot_fill_fields(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="sf0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="other1", text="含电 不含磁", chat="wr_air_1", mid="sf1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="collab_fields",
|
|
)
|
|
self.assertEqual(phase, "fields_denied")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertFalse(ticket.collab_facts.get("是否含电"))
|
|
|
|
def test_sales_deal(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="c0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text="成交", chat="wr_air_1", mid="c1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="deal_won",
|
|
)
|
|
self.assertEqual(phase, "done")
|
|
self.assertEqual(self.ledger.get_ticket(work_order_no=wo).status, "已成交")
|
|
|
|
def test_lock_requires_work_order(self) -> None:
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="锁舱", chat="wr_air_1", mid="l0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "cabin_need_work_order")
|
|
self.assertTrue(any("带上工单号" in t for t in self._texts()))
|
|
|
|
def test_lock_without_tms_quote(self) -> None:
|
|
wo = self._create_air(quote={}, status="询价中")
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_1", mid="l1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
self.assertTrue(any("执行成功" in t for t in self._texts()))
|
|
self.assertTrue(any("舱位选项：AIR-OPT-01" in t for t in self._texts()))
|
|
self.assertFalse(any("还没有可锁的 TMS 报价" in t for t in self._texts()))
|
|
|
|
def test_manual_quote_without_quote_id_can_lock(self) -> None:
|
|
wo = self._create_air(
|
|
quote={
|
|
"total": "100",
|
|
"source": "manual",
|
|
"source_label": "产品报价",
|
|
},
|
|
status="已报价",
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_lock_manual", mid="lm1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
self.assertTrue(any("执行成功" in t for t in self._texts()))
|
|
|
|
def test_lock_success_and_release_after_deal_blocked(self) -> None:
|
|
wo = self._create_air()
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_1", mid="l2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
self.assertTrue(any("执行成功" in t for t in self._texts()))
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text="成交", chat="wr_air_1", mid="l3"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="deal_won",
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}释放舱位", chat="wr_air_1", mid="l4"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_release",
|
|
)
|
|
self.assertEqual(phase, "release_deal_blocked")
|
|
self.assertTrue(any("已成交" in t for t in self._texts()))
|
|
|
|
def test_lost_can_release(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_1", mid="r0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text="未成交", chat="wr_air_1", mid="r1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="deal_lost",
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}释放舱位", chat="wr_air_1", mid="r2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_release",
|
|
)
|
|
self.assertEqual(phase, "release_ok")
|
|
|
|
def test_sales_cannot_lock(self) -> None:
|
|
wo = self._create_air()
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=f"{wo}锁舱", chat="wr_air_1", mid="s1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "not_airline")
|
|
|
|
def test_dual_role_can_lock(self) -> None:
|
|
self.ledger.upsert_staff(
|
|
{
|
|
"name": "WuJiLin",
|
|
"wecomId": "sales1",
|
|
"roleCode": "air",
|
|
"status": "active",
|
|
}
|
|
)
|
|
wo = self._create_air()
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=f"{wo}锁舱", chat="wr_air_1", mid="s2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
|
|
def test_multi_option_asks(self) -> None:
|
|
wo = self._create_air(
|
|
quote={
|
|
"total": "USD 800",
|
|
"source": "TMS",
|
|
"quoteId": "QT-AIR-2",
|
|
"airOptions": [
|
|
{"optionNo": "AIR-OPT-01", "airlineCode": "CZ"},
|
|
{"optionNo": "AIR-OPT-02", "airlineCode": "MU"},
|
|
],
|
|
}
|
|
)
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_1", mid="m1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "wait_lock_option")
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text="AIR-OPT-02", chat="wr_air_1", mid="m2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_pick_option",
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.collab_facts.get("__air_lock_option"), "AIR-OPT-02")
|
|
|
|
def _file(self, *, sender: str, filename: str, chat: str, mid: str, text: str = "") -> InboundMessage:
|
|
return InboundMessage(
|
|
sender_id=sender,
|
|
message_id=mid,
|
|
content=text,
|
|
msg_type="file",
|
|
chat_id=chat,
|
|
chat_type="group",
|
|
media={"filename": filename, "media_id": f"mid-{mid}"},
|
|
raw={"source": "aibot"},
|
|
)
|
|
|
|
def test_file_needs_work_order(self) -> None:
|
|
phase = handle_air_group(
|
|
self._file(sender="air1", filename="航线报价.xlsx", chat="wr_air_1", mid="af0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "need_work_order")
|
|
|
|
def test_sales_excel_archived_not_quote(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="af1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._file(sender="sales1", filename="对账单.xlsx", chat="wr_air_1", mid="af2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "file_archived")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.attachments[0]["file_name"], "对账单.xlsx")
|
|
self.assertNotEqual(ticket.quote.get("source_label"), "航线附件报价")
|
|
self.assertTrue(any(f"工单{wo}的1个附件已保存到后台工单" in t for t in self._texts()))
|
|
|
|
def test_airline_excel_quotes_and_keeps_tms_id(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="af3"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
self._file(sender="air1", filename="航线报价.pdf", chat="wr_air_1", mid="af4"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "airline_file_quote")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.attachments[0]["file_name"], "航线报价.pdf")
|
|
self.assertEqual(ticket.quote.get("source_label"), "航线附件报价")
|
|
self.assertEqual(ticket.quote.get("quoteId"), "QT-AIR-1")
|
|
self.assertEqual(ticket.status, "已报价")
|
|
self.assertTrue(any(f"工单{wo}的1个附件已保存到后台工单" in t for t in self._texts()))
|
|
self.assertTrue(any("报价单已发送，请确认成交状态" in t for t in self._texts()))
|
|
|
|
def test_image_rejected(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="af5"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
phase = handle_air_group(
|
|
InboundMessage(
|
|
sender_id="air1",
|
|
message_id="af6",
|
|
content="",
|
|
msg_type="image",
|
|
chat_id="wr_air_1",
|
|
chat_type="group",
|
|
media={"filename": "现场.jpg", "media_id": "mid-img"},
|
|
raw={"source": "aibot"},
|
|
),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "file_type_rejected")
|
|
self.assertFalse(self.ledger.get_ticket(work_order_no=wo).attachments)
|
|
|
|
def test_archive_file_without_at_quotes(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=wo, chat="wr_air_1", mid="ar0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
msg = self._file(sender="air1", filename="航线报价.xlsx", chat="wr_air_1", mid="ar1")
|
|
msg = InboundMessage(
|
|
sender_id=msg.sender_id,
|
|
message_id=msg.message_id,
|
|
content=msg.content,
|
|
msg_type="file",
|
|
chat_id=msg.chat_id,
|
|
chat_type="group",
|
|
media=dict(msg.media),
|
|
raw={"source": "wshoto_archive"},
|
|
)
|
|
import agent.handlers.air_group_collab as air_mod
|
|
orig = air_mod.get_air_text_flow
|
|
|
|
def _fake():
|
|
return self.flow
|
|
|
|
air_mod.get_air_text_flow = _fake # type: ignore
|
|
try:
|
|
name = dispatch_inbound(msg, RouteDecision(intent="ordinary_text_other"))
|
|
finally:
|
|
air_mod.get_air_text_flow = orig # type: ignore
|
|
self.assertEqual(name, "air_group_collab")
|
|
ticket = self.ledger.get_ticket(work_order_no=wo)
|
|
self.assertEqual(ticket.quote.get("source_label"), "航线附件报价")
|
|
self.assertTrue(any(f"工单{wo}的1个附件已保存到后台工单" in t for t in self._texts()))
|
|
self.assertTrue(any("成交" in t or "未成交" in t or "协商" in t for t in self._texts()))
|
|
|
|
def test_archive_text_ignored_until_at(self) -> None:
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="ar2"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
msg = self._msg(sender="sales1", text="成交", chat="wr_air_1", mid="ar3")
|
|
msg = InboundMessage(
|
|
sender_id=msg.sender_id,
|
|
message_id=msg.message_id,
|
|
content=msg.content,
|
|
chat_id=msg.chat_id,
|
|
chat_type="group",
|
|
raw={"source": "wshoto_archive"},
|
|
)
|
|
import agent.handlers.air_group_collab as air_mod
|
|
orig = air_mod.get_air_text_flow
|
|
|
|
def _fake():
|
|
return self.flow
|
|
|
|
air_mod.get_air_text_flow = _fake # type: ignore
|
|
try:
|
|
name = dispatch_inbound(msg, RouteDecision(intent="ordinary_text_other"))
|
|
finally:
|
|
air_mod.get_air_text_flow = orig # type: ignore
|
|
self.assertEqual(name, "air_archive_text_ignore")
|
|
self.assertEqual(self.ledger.get_ticket(work_order_no=wo).status, "已报价")
|
|
|
|
def test_archive_at_work_order_activates(self) -> None:
|
|
wo = self._create_air()
|
|
msg = InboundMessage(
|
|
sender_id="sales1",
|
|
message_id="ar-at-1",
|
|
content=f"@询价小助手 {wo}",
|
|
chat_id="wr_air_new",
|
|
chat_type="group",
|
|
raw={"source": "wshoto_archive"},
|
|
)
|
|
import agent.handlers.air_group_collab as air_mod
|
|
import agent.policy.air_text_flow as flow_mod
|
|
|
|
orig = air_mod.get_air_text_flow
|
|
orig2 = flow_mod.get_air_text_flow
|
|
|
|
def _fake():
|
|
return self.flow
|
|
|
|
air_mod.get_air_text_flow = _fake # type: ignore
|
|
flow_mod.get_air_text_flow = _fake # type: ignore
|
|
try:
|
|
name = dispatch_inbound(msg, RouteDecision(intent="ordinary_text_other"))
|
|
finally:
|
|
air_mod.get_air_text_flow = orig # type: ignore
|
|
flow_mod.get_air_text_flow = orig2 # type: ignore
|
|
self.assertEqual(name, "air_group_collab")
|
|
self.assertTrue(any(wo in t for t in self._texts()))
|
|
|
|
def test_file_with_work_order_activates(self) -> None:
|
|
wo = self._create_air()
|
|
phase = handle_air_group(
|
|
self._file(
|
|
sender="air1",
|
|
filename="航线报价.xlsx",
|
|
chat="wr_air_1",
|
|
mid="af7",
|
|
text=wo,
|
|
),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "airline_file_quote")
|
|
self.assertEqual(self.ledger.find_by_collab_chat(chat_id="wr_air_1").work_order_no, wo)
|
|
|
|
def test_archive_same_sentence_different_msgid_sends_one_sheet(self) -> None:
|
|
wo = self._create_air()
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_dup2", mid="bot-mid-012"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "activated")
|
|
self.assertEqual(sum(1 for x in self.groups.sent if x.get("kind") == "file"), 1)
|
|
phase = handle_air_group(
|
|
InboundMessage(
|
|
sender_id="sales1",
|
|
message_id="wshoto:archive-mid-999",
|
|
content=f"@询价小助手 {wo}",
|
|
chat_id="wr_air_dup2",
|
|
chat_type="group",
|
|
raw={"source": "wshoto_archive", "msgid": "archive-mid-999"},
|
|
),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "duplicate_inbound")
|
|
self.assertEqual(sum(1 for x in self.groups.sent if x.get("kind") == "file"), 1)
|
|
|
|
def test_aibot_and_archive_same_msgid_sends_one_sheet(self) -> None:
|
|
wo = self._create_air()
|
|
import agent.handlers.air_group_collab as air_mod
|
|
import agent.policy.air_text_flow as flow_mod
|
|
|
|
orig = air_mod.get_air_text_flow
|
|
orig2 = flow_mod.get_air_text_flow
|
|
|
|
def _fake():
|
|
return self.flow
|
|
|
|
air_mod.get_air_text_flow = _fake # type: ignore
|
|
flow_mod.get_air_text_flow = _fake # type: ignore
|
|
try:
|
|
aibot = self._msg(sender="sales1", text=wo, chat="wr_air_dup", mid="same-mid-011")
|
|
name = dispatch_inbound(aibot, RouteDecision(intent="ordinary_text_other"))
|
|
self.assertEqual(name, "air_group_collab")
|
|
files_after_bot = sum(1 for x in self.groups.sent if x.get("kind") == "file")
|
|
self.assertEqual(files_after_bot, 1)
|
|
archive = InboundMessage(
|
|
sender_id="sales1",
|
|
message_id="wshoto:same-mid-011",
|
|
content=f"@询价小助手 {wo}",
|
|
chat_id="wr_air_dup",
|
|
chat_type="group",
|
|
raw={"source": "wshoto_archive", "msgid": "same-mid-011"},
|
|
)
|
|
name = dispatch_inbound(archive, RouteDecision(intent="ordinary_text_other"))
|
|
self.assertEqual(name, "air_group_collab")
|
|
self.assertEqual(sum(1 for x in self.groups.sent if x.get("kind") == "file"), 1)
|
|
finally:
|
|
air_mod.get_air_text_flow = orig # type: ignore
|
|
flow_mod.get_air_text_flow = orig2 # type: ignore
|
|
|
|
def test_sea_ticket_rejected(self) -> None:
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales1",
|
|
business_line="SEA",
|
|
facts={"起运港": "上海", "目的港": "洛杉矶", "品名": "配件", "货量": "1x40HQ", "整柜或拼柜": "整柜", "箱型箱量": "1x40HQ", "报价日期": "2026-09-16"},
|
|
)
|
|
wo = created["work_order_no"]
|
|
phase = handle_air_group(
|
|
self._msg(sender="sales1", text=wo, chat="wr_air_1", mid="sea1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
)
|
|
self.assertEqual(phase, "not_air")
|
|
|
|
def test_lock_fail_tells_operator_and_tms_reason(self) -> None:
|
|
"""锁舱失败群里要回操作人和 TMS 原因，不能被系统异常兜底盖掉。"""
|
|
wo = self._create_air()
|
|
self.ledger._cabin_lock_result = {
|
|
"ok": False,
|
|
"error": "lock_fail",
|
|
"failReason": "该航班日期不存在可锁定的配载记录",
|
|
}
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_lock_fail", mid="lf1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "system_exception")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn(f"工单{wo} 空运锁舱失败。", joined)
|
|
self.assertIn("操作人：HangXian", joined)
|
|
self.assertIn("锁舱失败的原因：该航班日期不存在可锁定的配载记录", joined)
|
|
self.assertIn("已通知IT运维排查，请稍等。", joined)
|
|
self.assertNotIn("TMS接口异常，IT运维排查中", joined)
|
|
self.assertNotIn("报价选项不存在", joined)
|
|
self.assertNotIn("与工单不匹配", joined)
|
|
|
|
def test_lock_without_quote_still_shows_tms_reason(self) -> None:
|
|
"""没有报价也可以锁舱；群里只回 TMS 原因，不要说选项不匹配。"""
|
|
wo = self._create_air(quote={}, status="询价中")
|
|
self.ledger._cabin_lock_result = {
|
|
"ok": False,
|
|
"error": "lock_fail",
|
|
"failReason": "该航班日期不存在可锁定的配载记录",
|
|
}
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_lock_no_quote", mid="lnq1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertEqual(phase, "system_exception")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("锁舱失败的原因：该航班日期不存在可锁定的配载记录", joined)
|
|
self.assertNotIn("报价选项不存在", joined)
|
|
self.assertNotIn("与工单不匹配", joined)
|
|
|
|
def test_paused_lock_retries_and_shows_tms_reason(self) -> None:
|
|
"""上次锁舱失败标了系统异常后，再锁仍要打 TMS，不能只回兜底。"""
|
|
wo = self._create_air()
|
|
self.ledger.mark_system_exception(
|
|
work_order_no=wo,
|
|
system_exception="TMS",
|
|
system_exception_reason="空运锁舱接口调用失败",
|
|
step="tms_lock",
|
|
payload={},
|
|
)
|
|
hits = {"n": 0}
|
|
orig = self.ledger.lock_cabin
|
|
|
|
def counted(**kwargs):
|
|
hits["n"] += 1
|
|
return orig(**kwargs)
|
|
|
|
self.ledger.lock_cabin = counted # type: ignore[method-assign]
|
|
self.ledger._cabin_lock_result = {
|
|
"ok": False,
|
|
"error": "lock_fail",
|
|
"failReason": "该航班日期不存在可锁定的配载记录",
|
|
}
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_lock_paused", mid="lp1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.assertNotEqual(phase, "system_exception_paused")
|
|
self.assertGreaterEqual(hits["n"], 1)
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn("锁舱失败的原因：该航班日期不存在可锁定的配载记录", joined)
|
|
self.assertNotIn("TMS接口异常，IT运维排查中", joined)
|
|
|
|
def test_release_fail_tells_operator_and_tms_reason(self) -> None:
|
|
"""释放失败同理：操作人、原因、已通知IT，不要系统异常兜底。"""
|
|
wo = self._create_air()
|
|
handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}锁舱", chat="wr_air_rl_fail", mid="rf0"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_lock",
|
|
)
|
|
self.ledger._cabin_release_result = {
|
|
"ok": False,
|
|
"error": "release_fail",
|
|
"failReason": "舱位不存在",
|
|
}
|
|
phase = handle_air_group(
|
|
self._msg(sender="air1", text=f"{wo}释放舱位", chat="wr_air_rl_fail", mid="rf1"),
|
|
flow=self.flow,
|
|
ledger=self.ledger,
|
|
injected_intent="air_release",
|
|
)
|
|
self.assertEqual(phase, "system_exception")
|
|
joined = "\n".join(self._texts())
|
|
self.assertIn(f"工单{wo} 空运释放舱位失败。", joined)
|
|
self.assertIn("操作人：HangXian", joined)
|
|
self.assertIn("释放舱位失败的原因：舱位不存在", joined)
|
|
self.assertIn("已通知IT运维排查，请稍等。", joined)
|
|
self.assertNotIn("TMS接口异常，IT运维排查中", joined)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|