""" 入站选路：陆运书签/本句陆运交给 LandTextInquiryFlow；海运立即建单不变。 """ 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("YTD_ENV", "test") os.environ.setdefault("REDIS_BACKEND", "memory") os.environ.setdefault("CHECKPOINT_BACKEND", "memory") os.environ.setdefault("MESSAGE_STORE_BACKEND", "memory") os.environ.setdefault("LLM_DATA_USAGE_CONFIRMED", "false") os.environ.setdefault("LLM_ALLOW_NETWORK", "false") os.environ.setdefault("LEDGER_BACKEND", "memory") from agent.channel.queue import MemoryMessageStore, get_message_store from agent.channel.wecom.models import InboundMessage from agent.handlers.text_inquiry import handle_text_inquiry from agent.policy.air_text_flow import reset_air_text_flow_for_test from agent.policy.land_text_flow import get_land_text_flow, reset_land_text_flow_for_test from agent.policy.sea_text_flow import get_sea_text_flow, reset_sea_text_flow_for_test from agent.routing.decision import RouteDecision from agent.routing.dispatch import dispatch_inbound from tests.test_sea_text_inquiry import COMPLETE_SEA class LandDispatchTests(unittest.TestCase): def setUp(self) -> None: reset_sea_text_flow_for_test() reset_land_text_flow_for_test() reset_air_text_flow_for_test() self.store = MemoryMessageStore() def _outbox_text(self) -> str: return "\n".join(i.content for i in self.store._outbox.values()) def test_handler_routes_land_to_option_sheet(self) -> None: msg = InboundMessage(sender_id="ld1", message_id="m-land-1", content="陆运询价") phase = handle_text_inquiry( msg, injected_facts={"起运港": "广州", "目的港": "深圳", "品名": "衣服"}, injected_mode="LAND", store=self.store, ) self.assertEqual(phase, "need_land_options") body = self._outbox_text() self.assertIn("运输类型", body) self.assertIn("线路类别", body) self.assertIn("运输分类", body) self.assertNotIn("请补充陆运类型：", body) def test_air_owner_forwards_land_to_option_sheet(self) -> None: from agent.policy.air_text_flow import get_air_text_flow replies: list[str] = [] phase = get_air_text_flow().on_text( sender_id="ld-air", text="陆运询价", reply=lambda t, extra=None: replies.append(t) or "", injected_facts={"起运港": "广州", "目的港": "深圳", "品名": "衣服"}, injected_mode="LAND", ) self.assertEqual(phase, "need_land_options") text = "\n".join(replies) self.assertIn("线路类别", text) self.assertIn("运输分类", text) self.assertNotIn("请补充陆运类型：", text) def test_land_session_survives_confirm(self) -> None: msg1 = InboundMessage(sender_id="ld2", message_id="m-l2a", content="陆运") facts = { "起运港": "广州", "目的港": "深圳", "品名": "衣服", "体积": "100CBM", "毛重": "50kg", } phase = handle_text_inquiry( msg1, injected_facts=facts, injected_mode="LAND", store=self.store, ) # 补选项走同一 sender msg2 = InboundMessage( sender_id="ld2", message_id="m-l2b", content="国内运输拼车 国内长途/零担 急件", ) phase = handle_text_inquiry( msg2, injected_facts=facts, injected_mode="LAND", store=self.store, ) self.assertEqual(phase, "wait_confirm") msg3 = InboundMessage(sender_id="ld2", message_id="m-l2c", content="确定") phase = handle_text_inquiry(msg3, store=self.store) self.assertEqual(phase, "wait_collab") def test_confirm_misrouted_as_other_still_creates(self) -> None: store = get_message_store() facts = { "起运港": "广州", "目的港": "深圳", "品名": "衣服", "体积": "100CBM", "毛重": "50kg", } handle_text_inquiry( InboundMessage(sender_id="ld3", message_id="m-l3a", content="陆运"), injected_facts=facts, injected_mode="LAND", store=store, ) handle_text_inquiry( InboundMessage( sender_id="ld3", message_id="m-l3b", content="国内运输拼车 国内长途/零担 急件", ), injected_facts=facts, injected_mode="LAND", store=store, ) self.assertEqual(get_land_text_flow().session_of("ld3").phase, "wait_confirm") name = dispatch_inbound( InboundMessage(sender_id="ld3", message_id="m-l3c", content="确定"), RouteDecision(intent="ordinary_text_other"), ) self.assertEqual(name, "ordinary_text_inquiry") sess = get_land_text_flow().session_of("ld3") self.assertEqual(sess.phase, "wait_collab") self.assertTrue((sess.work_order_no or "").startswith("WO")) bodies = "\n".join(i.content for i in store._outbox.values()) self.assertIn(sess.work_order_no, bodies) last = list(store._outbox.values())[-1].content self.assertNotIn("你好，我是询价机器人", last) def test_confirm_after_memory_lost_still_creates(self) -> None: store = get_message_store() facts = { "起运港": "广州", "目的港": "深圳", "品名": "衣服", "体积": "100CBM", "毛重": "50kg", } handle_text_inquiry( InboundMessage(sender_id="ld4", message_id="m-l4a", content="陆运"), injected_facts=facts, injected_mode="LAND", store=store, ) handle_text_inquiry( InboundMessage( sender_id="ld4", message_id="m-l4b", content="国内运输拼车 国内长途/零担 急件", ), injected_facts=facts, injected_mode="LAND", store=store, ) self.assertEqual(get_land_text_flow().session_of("ld4").phase, "wait_confirm") reset_land_text_flow_for_test() self.assertIsNone(get_land_text_flow().session_of("ld4")) name = dispatch_inbound( InboundMessage(sender_id="ld4", message_id="m-l4c", content="确定"), RouteDecision(intent="ordinary_text_other"), ) self.assertEqual(name, "ordinary_text_inquiry") sess = get_land_text_flow().session_of("ld4") self.assertEqual(sess.phase, "wait_collab") self.assertTrue((sess.work_order_no or "").startswith("WO")) last = list(store._outbox.values())[-1].content self.assertNotIn("你好，我是询价机器人", last) def test_tms_tech_fail_pauses_until_admin_retry(self) -> None: from agent.policy.inquiry_copy import SYS_EXC_TMS_FALLBACK from tests.test_land_text_inquiry import CountingLedger class AlwaysFail(CountingLedger): def __init__(self) -> None: super().__init__() self.query_calls = 0 def query_tms(self, **kwargs): self.query_calls += 1 return { "ok": False, "classification": "TMS_TECH_FAILURE", "has_price": False, "quote": {}, } land = reset_land_text_flow_for_test() ledger = AlwaysFail() land.ledger = ledger get_sea_text_flow().ledger = ledger from agent.policy.air_text_flow import get_air_text_flow get_air_text_flow().ledger = ledger facts = { "起运港": "广州", "目的港": "深圳", "品名": "衣服", "体积": "100CBM", "毛重": "50kg", } handle_text_inquiry( InboundMessage(sender_id="ld5", message_id="m-l5a", content="陆运"), injected_facts=facts, injected_mode="LAND", store=self.store, ) handle_text_inquiry( InboundMessage( sender_id="ld5", message_id="m-l5b", content="国内运输拼车 国内长途/零担 急件", ), injected_facts=facts, injected_mode="LAND", store=self.store, ) handle_text_inquiry( InboundMessage(sender_id="ld5", message_id="m-l5c", content="确定"), store=self.store, ) self.assertGreaterEqual(ledger.query_calls, 3) wo = ledger.last_created_no ticket = ledger.get_ticket(work_order_no=wo) self.assertEqual(ticket.system_exception, "TMS异常") bodies = "\n".join(i.content for i in self.store._outbox.values()) self.assertIn(SYS_EXC_TMS_FALLBACK, bodies) self.assertIsNone(get_land_text_flow().session_of("ld5")) name = dispatch_inbound( InboundMessage(sender_id="ld5", message_id="m-l5d", content="确定"), RouteDecision(intent="ordinary_text_other"), ) self.assertEqual(name, "ordinary_text_other") sess = get_land_text_flow().session_of("ld5") if sess is not None: self.assertNotEqual(sess.phase, "wait_collab") handle_text_inquiry( InboundMessage(sender_id="ld5", message_id="m-l5e", content=f"看看 {wo}"), store=self.store, ) last = list(self.store._outbox.values())[-1].content self.assertIn(SYS_EXC_TMS_FALLBACK, last) def test_sea_still_creates_immediately(self) -> None: msg = InboundMessage(sender_id="s-sea", message_id="m-sea-keep", content="海运询价") phase = handle_text_inquiry( msg, injected_facts=COMPLETE_SEA, injected_mode="SEA", store=self.store, ) self.assertEqual(phase, "wait_collab") def _land_to_quoted(self, sender_id: str) -> None: store = get_message_store() facts = { "起运港": "广州", "目的港": "深圳", "品名": "衣服", "体积": "100CBM", "毛重": "50kg", } handle_text_inquiry( InboundMessage(sender_id=sender_id, message_id=f"{sender_id}-a", content="陆运"), injected_facts=facts, injected_mode="LAND", store=store, ) handle_text_inquiry( InboundMessage( sender_id=sender_id, message_id=f"{sender_id}-b", content="国内运输拼车 国内长途/零担 急件", ), injected_facts=facts, injected_mode="LAND", store=store, ) handle_text_inquiry( InboundMessage(sender_id=sender_id, message_id=f"{sender_id}-c", content="确定"), store=store, ) self.assertEqual(get_land_text_flow().session_of(sender_id).phase, "wait_collab") def test_quoted_rise_percent_not_first_chat(self) -> None: """已出 TMS 卡后，A 判成调价就必须追问费用名，不能回开场白。""" store = get_message_store() self._land_to_quoted("ld-rise") name = dispatch_inbound( InboundMessage(sender_id="ld-rise", message_id="m-rise", content="报价上升10%"), RouteDecision(intent="quote_adjust"), ) self.assertEqual(name, "quote_adjust") last = list(store._outbox.values())[-1].content self.assertNotIn("你好，我是询价机器人", last) self.assertIn("请说明要调哪一项费用", last) def test_quoted_land_wins_over_stale_sea(self) -> None: """旧海运书签不能把陆运调价抢走。""" store = get_message_store() handle_text_inquiry( InboundMessage(sender_id="ld-mix", message_id="m-mix-sea", content="海运"), injected_facts=COMPLETE_SEA, injected_mode="SEA", store=store, ) self._land_to_quoted("ld-mix") name = dispatch_inbound( InboundMessage(sender_id="ld-mix", message_id="m-mix-rise", content="报价上升10%"), RouteDecision(intent="quote_adjust"), ) self.assertEqual(name, "quote_adjust") from agent.handlers.text_inquiry import _pick_text_flow self.assertIs( _pick_text_flow(sender_id="ld-mix", text="报价上升10%", prefer_quoted=True), get_land_text_flow(), ) last = list(store._outbox.values())[-1].content self.assertNotIn("你好，我是询价机器人", last) self.assertIn("请说明要调哪一项费用", last) def test_sea_field_supplement_not_stolen_by_stale_land(self) -> None: """ 旧陆运选项草稿不能抢走海运补问。 测服：先陆运缺选项，再发海运缺贸易条款/运输分类，下一句只补这两项， 不能回陆运「参数提取不全」或填写样例图。 """ store = get_message_store() handle_text_inquiry( InboundMessage(sender_id="mix-sea", message_id="m-land-old", content="陆运询价"), injected_facts={"起运港": "广州", "目的港": "深圳", "品名": "衣服"}, injected_mode="LAND", store=store, ) self.assertEqual(get_land_text_flow().session_of("mix-sea").phase, "need_land_options") incomplete = { "起运港": "上海", "目的港": "洛杉矶", "品名": "普货", "货量": "20吨", "整柜或拼柜": "整柜", "箱型箱量": "1x40HQ", } phase = handle_text_inquiry( InboundMessage(sender_id="mix-sea", message_id="m-sea-ask", content="海运询价"), injected_facts=incomplete, injected_mode="SEA", store=store, ) self.assertEqual(phase, "clarify") self.assertEqual(get_sea_text_flow().session_of("mix-sea").phase, "clarify") self.assertIsNone(get_land_text_flow().session_of("mix-sea")) before = len(store._outbox) phase = handle_text_inquiry( InboundMessage( sender_id="mix-sea", message_id="m-sea-fill", content="贸易条款：FOB 运输分类：其他", ), injected_facts={"贸易条款": "FOB", "运输分类": "其他"}, store=store, ) self.assertNotEqual(phase, "need_land_options") self.assertEqual(get_sea_text_flow().session_of("mix-sea").business_line, "SEA") added = "\n".join(i.content for i in list(store._outbox.values())[before:]) self.assertNotIn("参数提取不全", added) self.assertNotIn("填写样例", added) self.assertNotIn("运输类型", added) def test_clarify_short_reply_misrouted_as_other_stays_on_inquiry(self) -> None: """补问阶段回「111」即使被标闲聊，也不能发开场白。""" store = get_message_store() facts = { "起运港": "广州", "目的港": "深圳", "运输类型": "国内运输拼车", "线路类别": "国内长途/零担", "运输分类": "急件", } phase = handle_text_inquiry( InboundMessage(sender_id="ld-111", message_id="m-111a", content="陆运广州到深圳"), injected_facts=facts, injected_mode="LAND", store=store, ) self.assertEqual(phase, "clarify") name = dispatch_inbound( InboundMessage(sender_id="ld-111", message_id="m-111b", content="111"), RouteDecision(intent="ordinary_text_other"), ) self.assertEqual(name, "ordinary_text_inquiry") sess = get_land_text_flow().session_of("ld-111") self.assertIsNotNone(sess) self.assertNotEqual(sess.phase, "") last = list(store._outbox.values())[-1].content self.assertNotIn("你好，我是询价机器人", last) self.assertNotIn("请直接描述客户需求", last) if __name__ == "__main__": unittest.main()