jillion886andCursor
1a2e1d6cb2
成交走询价消息服务；陆运/海运私聊下一句能记下原因；协商中回执写明顺延三个工作日和下次 9：00。 Co-authored-by: Cursor <cursoragent@cursor.com>
145 lines
4.3 KiB
Python
145 lines
4.3 KiB
Python
"""
|
|
成交跟进：工作日顺延、线路对负责人、未成交收下原因（不打外网）。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
from datetime import date, datetime
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from agent.policy.deal_followup import (
|
|
SHANGHAI,
|
|
air_route_keys,
|
|
leader_matches,
|
|
next_followup_at,
|
|
shift_workdays,
|
|
)
|
|
|
|
|
|
class TestWorkdays(unittest.TestCase):
|
|
def test_skip_weekend(self) -> None:
|
|
# 2026-09-18 周五，顺延 3 个工作日 → 21、22、23，第四个工作日 24
|
|
self.assertEqual(shift_workdays(date(2026, 9, 18), 3), date(2026, 9, 23))
|
|
self.assertEqual(shift_workdays(date(2026, 9, 18), 4), date(2026, 9, 24))
|
|
|
|
def test_skip_national_day(self) -> None:
|
|
# 2026-09-30 周三，国庆休 → 第四个工作日落到节后
|
|
fourth = shift_workdays(date(2026, 9, 30), 4)
|
|
self.assertGreaterEqual(fourth, date(2026, 10, 9))
|
|
self.assertNotEqual(fourth.weekday(), 5)
|
|
self.assertNotEqual(fourth.weekday(), 6)
|
|
|
|
def test_next_followup_is_9am(self) -> None:
|
|
when = next_followup_at(datetime(2026, 9, 18, 15, 30, tzinfo=SHANGHAI))
|
|
self.assertEqual(when.hour, 9)
|
|
self.assertEqual(when.minute, 0)
|
|
self.assertEqual(when.tzinfo, SHANGHAI)
|
|
self.assertEqual(when.date(), date(2026, 9, 24))
|
|
|
|
|
|
class TestAirRouteKeys(unittest.TestCase):
|
|
def test_pair_and_dest(self) -> None:
|
|
keys = air_route_keys({"起运港": "SZX", "目的港": "KUL"})
|
|
self.assertIn("SZX-KUL", keys)
|
|
self.assertIn("KUL", keys)
|
|
|
|
def test_chinese_city_projects_iata(self) -> None:
|
|
keys = air_route_keys({"起运港": "深圳", "目的港": "吉隆坡"})
|
|
self.assertIn("SZX-KUL", keys)
|
|
self.assertIn("KUL", keys)
|
|
|
|
|
|
class TestLeaderMatch(unittest.TestCase):
|
|
def test_sea_all_line(self) -> None:
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="sea",
|
|
routes="海运全线",
|
|
is_leader=True,
|
|
business_line="SEA",
|
|
facts={},
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
leader_matches(
|
|
role_code="sea",
|
|
routes="海运全线",
|
|
is_leader=False,
|
|
business_line="SEA",
|
|
facts={},
|
|
)
|
|
)
|
|
|
|
def test_land_route_category(self) -> None:
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="land",
|
|
routes="国内长途/零担,东南亚",
|
|
is_leader=True,
|
|
business_line="LAND",
|
|
facts={"线路类别": "国内长途/零担"},
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
leader_matches(
|
|
role_code="land",
|
|
routes="东南亚",
|
|
is_leader=True,
|
|
business_line="LAND",
|
|
facts={"线路类别": "国内长途/零担"},
|
|
)
|
|
)
|
|
|
|
def test_air_pair_dest_and_all(self) -> None:
|
|
facts = {"起运港": "SZX", "目的港": "KUL"}
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="air",
|
|
routes="SZX-KUL",
|
|
is_leader=True,
|
|
business_line="AIR",
|
|
facts=facts,
|
|
)
|
|
)
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="air",
|
|
routes="KUL",
|
|
is_leader=True,
|
|
business_line="AIR",
|
|
facts=facts,
|
|
)
|
|
)
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="air",
|
|
routes="全线",
|
|
is_leader=True,
|
|
business_line="AIR",
|
|
facts=facts,
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
leader_matches(
|
|
role_code="air",
|
|
routes="外采-空运",
|
|
is_leader=True,
|
|
business_line="AIR",
|
|
facts=facts,
|
|
)
|
|
)
|
|
self.assertTrue(
|
|
leader_matches(
|
|
role_code="air",
|
|
routes="外采-空运",
|
|
is_leader=True,
|
|
business_line="AIR",
|
|
facts={**facts, "外采": "1"},
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|