mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-23 16:36:23 +00:00
test(slack): cover send_message MEDIA delivery
Add standalone-sender media cases and route coverage; point the non-media platform assertions at SMS now that Slack supports MEDIA.
This commit is contained in:
parent
5f4c952ab0
commit
8df8f86784
1 changed files with 251 additions and 0 deletions
251
tests/tools/test_slack_send_message_media.py
Normal file
251
tests/tools/test_slack_send_message_media.py
Normal file
|
|
@ -0,0 +1,251 @@
|
|||
"""Slack media delivery for send_message.
|
||||
|
||||
Covers ``plugins/platforms/slack/adapter.py::_standalone_send`` media path:
|
||||
text+file, media-only, caption-on-upload, missing-file warnings.
|
||||
|
||||
``slack_sdk`` is optional in CI, so tests inject a fake module into
|
||||
``sys.modules`` (same pattern as ``tests/gateway/test_slack.py``).
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import contextlib
|
||||
import os
|
||||
import sys
|
||||
import tempfile
|
||||
from types import ModuleType, SimpleNamespace
|
||||
from unittest.mock import AsyncMock, MagicMock
|
||||
|
||||
import pytest
|
||||
|
||||
from plugins.platforms.slack.adapter import _standalone_send
|
||||
|
||||
|
||||
def _pconfig(token: str = "xoxb-test"):
|
||||
return SimpleNamespace(token=token, extra={})
|
||||
|
||||
|
||||
def _tmpfile(suffix: str) -> str:
|
||||
f = tempfile.NamedTemporaryFile(suffix=suffix, delete=False)
|
||||
f.write(b"%PDF-1.4 test")
|
||||
f.close()
|
||||
return f.name
|
||||
|
||||
|
||||
def _mock_client(*, post_ok=True, upload_ok=True):
|
||||
client = MagicMock()
|
||||
client.chat_postMessage = AsyncMock(
|
||||
return_value={
|
||||
"ok": post_ok,
|
||||
"ts": "111.222",
|
||||
"error": None if post_ok else "channel_not_found",
|
||||
}
|
||||
)
|
||||
if upload_ok:
|
||||
client.files_upload_v2 = AsyncMock(
|
||||
return_value={
|
||||
"ok": True,
|
||||
"file": {
|
||||
"id": "F123",
|
||||
"timestamp": 1234567890,
|
||||
"shares": {"public": {"C012AB3CD": [{"ts": "333.444"}]}},
|
||||
},
|
||||
}
|
||||
)
|
||||
else:
|
||||
client.files_upload_v2 = AsyncMock(
|
||||
return_value={"ok": False, "error": "not_in_channel"}
|
||||
)
|
||||
return client
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
def _fake_slack_sdk(client):
|
||||
"""Make ``from slack_sdk.web.async_client import AsyncWebClient`` resolve to a factory."""
|
||||
sdk = ModuleType("slack_sdk")
|
||||
web = ModuleType("slack_sdk.web")
|
||||
async_client = ModuleType("slack_sdk.web.async_client")
|
||||
async_client.AsyncWebClient = MagicMock(return_value=client)
|
||||
sdk.web = web
|
||||
web.async_client = async_client
|
||||
|
||||
modules = {
|
||||
"slack_sdk": sdk,
|
||||
"slack_sdk.web": web,
|
||||
"slack_sdk.web.async_client": async_client,
|
||||
}
|
||||
old = {name: sys.modules.get(name) for name in modules}
|
||||
sys.modules.update(modules)
|
||||
try:
|
||||
yield
|
||||
finally:
|
||||
for name, prev in old.items():
|
||||
if prev is None:
|
||||
sys.modules.pop(name, None)
|
||||
else:
|
||||
sys.modules[name] = prev
|
||||
|
||||
|
||||
def test_text_plus_pdf_uploads_via_files_upload_v2():
|
||||
pdf = _tmpfile(".pdf")
|
||||
client = _mock_client()
|
||||
try:
|
||||
with _fake_slack_sdk(client):
|
||||
result = asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"Here is the report",
|
||||
media_files=[(pdf, False)],
|
||||
)
|
||||
)
|
||||
assert result["success"] is True
|
||||
assert result["platform"] == "slack"
|
||||
client.chat_postMessage.assert_awaited_once()
|
||||
client.files_upload_v2.assert_awaited_once()
|
||||
upload_kwargs = client.files_upload_v2.await_args.kwargs
|
||||
assert upload_kwargs["channel"] == "C012AB3CD"
|
||||
assert upload_kwargs["file"] == pdf
|
||||
assert upload_kwargs["filename"] == os.path.basename(pdf)
|
||||
assert upload_kwargs["initial_comment"] == ""
|
||||
finally:
|
||||
os.unlink(pdf)
|
||||
|
||||
|
||||
def test_media_only_skips_text_post():
|
||||
pdf = _tmpfile(".pdf")
|
||||
client = _mock_client()
|
||||
try:
|
||||
with _fake_slack_sdk(client):
|
||||
result = asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"",
|
||||
media_files=[(pdf, False)],
|
||||
)
|
||||
)
|
||||
assert result["success"] is True
|
||||
client.chat_postMessage.assert_not_awaited()
|
||||
client.files_upload_v2.assert_awaited_once()
|
||||
finally:
|
||||
os.unlink(pdf)
|
||||
|
||||
|
||||
def test_caption_rides_initial_comment_no_separate_text():
|
||||
pdf = _tmpfile(".pdf")
|
||||
client = _mock_client()
|
||||
try:
|
||||
with _fake_slack_sdk(client):
|
||||
result = asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"",
|
||||
media_files=[(pdf, False)],
|
||||
caption="Q3 summary PDF",
|
||||
)
|
||||
)
|
||||
assert result["success"] is True
|
||||
client.chat_postMessage.assert_not_awaited()
|
||||
upload_kwargs = client.files_upload_v2.await_args.kwargs
|
||||
assert upload_kwargs["initial_comment"] == "Q3 summary PDF"
|
||||
finally:
|
||||
os.unlink(pdf)
|
||||
|
||||
|
||||
def test_missing_media_file_warns_and_falls_back_caption():
|
||||
client = _mock_client()
|
||||
with _fake_slack_sdk(client):
|
||||
result = asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"",
|
||||
media_files=[("/no/such/file.pdf", False)],
|
||||
caption="still deliver this",
|
||||
)
|
||||
)
|
||||
assert result["success"] is True
|
||||
assert result.get("warnings")
|
||||
assert any("not found" in w.lower() for w in result["warnings"])
|
||||
client.chat_postMessage.assert_awaited_once()
|
||||
assert client.chat_postMessage.await_args.kwargs["text"] == "still deliver this"
|
||||
client.files_upload_v2.assert_not_awaited()
|
||||
|
||||
|
||||
def test_missing_token_errors(monkeypatch):
|
||||
monkeypatch.delenv("SLACK_BOT_TOKEN", raising=False)
|
||||
result = asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(token=""),
|
||||
"C012AB3CD",
|
||||
"hi",
|
||||
media_files=[("/tmp/x.pdf", False)],
|
||||
)
|
||||
)
|
||||
assert "error" in result
|
||||
assert "SLACK_BOT_TOKEN" in result["error"]
|
||||
|
||||
|
||||
def test_thread_id_passed_to_upload():
|
||||
pdf = _tmpfile(".pdf")
|
||||
client = _mock_client()
|
||||
try:
|
||||
with _fake_slack_sdk(client):
|
||||
asyncio.run(
|
||||
_standalone_send(
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"",
|
||||
thread_id="999.000",
|
||||
media_files=[(pdf, False)],
|
||||
)
|
||||
)
|
||||
assert client.files_upload_v2.await_args.kwargs["thread_ts"] == "999.000"
|
||||
finally:
|
||||
os.unlink(pdf)
|
||||
|
||||
|
||||
def test_send_to_platform_routes_slack_media():
|
||||
"""_send_to_platform must call Slack standalone_sender with media_files."""
|
||||
import httpx
|
||||
|
||||
if not hasattr(httpx, "Proxy") or not hasattr(httpx, "URL"):
|
||||
pytest.skip("httpx type annotations incompatible with telegram library")
|
||||
|
||||
from gateway.config import Platform
|
||||
from hermes_cli.plugins import discover_plugins
|
||||
from gateway.platform_registry import platform_registry
|
||||
from tools.send_message_tool import _send_to_platform
|
||||
|
||||
pdf = _tmpfile(".pdf")
|
||||
discover_plugins()
|
||||
entry = platform_registry.get("slack")
|
||||
assert entry is not None and entry.standalone_sender_fn is not None
|
||||
original = entry.standalone_sender_fn
|
||||
mock_sender = AsyncMock(
|
||||
return_value={"success": True, "platform": "slack", "message_id": "1.2"}
|
||||
)
|
||||
entry.standalone_sender_fn = mock_sender
|
||||
try:
|
||||
result = asyncio.run(
|
||||
_send_to_platform(
|
||||
Platform.SLACK,
|
||||
_pconfig(),
|
||||
"C012AB3CD",
|
||||
"Here is the report",
|
||||
media_files=[(pdf, False)],
|
||||
)
|
||||
)
|
||||
assert result["success"] is True
|
||||
mock_sender.assert_awaited()
|
||||
call_kwargs = mock_sender.await_args.kwargs
|
||||
assert call_kwargs.get("media_files") == [(pdf, False)]
|
||||
# Single captionable file + short text → caption rides the upload.
|
||||
assert call_kwargs.get("caption") == "Here is the report"
|
||||
assert not result.get("warnings")
|
||||
finally:
|
||||
entry.standalone_sender_fn = original
|
||||
os.unlink(pdf)
|
||||
Loading…
Add table
Add a link
Reference in a new issue