238 lines
12 KiB
Python
238 lines
12 KiB
Python
"""Fast/slow comparisons preserve shipping conditions and comparable rate groups."""
|
||
|
||
from pathlib import Path
|
||
import tempfile
|
||
import unittest
|
||
from unittest.mock import patch
|
||
|
||
from logistics_chat import create_app, db
|
||
from logistics_chat.quote_data import import_snapshot
|
||
from logistics_chat.quote_engine import select_timed_rates
|
||
from test_quotes import fixture_snapshot, frames
|
||
|
||
|
||
def timing_snapshot():
|
||
book=fixture_snapshot()
|
||
rates=[]
|
||
for origin in ('义乌','华南'):
|
||
for zone,offset in (('美西',0),('美中',1),('美东',2)):
|
||
for service,price,eta in (('美森正班海派',8,'船开后15天左右'),
|
||
('合德海派',6,'船开后20-22天左右'),
|
||
('普船海派',4,'船开后25-30天左右')):
|
||
rates.append({'service':service,'kind':'parcel','origin':origin,'zone':zone,
|
||
'unit':'kg','tax':'included','min_weight':'12','price':str(price+offset),
|
||
'eta':eta,'destinations':[],'source_sheet':'合成时效','source_cell':'A1'})
|
||
book['rates']=rates
|
||
return book
|
||
|
||
|
||
class QuoteTimingTests(unittest.TestCase):
|
||
def setUp(self):
|
||
folder=tempfile.TemporaryDirectory();self.addCleanup(folder.cleanup)
|
||
root=Path(folder.name)
|
||
self.app=create_app({'TESTING':True,'DATABASE':root/'test.db','UPLOAD_DIR':root/'uploads',
|
||
'BACKUP_DIR':root/'backups','LOG_DIR':root/'logs','DEEPSEEK_API_KEY':'','RATE_LIMIT_PER_MIN':1000})
|
||
self.client=self.app.test_client()
|
||
self.client.post('/api/visitor/session',json={'key':'2'*64})
|
||
with self.app.app_context():
|
||
connection=db.connect()
|
||
import_snapshot(connection,timing_snapshot(),activate=True)
|
||
connection.commit();connection.close()
|
||
|
||
def ask(self,question,cid=None):
|
||
with patch('logistics_chat.ai_client.post_completion') as model:
|
||
events=frames(self.client.post('/stream_chat',json={'msg':question,'conv_id':cid}))
|
||
model.assert_not_called()
|
||
self.assertEqual(events[-1],{'d':1},events)
|
||
return events[0]['t']
|
||
|
||
def test_fastest_and_slowest_show_each_region_origin_and_the_correct_amount(self):
|
||
text=self.ask('我想发一批货，一吨发发到美国走海运，最快的和最慢的价格分别是多少？')
|
||
fast,slow=text.split('最慢参考方案')
|
||
self.assertIn('最快参考方案',fast)
|
||
self.assertIn('美西 8元/KG，1000KG = 8000元',fast)
|
||
self.assertIn('美东 10元/KG，1000KG = 10000元',fast)
|
||
self.assertIn('美森正班海派 · 华南',fast)
|
||
self.assertNotIn('普船海派',fast)
|
||
self.assertIn('美西 4元/KG，1000KG = 4000元',slow)
|
||
self.assertIn('美东 6元/KG，1000KG = 6000元',slow)
|
||
self.assertIn('船开后25-30天左右提取',slow)
|
||
self.assertNotIn('合德海派',text)
|
||
|
||
def test_short_followups_keep_weight_origin_zone_and_selected_comparison(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('1000KG到美国走海运，最快和最慢价格分别多少',cid)
|
||
for question in ('慢的呢？','义乌交仓','美东呢','请给我具体报价'):
|
||
text=self.ask(question,cid)
|
||
self.assertIn('美国 1000KG',text)
|
||
self.assertIn('最慢参考方案',text)
|
||
self.assertNotIn('最快参考方案',text)
|
||
self.assertNotIn('华南',text)
|
||
self.assertIn('1000KG = 6000元',text)
|
||
text=self.ask('50KG呢',cid)
|
||
self.assertIn('50KG = 300元',text)
|
||
self.assertNotIn('1000KG',text)
|
||
text=self.ask('全部渠道报价',cid)
|
||
self.assertIn('美森正班海派',text)
|
||
self.assertIn('合德海派',text)
|
||
self.assertNotIn('最慢参考方案',text)
|
||
text=self.ask('新的一批100KG发美国海运报价',cid)
|
||
self.assertIn('美国 100KG',text)
|
||
self.assertIn('华南',text)
|
||
self.assertNotIn('最慢参考方案',text)
|
||
|
||
def test_numeric_timing_groups_keep_destinations_tax_and_service_kinds_separate(self):
|
||
base={'zone':'美西','kind':'parcel','unit':'kg','tax':'included'}
|
||
rates=[{**base,'eta':'10-35天','service':'wide'},
|
||
{**base,'eta':'15-25天','service':'narrow'},
|
||
{**base,'zone':'美东','eta':'40-45天','service':'east'},
|
||
{**base,'tax':'excluded','eta':'20天','service':'excluded'},
|
||
{**base,'kind':'truck','eta':'50天','service':'truck'},
|
||
{**base,'eta':'未提供时效','service':'unknown'}]
|
||
self.assertEqual({r['service'] for r in select_timed_rates(rates)},
|
||
{'wide','east','excluded','truck'})
|
||
self.assertEqual({r['service'] for r in select_timed_rates(rates,slowest=True)},
|
||
{'wide','east','excluded','truck'})
|
||
self.assertEqual(select_timed_rates([{**base,'eta':'若干天'}]),[])
|
||
|
||
def test_english_comparison_keeps_numbers_and_renders_both_speeds(self):
|
||
text=self.ask('1000kg to USA by sea, fastest and slowest prices please')
|
||
self.assertIn('Fastest reference options',text)
|
||
self.assertIn('Slowest reference options',text)
|
||
self.assertIn('1000KG = 4000 CNY',text)
|
||
|
||
def test_different_regional_times_are_rendered_on_their_own_price_lines(self):
|
||
with self.app.app_context():
|
||
connection=db.connect()
|
||
connection.execute("UPDATE quote_rates SET eta='船开后18天左右' WHERE service='美森正班海派' AND zone='美东'")
|
||
connection.commit();connection.close()
|
||
text=self.ask('美国1000KG海运最快价格，义乌')
|
||
self.assertIn('美东 10元/KG，1000KG = 10000元（船开后18天左右提取）',text)
|
||
self.assertIn('美西 8元/KG，1000KG = 8000元',text)
|
||
|
||
def test_unknown_times_do_not_become_a_fastest_or_slowest_promise(self):
|
||
with self.app.app_context():
|
||
connection=db.connect()
|
||
connection.execute("UPDATE quote_rates SET eta='若干天'")
|
||
connection.commit();connection.close()
|
||
text=self.ask('美国1000KG海运最快和最慢价格')
|
||
self.assertIn('最快参考方案暂时无法确定',text)
|
||
self.assertIn('最慢参考方案暂时无法确定',text)
|
||
self.assertNotIn('1000KG =',text)
|
||
|
||
def test_selected_route_survives_total_calculation_time_and_tax_questions(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国1000KG义乌美东海运，便宜的报价',cid)
|
||
for question in ('那一共多少钱','怎么算出来的'):
|
||
text=self.ask(question,cid)
|
||
self.assertIn('普船海派',text)
|
||
self.assertIn('1000KG = 6000元',text)
|
||
self.assertNotIn('美森正班海派',text)
|
||
text=self.ask('多快能到',cid)
|
||
self.assertIn('普船海派',text)
|
||
self.assertIn('船开后25-30天左右提取',text)
|
||
self.assertNotIn('美森正班海派',text)
|
||
self.assertIn('标注包税',self.ask('含税吗',cid))
|
||
text=self.ask('具体多少',cid)
|
||
self.assertIn('1000KG = 6000元',text)
|
||
self.assertNotIn('美森正班海派',text)
|
||
text=self.ask('全部渠道报价',cid)
|
||
self.assertIn('美森正班海派',text)
|
||
self.assertIn('普船海派',text)
|
||
|
||
def test_polite_acknowledgements_keep_quote_conditions_and_selection(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国30KG义乌美西最快海派报价',cid)
|
||
for question in ('谢谢','好的，谢谢您','OK'):
|
||
self.ask(question,cid)
|
||
text=self.ask('改成50KG呢',cid)
|
||
self.assertIn('美森正班海派 · 义乌',text)
|
||
self.assertIn('50KG = 400元',text)
|
||
self.assertNotIn('普船海派',text)
|
||
text=self.ask('所有箱都是普通货',cid)
|
||
self.assertIn('最快参考方案',text)
|
||
self.assertNotIn('普船海派',text)
|
||
|
||
def test_missing_weight_unit_stays_pending_until_an_explicit_unit_is_supplied(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国1000KG义乌美西海运最快报价',cid)
|
||
for question in ('重量改成800，其他不变','换成华南','谢谢','美东呢'):
|
||
text=self.ask(question,cid)
|
||
if question!='谢谢':
|
||
self.assertIn('单位',text)
|
||
self.assertNotIn('1000KG =',text)
|
||
text=self.ask('单位是KG',cid)
|
||
self.assertIn('美森正班海派 · 华南',text)
|
||
self.assertIn('800KG = 8000元',text)
|
||
self.assertNotIn('1000KG',text)
|
||
self.assertIn('800KG = 8000元',self.ask('具体多少',cid))
|
||
|
||
def test_quote_resume_is_explicit_and_survives_later_field_edits(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国30KG义乌美西最快报价',cid)
|
||
self.ask('我有多少个订单',cid)
|
||
text=self.ask('50KG呢',cid)
|
||
self.assertNotIn('美森正班海派',text)
|
||
text=self.ask('回到刚才的报价',cid)
|
||
self.assertIn('30KG = 240元',text)
|
||
self.assertNotIn('50KG',text)
|
||
text=self.ask('换成华南',cid)
|
||
self.assertIn('美森正班海派 · 华南',text)
|
||
self.assertIn('30KG = 240元',text)
|
||
|
||
def test_resume_does_not_recover_us_conditions_across_a_new_country_quote(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国30KG义乌海运报价',cid)
|
||
with self.app.app_context():
|
||
db.save_message(cid,'user','新的一票100KG发德国海运报价')
|
||
db.save_message(cid,'assistant','请核对对应目的地的报价。')
|
||
self.ask('我有多少个订单',cid)
|
||
text=self.ask('回到刚才的报价',cid)
|
||
self.assertIn('目的国家',text)
|
||
self.assertNotIn('美国 30KG',text)
|
||
self.assertNotIn('元/KG',text)
|
||
|
||
def test_new_quote_with_a_topic_switch_becomes_the_next_followup_root(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国30KG义乌美西最快报价',cid)
|
||
self.ask('换个话题，美国50KG华南美东海运报价',cid)
|
||
text=self.ask('最慢的呢',cid)
|
||
self.assertIn('普船海派 · 华南',text)
|
||
self.assertIn('50KG = 300元',text)
|
||
self.assertNotIn('30KG',text)
|
||
self.ask('另外问美国海运报价',cid)
|
||
text=self.ask('20KG呢',cid)
|
||
self.assertIn('美国 20KG',text)
|
||
self.assertIn('义乌',text)
|
||
self.assertNotIn('50KG',text)
|
||
|
||
def test_explicit_selection_and_business_question_are_retained_independently(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国1000KG义乌美东海运报价',cid)
|
||
self.assertIn('标注包税',self.ask('最快的含税吗',cid))
|
||
text=self.ask('那一共多少钱',cid)
|
||
self.assertIn('1000KG = 10000元',text)
|
||
self.assertNotIn('普船海派',text)
|
||
text=self.ask('最快多久能提取',cid)
|
||
self.assertIn('15天左右提取',text)
|
||
self.assertNotIn('普船海派',text)
|
||
|
||
def test_missing_weight_units_before_a_destination_do_not_reuse_the_old_weight(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国1000KG义乌美西最快海运报价',cid)
|
||
self.assertIn('单位',self.ask('重量改成800美东，其他不变',cid))
|
||
text=self.ask('KG',cid)
|
||
self.assertIn('800KG = 8000元',text)
|
||
self.assertNotIn('1000KG',text)
|
||
|
||
def test_all_channels_clears_the_named_service_and_keeps_origin_destination_weight(self):
|
||
cid=self.client.post('/api/new_chat').json['id']
|
||
self.ask('美国1000KG义乌美东普船海派最便宜报价',cid)
|
||
text=self.ask('全部渠道报价',cid)
|
||
self.assertIn('美森正班海派',text)
|
||
self.assertIn('合德海派',text)
|
||
self.assertIn('普船海派',text)
|
||
self.assertIn('1000KG = 10000元',text)
|
||
self.assertNotIn('华南',text)
|
||
self.assertNotIn('美西 ',text)
|