237 lines
10 KiB
Python
237 lines
10 KiB
Python
"""
|
|
陆运选项单测：三项各自必填、不联动；中港拆进运输类型；补问只发复制清单，不再发填写样例图。
|
|
|
|
不连企微、主账、TMS。
|
|
"""
|
|
|
|
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.schema.land_options import (
|
|
harvest_land_options,
|
|
looks_like_confirm,
|
|
option_prompt,
|
|
option_prompt_needs_sample,
|
|
sample_pair_allowed,
|
|
validate_land_options,
|
|
)
|
|
|
|
|
|
class LandOptionsTest(unittest.TestCase):
|
|
def test_any_type_route_combo_ok(self):
|
|
got = validate_land_options(
|
|
{
|
|
"运输类型": "国内运输拼车",
|
|
"线路类别": "东南亚",
|
|
"运输分类": "急件",
|
|
}
|
|
)
|
|
self.assertTrue(got["ok"])
|
|
self.assertEqual(got["normalized"]["运输类型"], "国内运输拼车")
|
|
self.assertEqual(got["normalized"]["线路类别"], "东南亚")
|
|
|
|
def test_bare_hk_does_not_fill_type_or_route(self):
|
|
cur = harvest_land_options("中港 急件")
|
|
self.assertFalse(cur.get("运输类型"))
|
|
self.assertFalse(cur.get("线路类别"))
|
|
self.assertEqual(cur.get("运输分类"), "急件")
|
|
|
|
def test_hk_full_and_lcl_types(self):
|
|
self.assertEqual(
|
|
harvest_land_options("中港整车")["运输类型"], "中港整车"
|
|
)
|
|
self.assertEqual(
|
|
harvest_land_options("中港零担/集拼 东南亚")["运输类型"],
|
|
"中港零担/集拼",
|
|
)
|
|
self.assertEqual(
|
|
harvest_land_options("中港零担")["运输类型"], "中港零担/集拼"
|
|
)
|
|
|
|
def test_new_class_names(self):
|
|
cur = harvest_land_options(
|
|
"普件(东南亚、中港常用线路/常规车型)"
|
|
)
|
|
self.assertEqual(
|
|
cur["运输分类"], "普件(东南亚、中港常用线路/常规车型)"
|
|
)
|
|
self.assertFalse(
|
|
harvest_land_options("普件，常用线路/常规车型").get("运输分类")
|
|
)
|
|
|
|
def test_no_hk_load_field(self):
|
|
got = validate_land_options(
|
|
{
|
|
"运输类型": "中港整车",
|
|
"线路类别": "东南亚",
|
|
"运输分类": "急件",
|
|
}
|
|
)
|
|
self.assertTrue(got["ok"])
|
|
self.assertNotIn("中港装载方式", got["missing"])
|
|
self.assertFalse(got["normalized"].get("中港装载方式"))
|
|
|
|
def test_followup_route_class_keeps_type(self):
|
|
cur = harvest_land_options(
|
|
"国内城配/拖车/打包；普件(东南亚、中港常用线路/常规车型)",
|
|
current={"运输类型": "国内运输拼车"},
|
|
)
|
|
self.assertEqual(cur["运输类型"], "国内运输拼车")
|
|
self.assertEqual(cur["线路类别"], "国内城配/拖车/打包")
|
|
self.assertEqual(
|
|
cur["运输分类"], "普件(东南亚、中港常用线路/常规车型)"
|
|
)
|
|
|
|
def test_model_land_does_not_wipe_type(self):
|
|
dropped = harvest_land_options(
|
|
"国内城配/拖车/打包；急件",
|
|
current={"运输类型": "陆运"},
|
|
)
|
|
self.assertFalse(dropped.get("运输类型"))
|
|
cur = harvest_land_options(
|
|
"国内城配/拖车/打包；急件",
|
|
current={"运输类型": "国内运输拼车"},
|
|
)
|
|
junk = harvest_land_options("陆运", current=cur)
|
|
self.assertEqual(junk["运输类型"], "国内运输拼车")
|
|
self.assertEqual(junk["线路类别"], "国内城配/拖车/打包")
|
|
cur = harvest_land_options("广州到深圳国内拼车衣服100方")
|
|
self.assertEqual(cur["运输类型"], "国内运输拼车")
|
|
self.assertFalse(cur.get("线路类别"))
|
|
|
|
def test_full_prompt_when_empty(self):
|
|
text = option_prompt(current={})
|
|
self.assertIn("Step1: 请复制以下内容，粘贴到输入框；", text)
|
|
self.assertIn("Step2:在粘贴内容的每个字段的冒号后，回复数字或文字，告诉我对应字段的值：", text)
|
|
self.assertIn("---请复制以下内容---\n运输类型（必填）：", text)
|
|
self.assertIn("运输类型（必填）：\n1.国内运输拼车", text)
|
|
self.assertIn("6.中港零担/集拼", text)
|
|
self.assertIn("线路类别（必填）：\n1.1国内长途/零担", text)
|
|
self.assertIn("6.1中港/中亚/中欧", text)
|
|
self.assertIn("运输分类（必填）：\n1.普件(常用线路/常规车型)", text)
|
|
self.assertIn("2.普件(东南亚、中港常用线路/常规车型)", text)
|
|
self.assertIn("5.急件", text)
|
|
self.assertNotIn("请一次选择：运输类型", text)
|
|
self.assertNotIn("可选择回复", text)
|
|
self.assertNotIn("填写样例见图", text)
|
|
self.assertNotIn("国内运输整车 + 国内城配/拖车/打包", text)
|
|
self.assertNotIn("中港装载方式", text)
|
|
|
|
def test_copy_alias_lists_new_types(self):
|
|
from agent.policy import inquiry_copy as copy
|
|
|
|
text = copy.ASK_LAND_SUBTYPE
|
|
self.assertIn("中港整车", text)
|
|
self.assertIn("Step1: 请复制以下内容，粘贴到输入框；", text)
|
|
self.assertIn("---请复制以下内容---", text)
|
|
self.assertNotIn("请补充陆运类型：国内运输拼车", text)
|
|
|
|
def test_sample_table_when_route_missing(self):
|
|
text = option_prompt(current={"运输类型": "国内运输拼车"})
|
|
self.assertIn("Step1: 请复制以下内容，粘贴到输入框；", text)
|
|
self.assertIn("---请复制以下内容---\n线路类别（必填）：", text)
|
|
self.assertIn("线路类别（必填）：\n1.1国内长途/零担", text)
|
|
self.assertIn("运输分类（必填）：\n1.普件", text)
|
|
self.assertFalse(option_prompt_needs_sample(current={"运输类型": "国内运输拼车"}))
|
|
self.assertFalse(option_prompt_needs_sample(current={}))
|
|
self.assertNotIn("国内运输拼车 + 国内长途/零担", text)
|
|
self.assertNotIn("请一次选择：运输类型", text)
|
|
self.assertNotIn("运输类型（必填）", text)
|
|
|
|
def test_sample_pairs_go_to_image_not_text(self):
|
|
text = option_prompt(current={})
|
|
self.assertIn("Step1: 请复制以下内容，粘贴到输入框；", text)
|
|
self.assertIn("---请复制以下内容---", text)
|
|
self.assertNotIn("填写样例见图", text)
|
|
self.assertNotIn("填写样例", text)
|
|
self.assertNotIn("国内运输拼车 + 国内长途/零担", text)
|
|
|
|
def test_no_sample_when_only_class_missing(self):
|
|
text = option_prompt(
|
|
current={
|
|
"运输类型": "国内运输拼车",
|
|
"线路类别": "东南亚",
|
|
}
|
|
)
|
|
self.assertIn("运输分类（必填）", text)
|
|
self.assertNotIn("填写样例", text)
|
|
|
|
def test_sample_pair_gate(self):
|
|
self.assertTrue(sample_pair_allowed("国内运输拼车", "国内长途/零担"))
|
|
self.assertTrue(sample_pair_allowed("国内运输整车", "国内城配/拖车/打包"))
|
|
self.assertTrue(sample_pair_allowed("国内运输整车", "国内长途/零担"))
|
|
self.assertTrue(sample_pair_allowed("中港整车", "中港/中亚/中欧"))
|
|
self.assertFalse(sample_pair_allowed("国内运输拼车", "东南亚"))
|
|
self.assertFalse(sample_pair_allowed("中港整车", "东南亚"))
|
|
|
|
def test_scrub_template_ports(self):
|
|
from agent.schema.land_options import scrub_land_placeholder_ports
|
|
|
|
out = scrub_land_placeholder_ports(
|
|
{"起运港": "DOM", "目的港": "FTL", "品名": "衣服", "运输类型": "国内运输整车"}
|
|
)
|
|
self.assertNotIn("起运港", out)
|
|
self.assertNotIn("目的港", out)
|
|
self.assertEqual(out.get("品名"), "衣服")
|
|
|
|
def test_paste_numbers_and_codes(self):
|
|
"""复制清单后在冒号后填数字/编号，要收到标准项。"""
|
|
pasted = (
|
|
"运输类型（必填）：2\n"
|
|
"线路类别（必填）：2.1\n"
|
|
"运输分类（必填）：5"
|
|
)
|
|
got = harvest_land_options(pasted)
|
|
self.assertEqual(got.get("运输类型"), "国内运输整车")
|
|
self.assertEqual(got.get("线路类别"), "国内长途/零担")
|
|
self.assertEqual(got.get("运输分类"), "急件")
|
|
|
|
def test_common_route_class_only_full_text_or_number(self):
|
|
"""新项只认编号 1 或全文；短话和逗号写法不收。"""
|
|
full = "普件(常用线路/常规车型)"
|
|
self.assertEqual(harvest_land_options(full).get("运输分类"), full)
|
|
numbered = harvest_land_options(
|
|
"1",
|
|
current={"运输类型": "国内运输拼车", "线路类别": "国内长途/零担"},
|
|
)
|
|
self.assertEqual(numbered.get("运输分类"), full)
|
|
self.assertFalse(harvest_land_options("常用线路").get("运输分类"))
|
|
self.assertFalse(harvest_land_options("常规车型").get("运输分类"))
|
|
self.assertFalse(harvest_land_options("普件，常用线路/常规车型").get("运输分类"))
|
|
|
|
def test_route_code_infers_type_when_empty(self):
|
|
got = harvest_land_options("1.1")
|
|
self.assertEqual(got.get("运输类型"), "国内运输拼车")
|
|
self.assertEqual(got.get("线路类别"), "国内长途/零担")
|
|
|
|
def test_unfilled_prompt_does_not_eat_menu(self):
|
|
"""原样粘贴未填的清单，不能把选项名当成已选。"""
|
|
got = harvest_land_options(option_prompt(current={}))
|
|
self.assertFalse(got.get("运输类型"))
|
|
self.assertFalse(got.get("线路类别"))
|
|
self.assertFalse(got.get("运输分类"))
|
|
|
|
def test_confirm_intent(self):
|
|
self.assertTrue(looks_like_confirm("确认"))
|
|
self.assertTrue(looks_like_confirm("确定"))
|
|
self.assertTrue(looks_like_confirm("可以查了"))
|
|
self.assertTrue(looks_like_confirm("查价"))
|
|
self.assertFalse(looks_like_confirm("广州到深圳衣服"))
|
|
# 询价正文里的「确认」不是点核对卡。
|
|
self.assertFalse(
|
|
looks_like_confirm(
|
|
"服务器、不带电，预计7月初货好，因货值较高 派送需确认安全运输方案，比如押运。"
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|