421 lines
16 KiB
Python
421 lines
16 KiB
Python
"""多段群内：按岗位归段报价，全部出单才整票成交跟进。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
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.ledger.memory_ledger import MemoryLedger
|
|
from agent.policy.multi_group_ops import deal_text, handle_multi_inbound
|
|
from agent.policy.multi_segments import Segment
|
|
|
|
|
|
class _Client:
|
|
def __init__(self) -> None:
|
|
self.texts: list[str] = []
|
|
self.cards: list[dict] = []
|
|
self.files: list[str] = []
|
|
|
|
def send_group(self, chat_id: str, content: str, mention_userids: list[str] | None = None) -> dict:
|
|
_ = chat_id, mention_userids
|
|
self.texts.append(content)
|
|
return {"ok": True}
|
|
|
|
def send_group_card(self, chat_id: str, template_card: dict) -> dict:
|
|
_ = chat_id
|
|
self.cards.append(template_card)
|
|
return {"ok": True}
|
|
|
|
def send_group_file(self, *, chat_id: str, filename: str, filepath: str = "", file_b64: str = "") -> dict:
|
|
_ = chat_id, filepath, file_b64
|
|
self.files.append(filename)
|
|
return {"ok": True}
|
|
|
|
|
|
class _Flow:
|
|
def __init__(self, ledger: MemoryLedger, client: _Client) -> None:
|
|
self.ledger = ledger
|
|
self._ledger = ledger
|
|
self._group_client = client
|
|
|
|
def group_client(self) -> _Client:
|
|
return self._group_client
|
|
|
|
|
|
def _ticket(ledger: MemoryLedger) -> object:
|
|
created = ledger.create_ticket(
|
|
sender_id="sales",
|
|
business_line="MULTI",
|
|
facts={
|
|
"segments_json": json.dumps(
|
|
[
|
|
{"index": 1, "mode": "SEA", "facts": {"起运港": "广州港", "目的港": "巴生港"}},
|
|
{"index": 2, "mode": "LAND", "facts": {"线路类别": "国内长途/零担"}},
|
|
],
|
|
ensure_ascii=False,
|
|
)
|
|
},
|
|
)
|
|
ledger.set_sea_staff(
|
|
[{"wecomId": "sea1", "name": "海产", "roleCode": "sea", "status": "active", "routes": "海运·全部线路"}]
|
|
)
|
|
ledger.set_land_staff(
|
|
[{"wecomId": "land1", "name": "陆产", "roleCode": "land", "status": "active", "routes": "国内长途/零担"}]
|
|
)
|
|
ledger.upsert_staff({"wecomId": "sales", "name": "销售甲", "roleCode": "sales", "status": "active"})
|
|
no = created["work_order_no"]
|
|
ledger.bind_collab_group(
|
|
work_order_no=no,
|
|
chat_id="room-1",
|
|
member_ids=["sales", "sea1", "land1"],
|
|
product_names=["海产", "陆产"],
|
|
product_ids=["sea1", "land1"],
|
|
)
|
|
return ledger.get_ticket(work_order_no=no)
|
|
|
|
|
|
class QuoteTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.ledger = MemoryLedger()
|
|
self.client = _Client()
|
|
self.flow = _Flow(self.ledger, self.client)
|
|
self.ticket = _ticket(self.ledger)
|
|
self.sheets: list[str] = []
|
|
|
|
def sheet(self, **kwargs: object) -> dict:
|
|
self.sheets.append(str(kwargs.get("business_line") or ""))
|
|
return {"ok": True}
|
|
|
|
def test_one_segment_quote_does_not_open_deal(self) -> None:
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ticket,
|
|
chat_id="room-1",
|
|
sender_id="sea1",
|
|
text="海运费 100CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "segment_quoted")
|
|
self.assertEqual(self.sheets, ["SEA"])
|
|
self.assertFalse(self.client.cards)
|
|
self.assertTrue(any("其他段报完" in text for text in self.client.texts))
|
|
|
|
def test_both_quotes_one_deal_without_sum(self) -> None:
|
|
handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ledger.get_ticket(work_order_no=self.ticket.work_order_no),
|
|
chat_id="room-1",
|
|
sender_id="sea1",
|
|
text="海运费 100CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ledger.get_ticket(work_order_no=self.ticket.work_order_no),
|
|
chat_id="room-1",
|
|
sender_id="land1",
|
|
text="陆运费 80CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "multi_deal_sent")
|
|
self.assertEqual(self.sheets, ["SEA", "LAND"])
|
|
body = self.client.texts[-1]
|
|
self.assertIn("段1：海运", body)
|
|
self.assertIn("段2：陆运", body)
|
|
self.assertNotIn("180", body)
|
|
rows = self.client.cards[-1]["horizontal_content_list"]
|
|
self.assertEqual([row["keyname"] for row in rows], ["段1海运", "段2陆运"])
|
|
joined = " ".join(row["value"] for row in rows)
|
|
self.assertNotIn("180", joined)
|
|
|
|
def test_land_person_cannot_fill_sea_segment(self) -> None:
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ticket,
|
|
chat_id="room-1",
|
|
sender_id="land1",
|
|
text="海运费 100CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "segment_quoted")
|
|
self.assertEqual(self.sheets, ["LAND"])
|
|
|
|
def test_requote_refreshes_deal(self) -> None:
|
|
handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ticket,
|
|
chat_id="room-1",
|
|
sender_id="sea1",
|
|
text="海运费 100CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ledger.get_ticket(work_order_no=self.ticket.work_order_no),
|
|
chat_id="room-1",
|
|
sender_id="land1",
|
|
text="陆运费 80CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ledger.get_ticket(work_order_no=self.ticket.work_order_no),
|
|
chat_id="room-1",
|
|
sender_id="sea1",
|
|
text="海运费 120CNY",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "multi_deal_sent")
|
|
self.assertIn("120", self.client.texts[-1])
|
|
self.assertNotIn("100", deal_text(
|
|
work_order_no=self.ticket.work_order_no,
|
|
segments=[Segment(index=1, mode="SEA"), Segment(index=2, mode="LAND")],
|
|
quotes={"SEA": {"total": "120", "currency": "CNY"}, "LAND": {"total": "80", "currency": "CNY"}},
|
|
))
|
|
|
|
def test_work_order_in_group_sends_brief_and_templates(self) -> None:
|
|
no = self.ticket.work_order_no
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=self.ticket,
|
|
chat_id="room-9",
|
|
sender_id="sales",
|
|
text=no,
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "multi_activated")
|
|
self.assertEqual(self.sheets, ["SEA", "LAND"])
|
|
brief = self.client.texts[-1]
|
|
self.assertIn("段1：海运", brief)
|
|
self.assertIn("段2：陆运", brief)
|
|
self.assertNotIn("TMS报价", brief)
|
|
self.assertNotIn("其他段报完", brief)
|
|
self.assertEqual(self.ledger.get_ticket(work_order_no=no).collab_chat_id, "room-9")
|
|
|
|
def test_at_bot_work_order_activates_not_quote(self) -> None:
|
|
self.ledger.upsert_staff(
|
|
{"wecomId": "air1", "name": "航线甲", "roleCode": "air", "status": "active"}
|
|
)
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales",
|
|
business_line="MULTI",
|
|
facts={
|
|
"segments_json": json.dumps(
|
|
[
|
|
{
|
|
"index": 1,
|
|
"mode": "LAND",
|
|
"facts": {
|
|
"运输类型": "国内运输拼车",
|
|
"起运港": "广州",
|
|
"目的港": "深圳",
|
|
"品名": "LMM振动测试定位治具",
|
|
"体积": "100CBM",
|
|
"毛重": "1.5kg",
|
|
},
|
|
},
|
|
{
|
|
"index": 2,
|
|
"mode": "AIR",
|
|
"facts": {
|
|
"目的港": "CRK",
|
|
"品名": "LMM振动测试定位治具",
|
|
"件数": "1件",
|
|
"毛重": "1.5",
|
|
"体积": "100CBM",
|
|
"包装方式": "托盘",
|
|
"起运港": "ZUH",
|
|
"贸易条款": "DAP",
|
|
},
|
|
},
|
|
],
|
|
ensure_ascii=False,
|
|
)
|
|
},
|
|
)
|
|
no = created["work_order_no"]
|
|
ticket = self.ledger.get_ticket(work_order_no=no)
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=ticket,
|
|
chat_id="air-room",
|
|
sender_id="sales",
|
|
text=f"@询价小助手 {no}",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "multi_activated")
|
|
self.assertEqual(self.sheets, ["LAND", "AIR"])
|
|
brief = self.client.texts[0]
|
|
self.assertIn("@销售甲", brief)
|
|
self.assertIn("@陆产", brief)
|
|
self.assertIn("@航线甲", brief)
|
|
self.assertIn("段1：陆运", brief)
|
|
self.assertIn("运输类型（必填）：国内运输拼车", brief)
|
|
self.assertIn("始发站（必填）：广州", brief)
|
|
self.assertIn("段2：空运", brief)
|
|
self.assertIn("目的地（必填）：CRK", brief)
|
|
self.assertIn("贸易条款（非必填）：DAP", brief)
|
|
self.assertNotIn("其他段报完", "\n".join(self.client.texts))
|
|
self.assertNotIn("询价字段：", brief)
|
|
|
|
def test_http_snapshot_activation_mentions_real_sales_not_job_word(self) -> None:
|
|
"""主账快照没带回销售企微号时，群里仍要 @ 具体人，不能写岗位词「销售」。"""
|
|
self.ledger.upsert_staff(
|
|
{"wecomId": "WuJiLin", "name": "吴吉林", "roleCode": "sales", "status": "active"}
|
|
)
|
|
self.ledger.upsert_staff(
|
|
{"wecomId": "air1", "name": "航线甲", "roleCode": "air", "status": "active"}
|
|
)
|
|
created = self.ledger.create_ticket(
|
|
sender_id="WuJiLin",
|
|
business_line="MULTI",
|
|
facts={
|
|
"segments_json": json.dumps(
|
|
[
|
|
{
|
|
"index": 1,
|
|
"mode": "LAND",
|
|
"facts": {
|
|
"运输类型": "国内运输拼车",
|
|
"起运港": "广州",
|
|
"目的港": "深圳",
|
|
"品名": "治具",
|
|
},
|
|
},
|
|
{"index": 2, "mode": "AIR", "facts": {"起运港": "ZUH", "目的港": "CRK", "品名": "治具"}},
|
|
],
|
|
ensure_ascii=False,
|
|
)
|
|
},
|
|
)
|
|
no = created["work_order_no"]
|
|
real_get = self.ledger.get_ticket
|
|
|
|
def http_snap(*, work_order_no: str):
|
|
ticket = real_get(work_order_no=work_order_no)
|
|
if ticket is None:
|
|
return None
|
|
return {
|
|
"ok": True,
|
|
"found": True,
|
|
"work_order_no": ticket.work_order_no,
|
|
"business_line": "MULTI",
|
|
"facts": dict(ticket.facts),
|
|
"sales_name": "销售",
|
|
"salesName": "销售",
|
|
"collab_chat_id": ticket.collab_chat_id,
|
|
}
|
|
|
|
self.ledger.get_ticket = http_snap # type: ignore[method-assign]
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=http_snap(work_order_no=no),
|
|
chat_id="live-room",
|
|
sender_id="WuJiLin",
|
|
text=f"@询价小助手 {no}",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "multi_activated")
|
|
brief = self.client.texts[0]
|
|
self.assertIn("@吴吉林", brief)
|
|
self.assertNotRegex(brief, r"@销售(?:@|\n|$)")
|
|
|
|
def test_land_sheet_matches_when_route_missing_and_no_default(self) -> None:
|
|
"""多段陆运常缺线路类别；后台陆运模板只按线路关键词时，仍要发出陆运表。"""
|
|
self.ledger.set_quote_templates(
|
|
[
|
|
{
|
|
"templateId": "air-default",
|
|
"name": "空运默认",
|
|
"bizType": "空运",
|
|
"keywords": [],
|
|
"isDefault": True,
|
|
},
|
|
{
|
|
"templateId": "land-lcl",
|
|
"name": "陆运国内运输拼车报价单",
|
|
"bizType": "陆运",
|
|
"keywords": ["国内拼车", "国内零担"],
|
|
"isDefault": False,
|
|
},
|
|
]
|
|
)
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales",
|
|
business_line="MULTI",
|
|
facts={
|
|
"segments_json": json.dumps(
|
|
[
|
|
{
|
|
"index": 1,
|
|
"mode": "LAND",
|
|
"facts": {
|
|
"运输类型": "国内运输拼车",
|
|
"起运港": "广州",
|
|
"目的港": "深圳",
|
|
"品名": "治具",
|
|
"体积": "100CBM",
|
|
"毛重": "1.5kg",
|
|
},
|
|
},
|
|
{
|
|
"index": 2,
|
|
"mode": "AIR",
|
|
"facts": {"起运港": "ZUH", "目的港": "CRK", "品名": "治具"},
|
|
},
|
|
],
|
|
ensure_ascii=False,
|
|
)
|
|
},
|
|
)
|
|
no = created["work_order_no"]
|
|
ticket = self.ledger.get_ticket(work_order_no=no)
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=ticket,
|
|
chat_id="sheet-room",
|
|
sender_id="sales",
|
|
text=f"@询价小助手 {no}",
|
|
)
|
|
self.assertEqual(phase, "multi_activated")
|
|
self.assertEqual(len(self.client.files), 2)
|
|
self.assertTrue(self.ledger.last_render.get("ok"))
|
|
|
|
def test_lock_on_air_segment_does_not_wait_for_other_quotes(self) -> None:
|
|
created = self.ledger.create_ticket(
|
|
sender_id="sales",
|
|
business_line="MULTI",
|
|
facts={
|
|
"segments_json": json.dumps(
|
|
[
|
|
{"index": 1, "mode": "SEA", "facts": {"起运港": "广州港"}},
|
|
{"index": 2, "mode": "AIR", "facts": {"目的港": "CRK"}},
|
|
],
|
|
ensure_ascii=False,
|
|
)
|
|
},
|
|
)
|
|
self.ledger.upsert_staff({"wecomId": "air1", "name": "航线甲", "roleCode": "air", "status": "active"})
|
|
no = created["work_order_no"]
|
|
ticket = self.ledger.get_ticket(work_order_no=no)
|
|
phase = handle_multi_inbound(
|
|
flow=self.flow,
|
|
ticket=ticket,
|
|
chat_id="air-room",
|
|
sender_id="air1",
|
|
text=f"{no}锁舱",
|
|
sheet_fn=self.sheet,
|
|
)
|
|
self.assertEqual(phase, "lock_ok")
|
|
self.assertFalse(any("其他段报完" in text for text in self.client.texts))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|