mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-14 06:30:57 +00:00
fdc310
e211d3aae6
- filter the bot's own messages to prevent reply loops
- use startswith('@@') for group detection
- use ActualUserName as stable GroupMember id
- fill Friend.remark from RemarkName
- strip @mention prefix with a regex
131 lines
4.0 KiB
Python
131 lines
4.0 KiB
Python
"""Tests for itchat adapter group/private message conversion."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from types import SimpleNamespace
|
||
|
||
import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
|
||
import langbot_plugin.api.entities.builtin.platform.events as platform_events
|
||
import langbot_plugin.api.entities.builtin.platform.message as platform_message
|
||
|
||
from langbot.pkg.platform import botmgr as _botmgr # noqa: F401
|
||
|
||
from langbot.pkg.platform.sources.itchat import ItchatAdapter, ItchatEventConverter
|
||
|
||
|
||
def _make_adapter(bot_account_id: str = '@bot_wxid', bot_nickname: str = 'MyBot'):
|
||
adapter = SimpleNamespace(
|
||
bot_account_id=bot_account_id,
|
||
_bot_nickname=bot_nickname,
|
||
_core=SimpleNamespace(storageClass=SimpleNamespace(userName=bot_account_id)),
|
||
_get_obj_value=ItchatAdapter._get_obj_value,
|
||
)
|
||
return adapter
|
||
|
||
|
||
def _make_converter(adapter) -> ItchatEventConverter:
|
||
return ItchatEventConverter(adapter_ref=lambda: adapter)
|
||
|
||
|
||
def test_group_text_becomes_group_message():
|
||
converter = _make_converter(_make_adapter())
|
||
|
||
msg = {
|
||
'FromUserName': '@@group_wxid',
|
||
'Type': 'Text',
|
||
'Text': 'hello',
|
||
'ActualUserName': '@member_wxid',
|
||
'ActualNickName': 'MemberNick',
|
||
'IsAt': False,
|
||
'CreateTime': 123456,
|
||
'User': SimpleNamespace(NickName='Group Name'),
|
||
}
|
||
|
||
event = converter.target2yiri(msg)
|
||
|
||
assert isinstance(event, platform_events.GroupMessage)
|
||
assert isinstance(event.sender, platform_entities.GroupMember)
|
||
assert event.sender.id == '@member_wxid'
|
||
assert event.sender.member_name == 'MemberNick'
|
||
assert event.sender.group.id == '@@group_wxid'
|
||
assert event.sender.group.name == 'Group Name'
|
||
|
||
components = list(event.message_chain)
|
||
assert len(components) == 1
|
||
assert isinstance(components[0], platform_message.Plain)
|
||
assert components[0].text == 'hello'
|
||
|
||
|
||
def test_private_text_becomes_friend_message():
|
||
converter = _make_converter(_make_adapter())
|
||
|
||
msg = {
|
||
'FromUserName': '@friend_wxid',
|
||
'Type': 'Text',
|
||
'Text': 'hi',
|
||
'CreateTime': 123456,
|
||
'User': SimpleNamespace(NickName='FriendNick', RemarkName='FriendRemark'),
|
||
}
|
||
|
||
event = converter.target2yiri(msg)
|
||
|
||
assert isinstance(event, platform_events.FriendMessage)
|
||
assert isinstance(event.sender, platform_entities.Friend)
|
||
assert event.sender.id == '@friend_wxid'
|
||
assert event.sender.nickname == 'FriendNick'
|
||
assert event.sender.remark == 'FriendRemark'
|
||
|
||
|
||
def test_bot_own_message_is_ignored():
|
||
converter = _make_converter(_make_adapter(bot_account_id='@bot_wxid'))
|
||
|
||
msg = {
|
||
'FromUserName': '@bot_wxid',
|
||
'Type': 'Text',
|
||
'Text': 'self echo',
|
||
}
|
||
|
||
assert converter.target2yiri(msg) is None
|
||
|
||
|
||
def test_group_at_bot_strips_prefix_and_adds_at():
|
||
converter = _make_converter(_make_adapter(bot_account_id='@bot_wxid', bot_nickname='MyBot'))
|
||
|
||
msg = {
|
||
'FromUserName': '@@group_wxid',
|
||
'Type': 'Text',
|
||
'Text': '@MyBot hello world',
|
||
'ActualUserName': '@member_wxid',
|
||
'ActualNickName': 'MemberNick',
|
||
'IsAt': True,
|
||
'CreateTime': 123456,
|
||
'User': SimpleNamespace(NickName='Group Name'),
|
||
}
|
||
|
||
event = converter.target2yiri(msg)
|
||
|
||
assert isinstance(event, platform_events.GroupMessage)
|
||
components = list(event.message_chain)
|
||
assert len(components) == 2
|
||
assert isinstance(components[0], platform_message.At)
|
||
assert components[0].target == '@bot_wxid'
|
||
assert isinstance(components[1], platform_message.Plain)
|
||
assert components[1].text == 'hello world'
|
||
|
||
|
||
def test_group_message_without_sender_is_ignored():
|
||
converter = _make_converter(_make_adapter())
|
||
|
||
msg = {
|
||
'FromUserName': '@@group_wxid',
|
||
'Type': 'Text',
|
||
'Text': 'system note',
|
||
'ActualUserName': '',
|
||
'ActualNickName': '',
|
||
'IsAt': False,
|
||
'CreateTime': 123456,
|
||
'User': SimpleNamespace(NickName='Group Name'),
|
||
}
|
||
|
||
assert converter.target2yiri(msg) is None
|