from __future__ import annotations import pytest # Check optional Slack dependencies before running tests try: import slack_sdk # noqa: F401 except ImportError: pytest.skip("Slack dependencies not installed (slack-sdk)", allow_module_level=True) from nanobot.bus.events import OutboundMessage from nanobot.bus.queue import MessageBus from nanobot.channels.slack import SlackChannel, 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]] = [] self.reactions_add_calls: list[dict[str, object | None]] = [] self.reactions_remove_calls: list[dict[str, object | None]] = [] self.conversations_list_calls: list[dict[str, object | None]] = [] self.conversations_replies_calls: list[dict[str, object | None]] = [] self.users_list_calls: list[dict[str, object | None]] = [] self.conversations_open_calls: list[dict[str, object | None]] = [] self._conversations_pages: list[dict[str, object]] = [] self._conversations_replies_response: dict[str, object] = {"messages": []} self._users_pages: list[dict[str, object]] = [] self._open_dm_response: dict[str, object] = {"channel": {"id": "D_OPENED"}} 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, } ) async def reactions_add( self, *, channel: str, name: str, timestamp: str, ) -> None: self.reactions_add_calls.append( { "channel": channel, "name": name, "timestamp": timestamp, } ) async def reactions_remove( self, *, channel: str, name: str, timestamp: str, ) -> None: self.reactions_remove_calls.append( { "channel": channel, "name": name, "timestamp": timestamp, } ) async def conversations_list(self, **kwargs): self.conversations_list_calls.append(kwargs) if self._conversations_pages: return self._conversations_pages.pop(0) return {"channels": [], "response_metadata": {"next_cursor": ""}} async def conversations_replies(self, **kwargs): self.conversations_replies_calls.append(kwargs) return self._conversations_replies_response async def users_list(self, **kwargs): self.users_list_calls.append(kwargs) if self._users_pages: return self._users_pages.pop(0) return {"members": [], "response_metadata": {"next_cursor": ""}} async def conversations_open(self, **kwargs): self.conversations_open_calls.append(kwargs) return self._open_dm_response @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]["text"] == "hello\n" 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]["text"] == "hello\n" 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 @pytest.mark.asyncio async def test_send_updates_reaction_when_final_response_sent() -> None: channel = SlackChannel(SlackConfig(enabled=True, react_emoji="eyes"), MessageBus()) fake_web = _FakeAsyncWebClient() channel._web_client = fake_web await channel.send( OutboundMessage( channel="slack", chat_id="C123", content="done", metadata={ "slack": {"event": {"ts": "1700000000.000100"}, "channel_type": "channel"}, }, ) ) assert fake_web.reactions_remove_calls == [ {"channel": "C123", "name": "eyes", "timestamp": "1700000000.000100"} ] assert fake_web.reactions_add_calls == [ {"channel": "C123", "name": "white_check_mark", "timestamp": "1700000000.000100"} ] @pytest.mark.asyncio async def test_send_resolves_channel_name_to_channel_id() -> None: channel = SlackChannel(SlackConfig(enabled=True), MessageBus()) fake_web = _FakeAsyncWebClient() fake_web._conversations_pages = [ { "channels": [{"id": "C999", "name": "channel_x"}], "response_metadata": {"next_cursor": ""}, } ] channel._web_client = fake_web await channel.send( OutboundMessage( channel="slack", chat_id="#channel_x", content="hello", ) ) assert fake_web.chat_post_calls == [ {"channel": "C999", "text": "hello\n", "thread_ts": None} ] assert len(fake_web.conversations_list_calls) == 1 @pytest.mark.asyncio async def test_send_resolves_user_handle_to_dm_channel() -> None: channel = SlackChannel(SlackConfig(enabled=True), MessageBus()) fake_web = _FakeAsyncWebClient() fake_web._users_pages = [ { "members": [ { "id": "U234", "name": "alice", "profile": {"display_name": "Alice"}, } ], "response_metadata": {"next_cursor": ""}, } ] fake_web._open_dm_response = {"channel": {"id": "D234"}} channel._web_client = fake_web await channel.send( OutboundMessage( channel="slack", chat_id="@alice", content="hello", ) ) assert fake_web.conversations_open_calls == [{"users": "U234"}] assert fake_web.chat_post_calls == [ {"channel": "D234", "text": "hello\n", "thread_ts": None} ] @pytest.mark.asyncio async def test_send_updates_reaction_on_origin_channel_for_cross_channel_send() -> None: channel = SlackChannel(SlackConfig(enabled=True, react_emoji="eyes"), MessageBus()) fake_web = _FakeAsyncWebClient() fake_web._conversations_pages = [ { "channels": [{"id": "C999", "name": "channel_x"}], "response_metadata": {"next_cursor": ""}, } ] channel._web_client = fake_web await channel.send( OutboundMessage( channel="slack", chat_id="channel_x", content="done", metadata={ "slack": { "event": {"ts": "1700000000.000100", "channel": "D_ORIGIN"}, "channel_type": "im", }, }, ) ) assert fake_web.chat_post_calls == [ {"channel": "C999", "text": "done\n", "thread_ts": None} ] assert fake_web.reactions_remove_calls == [ {"channel": "D_ORIGIN", "name": "eyes", "timestamp": "1700000000.000100"} ] assert fake_web.reactions_add_calls == [ {"channel": "D_ORIGIN", "name": "white_check_mark", "timestamp": "1700000000.000100"} ] @pytest.mark.asyncio async def test_send_does_not_reuse_origin_thread_ts_for_cross_channel_send() -> None: channel = SlackChannel(SlackConfig(enabled=True), MessageBus()) fake_web = _FakeAsyncWebClient() fake_web._conversations_pages = [ { "channels": [{"id": "C999", "name": "channel_x"}], "response_metadata": {"next_cursor": ""}, } ] channel._web_client = fake_web await channel.send( OutboundMessage( channel="slack", chat_id="channel_x", content="done", metadata={ "slack": { "event": {"ts": "1700000000.000100", "channel": "C_ORIGIN"}, "thread_ts": "1700000000.000200", "channel_type": "channel", }, }, ) ) assert fake_web.chat_post_calls == [ {"channel": "C999", "text": "done\n", "thread_ts": None} ] @pytest.mark.asyncio async def test_send_raises_when_named_target_cannot_be_resolved() -> None: channel = SlackChannel(SlackConfig(enabled=True), MessageBus()) fake_web = _FakeAsyncWebClient() channel._web_client = fake_web with pytest.raises(ValueError, match="was not found"): await channel.send( OutboundMessage( channel="slack", chat_id="#missing-channel", content="hello", ) ) @pytest.mark.asyncio async def test_with_thread_context_fetches_root_once() -> None: channel = SlackChannel(SlackConfig(enabled=True), MessageBus()) channel._bot_user_id = "UBOT" fake_web = _FakeAsyncWebClient() fake_web._conversations_replies_response = { "messages": [ {"ts": "111.000", "user": "UROOT", "text": "drink water"}, {"ts": "112.000", "user": "U2", "text": "good idea"}, {"ts": "113.000", "user": "U3", "text": "<@UBOT> what did you see?"}, ] } channel._web_client = fake_web content = await channel._with_thread_context( "what did you see?", chat_id="C123", channel_type="channel", thread_ts="111.000", raw_thread_ts="111.000", current_ts="113.000", ) assert fake_web.conversations_replies_calls == [ {"channel": "C123", "ts": "111.000", "limit": 20} ] assert "Slack thread context before this mention:" in content assert "- <@UROOT>: drink water" in content assert "- <@U2>: good idea" in content assert "U3" not in content assert content.endswith("Current message:\nwhat did you see?") second = await channel._with_thread_context( "again", chat_id="C123", channel_type="channel", thread_ts="111.000", raw_thread_ts="111.000", current_ts="114.000", ) assert second == "again" assert len(fake_web.conversations_replies_calls) == 1