ab889f9f1d
# PR Description
* Adds/updates the following files to all toolkits:
- `.pre-commit-config.yaml`
- `.ruff.toml`
- `LICENSE`
- `Makefile`
- `pyproject.toml`
* Lint all toolkits such that they pass `make check` and `make test` (a
total doozy). This includes adding some unit tests and evals.
* Github workflow for testing toolkits before merge into main (courtesy
of @sdreyer)
* Added a QOL improvement for tool developers for when they need to get
the context's auth token.
* Minor updates to `arcade new` template.
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from arcade.sdk import ToolAuthorizationContext, ToolContext
|
|
|
|
from arcade_slack.tools.chat import send_dm_to_user, send_message_to_channel
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_context():
|
|
mock_auth = ToolAuthorizationContext(token="fake-token") # noqa: S106
|
|
return ToolContext(authorization=mock_auth)
|
|
|
|
|
|
def test_send_dm_to_user(mock_context):
|
|
with patch("arcade_slack.tools.chat.WebClient") as MockWebClient:
|
|
mock_client = MockWebClient.return_value
|
|
mock_client.users_list.return_value = {"members": [{"name": "testuser", "id": "U12345"}]}
|
|
mock_client.conversations_open.return_value = {"channel": {"id": "D12345"}}
|
|
mock_client.chat_postMessage.return_value = MagicMock(data={"ok": True})
|
|
|
|
response = send_dm_to_user(mock_context, "testuser", "Hello!")
|
|
|
|
assert response["ok"] is True
|
|
mock_client.users_list.assert_called_once()
|
|
mock_client.conversations_open.assert_called_once_with(users=["U12345"])
|
|
mock_client.chat_postMessage.assert_called_once_with(channel="D12345", text="Hello!")
|
|
|
|
|
|
def test_send_message_to_channel(mock_context):
|
|
with patch("arcade_slack.tools.chat.WebClient") as MockWebClient:
|
|
mock_client = MockWebClient.return_value
|
|
mock_client.conversations_list.return_value = {
|
|
"channels": [{"name": "general", "id": "C12345"}]
|
|
}
|
|
mock_client.chat_postMessage.return_value = MagicMock(data={"ok": True})
|
|
|
|
response = send_message_to_channel(mock_context, "general", "Hello, channel!")
|
|
|
|
assert response["ok"] is True
|
|
mock_client.conversations_list.assert_called_once()
|
|
mock_client.chat_postMessage.assert_called_once_with(
|
|
channel="C12345", text="Hello, channel!"
|
|
)
|