落地陆运私聊文字询价:选项核对后再建单查价,并按线路拉产品进群。
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
"""
|
||||
内存账本:陆运按线路类别整词匹配产品岗。
|
||||
"""
|
||||
|
||||
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("LEDGER_BACKEND", "memory")
|
||||
|
||||
from agent.ledger.memory_ledger import MemoryLedger
|
||||
|
||||
|
||||
class MemoryLandStaffTest(unittest.TestCase):
|
||||
def test_match_route_token(self):
|
||||
book = MemoryLedger()
|
||||
book.set_land_staff(
|
||||
[
|
||||
{"roleCode": "land", "status": "active", "wecomId": "a", "name": "甲", "routes": "东南亚"},
|
||||
{"roleCode": "land", "status": "active", "wecomId": "b", "name": "乙", "routes": "国内长途/零担"},
|
||||
{"roleCode": "land", "status": "active", "wecomId": "c", "name": "丙", "routes": "中港/中亚/中欧,东南亚"},
|
||||
]
|
||||
)
|
||||
hits = book.match_land_staff(route_category="东南亚")
|
||||
names = [x.get("name") for x in hits]
|
||||
self.assertEqual(names, ["甲", "丙"])
|
||||
|
||||
def test_sea_staff_not_pulled(self):
|
||||
book = MemoryLedger()
|
||||
book.set_land_staff(
|
||||
[{"roleCode": "sea", "status": "active", "wecomId": "s", "name": "海", "routes": "东南亚"}]
|
||||
)
|
||||
self.assertEqual(book.match_land_staff(route_category="东南亚"), [])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user