wehub-resource-sync
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import AsyncIterator
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
from opensquilla.channels.discord import DiscordChannel, DiscordChannelConfig
|
|
from opensquilla.channels.slack import SLACK_API_BASE, SlackChannel
|
|
|
|
|
|
async def _two_chunks() -> AsyncIterator[str]:
|
|
yield "first chunk"
|
|
yield "second chunk"
|
|
|
|
|
|
async def test_discord_send_streaming_surfaces_rejected_edit() -> None:
|
|
config = DiscordChannelConfig(token="token", default_channel_id="123")
|
|
channel = DiscordChannel(config=config)
|
|
|
|
def handler(request: httpx.Request) -> httpx.Response:
|
|
if request.method == "POST":
|
|
return httpx.Response(200, json={"id": "m1"})
|
|
return httpx.Response(400, json={"code": 50035, "message": "Invalid Form Body"})
|
|
|
|
channel._client = httpx.AsyncClient(
|
|
base_url=config.api_base, transport=httpx.MockTransport(handler)
|
|
)
|
|
try:
|
|
with pytest.raises(httpx.HTTPStatusError):
|
|
await channel.send_streaming(_two_chunks(), update_interval_ms=0)
|
|
finally:
|
|
await channel._client.aclose()
|
|
|
|
|
|
async def test_slack_send_streaming_surfaces_edit_api_error() -> None:
|
|
channel = SlackChannel(token="xoxb-dummy", slack_channel_id="C123")
|
|
|
|
def handler(request: httpx.Request) -> httpx.Response:
|
|
if request.url.path.endswith("/chat.postMessage"):
|
|
return httpx.Response(200, json={"ok": True, "ts": "111.222"})
|
|
return httpx.Response(200, json={"ok": False, "error": "msg_too_long"})
|
|
|
|
channel._client = httpx.AsyncClient(
|
|
base_url=SLACK_API_BASE, transport=httpx.MockTransport(handler)
|
|
)
|
|
try:
|
|
with pytest.raises(RuntimeError, match="msg_too_long"):
|
|
await channel.send_streaming(_two_chunks(), update_interval_ms=0)
|
|
finally:
|
|
await channel._client.aclose()
|