test(chat): 覆盖系统撤回/群名片/实时会话同步相关用例

- 新增系统撤回消息 replacemsg 解析与导出语义测试

- 新增群聊会话预览格式化与群名片 ext_buffer 解析测试

- 新增 realtime 会话列表与 sync_all 落库 last_sender_display_name 测试
This commit is contained in:
2977094657
2026-02-09 18:31:47 +08:00
parent 2c832aa861
commit cf58d26e6f
6 changed files with 479 additions and 0 deletions

View File

@@ -0,0 +1,103 @@
import sys
import threading
import unittest
from pathlib import Path
from tempfile import TemporaryDirectory
from unittest.mock import patch
ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT / "src"))
from wechat_decrypt_tool.routers import chat as chat_router
class _DummyRequest:
base_url = "http://testserver/"
class _DummyConn:
def __init__(self) -> None:
self.handle = 1
self.lock = threading.Lock()
class TestChatSessionsRealtimeSenderPreview(unittest.TestCase):
def _run(self, sessions_rows: list[dict]) -> dict:
with TemporaryDirectory() as td:
account_dir = Path(td) / "acc"
account_dir.mkdir(parents=True, exist_ok=True)
conn = _DummyConn()
with (
patch.object(chat_router, "_resolve_account_dir", return_value=account_dir),
patch.object(chat_router.WCDB_REALTIME, "ensure_connected", return_value=conn),
patch.object(chat_router, "_wcdb_get_sessions", return_value=sessions_rows),
patch.object(chat_router, "_wcdb_get_display_names", return_value={}),
patch.object(chat_router, "_wcdb_get_avatar_urls", return_value={}),
patch.object(chat_router, "_load_contact_rows", return_value={}),
patch.object(chat_router, "_query_head_image_usernames", return_value=set()),
patch.object(chat_router, "_should_keep_session", return_value=True),
patch.object(chat_router, "_avatar_url_unified", return_value="/avatar"),
):
return chat_router.list_chat_sessions(
_DummyRequest(),
account="acc",
limit=50,
include_hidden=True,
include_official=True,
preview="latest",
source="realtime",
)
def test_realtime_sessions_group_summary_prefixed_by_sender_display_name(self):
resp = self._run(
[
{
"username": "demo@chatroom",
"summary": "hello",
"draft": "",
"unread_count": 0,
"is_hidden": 0,
"last_timestamp": 123,
"sort_timestamp": 123,
"last_msg_type": 1,
"last_msg_sub_type": 0,
"last_msg_sender": "wxid_demo",
"last_sender_display_name": "群名片A",
}
]
)
self.assertEqual(resp.get("status"), "success")
sessions = resp.get("sessions") or []
self.assertEqual(len(sessions), 1)
self.assertEqual(sessions[0].get("lastMessage"), "群名片A: hello")
def test_realtime_sessions_group_url_summary_keeps_scheme(self):
resp = self._run(
[
{
"username": "url@chatroom",
"summary": "https://example.com/x",
"draft": "",
"unread_count": 0,
"is_hidden": 0,
"last_timestamp": 123,
"sort_timestamp": 123,
"last_msg_type": 1,
"last_msg_sub_type": 0,
"last_msg_sender": "wxid_demo",
"last_sender_display_name": "群名片B",
}
]
)
self.assertEqual(resp.get("status"), "success")
sessions = resp.get("sessions") or []
self.assertEqual(len(sessions), 1)
self.assertEqual(sessions[0].get("lastMessage"), "群名片B: https://example.com/x")
if __name__ == "__main__":
unittest.main()