mirror of
https://github.com/LifeArchiveProject/WeChatDataAnalysis.git
synced 2026-02-19 22:30:49 +08:00
cf58d26e6f
- 新增系统撤回消息 replacemsg 解析与导出语义测试 - 新增群聊会话预览格式化与群名片 ext_buffer 解析测试 - 新增 realtime 会话列表与 sync_all 落库 last_sender_display_name 测试
104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
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()
|
|
|