89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from nanobot.bus.events import OutboundMessage
|
|
from nanobot.bus.queue import MessageBus
|
|
from nanobot.channels.slack import SlackChannel
|
|
from nanobot.config.schema import SlackConfig
|
|
|
|
|
|
class _FakeAsyncWebClient:
|
|
def __init__(self) -> None:
|
|
self.chat_post_calls: list[dict[str, object | None]] = []
|
|
self.file_upload_calls: list[dict[str, object | None]] = []
|
|
|
|
async def chat_postMessage(
|
|
self,
|
|
*,
|
|
channel: str,
|
|
text: str,
|
|
thread_ts: str | None = None,
|
|
) -> None:
|
|
self.chat_post_calls.append(
|
|
{
|
|
"channel": channel,
|
|
"text": text,
|
|
"thread_ts": thread_ts,
|
|
}
|
|
)
|
|
|
|
async def files_upload_v2(
|
|
self,
|
|
*,
|
|
channel: str,
|
|
file: str,
|
|
thread_ts: str | None = None,
|
|
) -> None:
|
|
self.file_upload_calls.append(
|
|
{
|
|
"channel": channel,
|
|
"file": file,
|
|
"thread_ts": thread_ts,
|
|
}
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_uses_thread_for_channel_messages() -> None:
|
|
channel = SlackChannel(SlackConfig(enabled=True), MessageBus())
|
|
fake_web = _FakeAsyncWebClient()
|
|
channel._web_client = fake_web
|
|
|
|
await channel.send(
|
|
OutboundMessage(
|
|
channel="slack",
|
|
chat_id="C123",
|
|
content="hello",
|
|
media=["/tmp/demo.txt"],
|
|
metadata={"slack": {"thread_ts": "1700000000.000100", "channel_type": "channel"}},
|
|
)
|
|
)
|
|
|
|
assert len(fake_web.chat_post_calls) == 1
|
|
assert fake_web.chat_post_calls[0]["thread_ts"] == "1700000000.000100"
|
|
assert len(fake_web.file_upload_calls) == 1
|
|
assert fake_web.file_upload_calls[0]["thread_ts"] == "1700000000.000100"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_omits_thread_for_dm_messages() -> None:
|
|
channel = SlackChannel(SlackConfig(enabled=True), MessageBus())
|
|
fake_web = _FakeAsyncWebClient()
|
|
channel._web_client = fake_web
|
|
|
|
await channel.send(
|
|
OutboundMessage(
|
|
channel="slack",
|
|
chat_id="D123",
|
|
content="hello",
|
|
media=["/tmp/demo.txt"],
|
|
metadata={"slack": {"thread_ts": "1700000000.000100", "channel_type": "im"}},
|
|
)
|
|
)
|
|
|
|
assert len(fake_web.chat_post_calls) == 1
|
|
assert fake_web.chat_post_calls[0]["thread_ts"] is None
|
|
assert len(fake_web.file_upload_calls) == 1
|
|
assert fake_web.file_upload_calls[0]["thread_ts"] is None
|