117 lines
4.7 KiB
Python
117 lines
4.7 KiB
Python
"""
|
|
私聊附件等材料波次：发卡等待、发图/H5 并入、30 分钟时效。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
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.policy.attachment_wave import (
|
|
WAIT_TTL_SECONDS,
|
|
AttachWaveBook,
|
|
reset_attach_wave_book_for_test,
|
|
)
|
|
|
|
|
|
class AttachWaveTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.book = reset_attach_wave_book_for_test()
|
|
|
|
def test_wait_ttl_is_30_minutes(self) -> None:
|
|
self.assertEqual(WAIT_TTL_SECONDS, 1800)
|
|
|
|
def test_open_wait_is_active_without_files(self) -> None:
|
|
wave = self.book.open_wait("s1", precede_text="海运上海到纽约", token="tok1", now=100.0)
|
|
self.assertTrue(wave.wave_id)
|
|
self.assertEqual(wave.precede_text, "海运上海到纽约")
|
|
self.assertFalse(wave.started)
|
|
self.assertIsNotNone(self.book.active("s1", now=100.0))
|
|
|
|
def test_wait_expires_after_30_minutes_if_not_started(self) -> None:
|
|
self.book.open_wait("s1", precede_text="海运", token="tok1", now=100.0)
|
|
self.assertIsNone(self.book.active("s1", now=100.0 + 1800.1))
|
|
|
|
def test_follow_text_while_waiting(self) -> None:
|
|
self.book.open_wait("s1", precede_text="", token="tok1", now=1.0)
|
|
self.assertTrue(self.book.append_follow_text("s1", "只要整柜"))
|
|
self.assertFalse(self.book.append_follow_text("s1", ""))
|
|
|
|
def test_join_image_starts_recognition(self) -> None:
|
|
self.book.open_wait("s1", precede_text="海运", token="tok1", now=1.0)
|
|
wave, how = self.book.join_image("s1", "mid-a", filename="a.jpg", now=2.0)
|
|
self.assertEqual(how, "joined")
|
|
self.assertTrue(wave.started)
|
|
self.assertEqual(self.book.pending(wave), 1)
|
|
|
|
def test_add_files_keeps_upload_order(self) -> None:
|
|
self.book.open_wait("s1", precede_text="", token="tok1", now=1.0)
|
|
wave = self.book.add_files("s1", ["sea.xlsx", "air.pdf"], now=2.0)
|
|
self.assertTrue(wave.started)
|
|
self.assertEqual([m.filename for m in wave.materials], ["sea.xlsx", "air.pdf"])
|
|
self.assertEqual(self.book.pending(wave), 2)
|
|
|
|
def test_compose_oral_uses_material_order(self) -> None:
|
|
self.book.open_wait("s1", precede_text="开口字", token="tok1", now=1.0)
|
|
self.book.add_files("s1", ["a.xlsx", "b.pdf"], now=2.0)
|
|
self.book.append_follow_text("s1", "只要40尺")
|
|
wave = self.book.active("s1", now=2.0)
|
|
self.assertIsNotNone(wave)
|
|
self.book.complete_material("s1", 1, text="空运 上海到纽约")
|
|
self.book.complete_material("s1", 0, text="海运 宁波到汉堡")
|
|
oral = self.book.compose_oral("s1")
|
|
self.assertEqual(oral, "开口字\n海运 宁波到汉堡\n空运 上海到纽约\n只要40尺")
|
|
|
|
def test_after_business_done_not_active(self) -> None:
|
|
self.book.open_wait("s1", precede_text="", token="tok1", now=1.0)
|
|
self.book.add_files("s1", ["a.xlsx"], now=2.0)
|
|
self.book.complete_material("s1", 0, text="海运")
|
|
self.book.mark_business_done("s1")
|
|
self.assertIsNone(self.book.active("s1", now=3.0))
|
|
self.assertFalse(self.book.append_follow_text("s1", "补一句"))
|
|
|
|
def test_image_after_business_rejected(self) -> None:
|
|
self.book.open_wait("s1", precede_text="", token="tok1", now=1.0)
|
|
self.book.mark_business_done("s1")
|
|
wave, how = self.book.join_image("s1", "mid-b", filename="b.jpg", now=2.0)
|
|
self.assertEqual(how, "inactive")
|
|
self.assertIsNone(wave)
|
|
|
|
def test_http_and_worker_share_kv(self) -> None:
|
|
kv = _MemoryWaveKV()
|
|
http = reset_attach_wave_book_for_test(kv=kv)
|
|
worker = AttachWaveBook(kv=kv)
|
|
http.open_wait("s1", precede_text="海运", token="tok", now=1.0)
|
|
self.assertIsNotNone(worker.active("s1", now=1.0))
|
|
worker.add_files("s1", ["a.xlsx"], now=2.0)
|
|
worker.complete_material("s1", 0, text="宁波到汉堡")
|
|
worker.mark_business_done("s1")
|
|
self.assertIsNone(http.active("s1", now=3.0))
|
|
|
|
|
|
class _MemoryWaveKV:
|
|
"""单测假 Redis：两份账本共用同一字典。"""
|
|
|
|
def __init__(self) -> None:
|
|
self._d: dict[str, dict] = {}
|
|
|
|
def get_json(self, kind: str, sender_id: str):
|
|
row = self._d.get(f"{kind}:{sender_id}")
|
|
return json.loads(json.dumps(row)) if row is not None else None
|
|
|
|
def set_json(self, kind: str, sender_id: str, data: dict) -> None:
|
|
self._d[f"{kind}:{sender_id}"] = json.loads(json.dumps(data))
|
|
|
|
def delete(self, kind: str, sender_id: str) -> None:
|
|
self._d.pop(f"{kind}:{sender_id}", None)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|