260 lines
9.7 KiB
Python
260 lines
9.7 KiB
Python
"""
|
|
入站选路：陆运书签/本句陆运交给 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 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_retry_misrouted_as_other(self) -> None:
|
|
from tests.test_land_text_inquiry import CountingLedger
|
|
|
|
class Flaky(CountingLedger):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.query_calls = 0
|
|
|
|
def query_tms(self, **kwargs):
|
|
self.query_calls += 1
|
|
if self.query_calls == 1:
|
|
return {
|
|
"ok": False,
|
|
"classification": "TMS_TECH_FAILURE",
|
|
"has_price": False,
|
|
"quote": {},
|
|
}
|
|
return super().query_tms(**kwargs)
|
|
|
|
land = reset_land_text_flow_for_test()
|
|
land.ledger = Flaky()
|
|
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.assertEqual(get_land_text_flow().session_of("ld5").phase, "tms_tech")
|
|
wo = get_land_text_flow().session_of("ld5").work_order_no
|
|
name = dispatch_inbound(
|
|
InboundMessage(sender_id="ld5", message_id="m-l5d", content="确定"),
|
|
RouteDecision(intent="ordinary_text_other"),
|
|
)
|
|
self.assertEqual(name, "ordinary_text_inquiry")
|
|
sess = get_land_text_flow().session_of("ld5")
|
|
self.assertEqual(sess.phase, "wait_collab")
|
|
self.assertEqual(sess.work_order_no, wo)
|
|
last = list(self.store._outbox.values())[-1].content
|
|
self.assertNotIn("你好，我是询价机器人", 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")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|