mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-27 12:47:13 +00:00
Yang
cabde423a1
* Update wecomcs.py fix bug wecomcs send_message open_kfid event.receiver_id is open_kfid * Update wecomcs.py fix bug msgid exceeds the 32-byte limit * test(wecomcs): cover bounded message IDs and images --------- Co-authored-by: dadachann <185672915+dadachann@users.noreply.github.com>
115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
import uuid
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
import langbot_plugin.api.definition.abstract.platform.event_logger as abstract_platform_logger
|
|
import langbot_plugin.api.entities.builtin.platform.message as platform_message
|
|
from langbot.pkg.platform.sources.wecomcs import WecomCSAdapter
|
|
|
|
|
|
class DummyLogger(abstract_platform_logger.AbstractEventLogger):
|
|
async def info(self, *args, **kwargs):
|
|
pass
|
|
|
|
async def debug(self, *args, **kwargs):
|
|
pass
|
|
|
|
async def warning(self, *args, **kwargs):
|
|
pass
|
|
|
|
async def error(self, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
def make_adapter():
|
|
return WecomCSAdapter(
|
|
config={
|
|
'corpid': 'corp-id',
|
|
'secret': 'secret',
|
|
'token': 'token',
|
|
'EncodingAESKey': 'encoding-key',
|
|
},
|
|
logger=DummyLogger(),
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_sends_text_to_customer_service_user():
|
|
adapter = make_adapter()
|
|
adapter.bot_account_id = 'kf-test'
|
|
adapter.bot = SimpleNamespace(send_text_msg=AsyncMock())
|
|
|
|
message = platform_message.MessageChain([platform_message.Plain(text='hello')])
|
|
|
|
await adapter.send_message('person', 'uexternal-user', message)
|
|
|
|
adapter.bot.send_text_msg.assert_awaited_once()
|
|
kwargs = adapter.bot.send_text_msg.await_args.kwargs
|
|
assert kwargs['open_kfid'] == 'kf-test'
|
|
assert kwargs['external_userid'] == 'external-user'
|
|
assert kwargs['content'] == 'hello'
|
|
assert len(kwargs['msgid'].encode()) <= 32
|
|
assert uuid.UUID(hex=kwargs['msgid']).hex == kwargs['msgid']
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_sends_image_to_customer_service_user():
|
|
adapter = make_adapter()
|
|
adapter.bot_account_id = 'kf-test'
|
|
adapter.bot = SimpleNamespace(
|
|
get_media_id=AsyncMock(return_value='media-id'),
|
|
send_image_msg=AsyncMock(),
|
|
)
|
|
|
|
message = platform_message.MessageChain([platform_message.Image(base64='aW1hZ2U=')])
|
|
|
|
await adapter.send_message('person', 'uexternal-user', message)
|
|
|
|
adapter.bot.send_image_msg.assert_awaited_once()
|
|
kwargs = adapter.bot.send_image_msg.await_args.kwargs
|
|
assert kwargs['open_kfid'] == 'kf-test'
|
|
assert kwargs['external_userid'] == 'external-user'
|
|
assert kwargs['media_id'] == 'media-id'
|
|
assert len(kwargs['msgid'].encode()) <= 32
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_allows_explicit_open_kfid_in_target_id():
|
|
adapter = make_adapter()
|
|
adapter.bot = SimpleNamespace(send_text_msg=AsyncMock())
|
|
|
|
message = platform_message.MessageChain([platform_message.Plain(text='hello')])
|
|
|
|
await adapter.send_message('person', 'kf-explicit|uexternal-user', message)
|
|
|
|
kwargs = adapter.bot.send_text_msg.await_args.kwargs
|
|
assert kwargs['open_kfid'] == 'kf-explicit'
|
|
assert kwargs['external_userid'] == 'external-user'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_requires_open_kfid():
|
|
adapter = make_adapter()
|
|
adapter.bot = SimpleNamespace(send_text_msg=AsyncMock())
|
|
message = platform_message.MessageChain([platform_message.Plain(text='hello')])
|
|
|
|
with pytest.raises(ValueError, match='open_kfid is required'):
|
|
await adapter.send_message('person', 'uexternal-user', message)
|
|
|
|
adapter.bot.send_text_msg.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_rejects_group_targets():
|
|
adapter = make_adapter()
|
|
adapter.bot_account_id = 'kf-test'
|
|
adapter.bot = SimpleNamespace(send_text_msg=AsyncMock())
|
|
message = platform_message.MessageChain([platform_message.Plain(text='hello')])
|
|
|
|
with pytest.raises(ValueError, match='only supports sending messages to person'):
|
|
await adapter.send_message('group', 'group-id', message)
|
|
|
|
adapter.bot.send_text_msg.assert_not_called()
|