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>
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
from langbot.libs.wecom_customer_service_api.api import WecomCSClient
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_image_msg_posts_customer_service_image_payload() -> None:
|
|
captured_request: httpx.Request | None = None
|
|
|
|
def handle_request(request: httpx.Request) -> httpx.Response:
|
|
nonlocal captured_request
|
|
captured_request = request
|
|
return httpx.Response(200, json={'errcode': 0})
|
|
|
|
client = WecomCSClient(
|
|
corpid='corp-id',
|
|
secret='secret',
|
|
token='token',
|
|
EncodingAESKey='encoding-key',
|
|
logger=None,
|
|
unified_mode=True,
|
|
)
|
|
client.access_token = 'access-token'
|
|
client._http_client = httpx.AsyncClient(transport=httpx.MockTransport(handle_request))
|
|
|
|
try:
|
|
await client.send_image_msg(
|
|
open_kfid='kf-test',
|
|
external_userid='external-user',
|
|
msgid='a' * 32,
|
|
media_id='media-id',
|
|
)
|
|
finally:
|
|
await client.close()
|
|
|
|
assert captured_request is not None
|
|
assert captured_request.url.path == '/cgi-bin/kf/send_msg'
|
|
assert captured_request.url.params['access_token'] == 'access-token'
|
|
assert captured_request.method == 'POST'
|
|
assert captured_request.read().decode() == (
|
|
'{"touser":"external-user","open_kfid":"kf-test","msgid":"'
|
|
+ 'a' * 32
|
|
+ '","msgtype":"image","image":{"media_id":"media-id"}}'
|
|
)
|