107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
"""Test slack notifications."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from unittest.mock import AsyncMock, Mock
|
|
|
|
from _pytest.logging import LogCaptureFixture
|
|
|
|
from homeassistant.components import notify
|
|
from homeassistant.components.slack import DOMAIN
|
|
from homeassistant.components.slack.notify import (
|
|
CONF_DEFAULT_CHANNEL,
|
|
SlackNotificationService,
|
|
)
|
|
from homeassistant.const import ATTR_ICON, CONF_API_KEY, CONF_NAME, CONF_PLATFORM
|
|
|
|
from . import CONF_DATA
|
|
|
|
MODULE_PATH = "homeassistant.components.slack.notify"
|
|
SERVICE_NAME = f"notify_{DOMAIN}"
|
|
|
|
DEFAULT_CONFIG = {
|
|
notify.DOMAIN: [
|
|
{
|
|
CONF_PLATFORM: DOMAIN,
|
|
CONF_NAME: SERVICE_NAME,
|
|
CONF_API_KEY: "12345",
|
|
CONF_DEFAULT_CHANNEL: "channel",
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
def filter_log_records(caplog: LogCaptureFixture) -> list[logging.LogRecord]:
|
|
"""Filter all unrelated log records."""
|
|
return [
|
|
rec for rec in caplog.records if rec.name.endswith(f"{DOMAIN}.{notify.DOMAIN}")
|
|
]
|
|
|
|
|
|
async def test_message_includes_default_emoji():
|
|
"""Tests that default icon is used when no message icon is given."""
|
|
mock_client = Mock()
|
|
mock_client.chat_postMessage = AsyncMock()
|
|
expected_icon = ":robot_face:"
|
|
service = SlackNotificationService(
|
|
None, mock_client, CONF_DATA | {ATTR_ICON: expected_icon}
|
|
)
|
|
|
|
await service.async_send_message("test")
|
|
|
|
mock_fn = mock_client.chat_postMessage
|
|
mock_fn.assert_called_once()
|
|
_, kwargs = mock_fn.call_args
|
|
assert kwargs["icon_emoji"] == expected_icon
|
|
|
|
|
|
async def test_message_emoji_overrides_default():
|
|
"""Tests that overriding the default icon emoji when sending a message works."""
|
|
mock_client = Mock()
|
|
mock_client.chat_postMessage = AsyncMock()
|
|
service = SlackNotificationService(
|
|
None, mock_client, CONF_DATA | {ATTR_ICON: "default_icon"}
|
|
)
|
|
|
|
expected_icon = ":new:"
|
|
await service.async_send_message("test", data={"icon": expected_icon})
|
|
|
|
mock_fn = mock_client.chat_postMessage
|
|
mock_fn.assert_called_once()
|
|
_, kwargs = mock_fn.call_args
|
|
assert kwargs["icon_emoji"] == expected_icon
|
|
|
|
|
|
async def test_message_includes_default_icon_url():
|
|
"""Tests that overriding the default icon url when sending a message works."""
|
|
mock_client = Mock()
|
|
mock_client.chat_postMessage = AsyncMock()
|
|
expected_icon = "https://example.com/hass.png"
|
|
service = SlackNotificationService(
|
|
None, mock_client, CONF_DATA | {ATTR_ICON: expected_icon}
|
|
)
|
|
|
|
await service.async_send_message("test")
|
|
|
|
mock_fn = mock_client.chat_postMessage
|
|
mock_fn.assert_called_once()
|
|
_, kwargs = mock_fn.call_args
|
|
assert kwargs["icon_url"] == expected_icon
|
|
|
|
|
|
async def test_message_icon_url_overrides_default():
|
|
"""Tests that overriding the default icon url when sending a message works."""
|
|
mock_client = Mock()
|
|
mock_client.chat_postMessage = AsyncMock()
|
|
service = SlackNotificationService(
|
|
None, mock_client, CONF_DATA | {ATTR_ICON: "default_icon"}
|
|
)
|
|
|
|
expected_icon = "https://example.com/hass.png"
|
|
await service.async_send_message("test", data={ATTR_ICON: expected_icon})
|
|
|
|
mock_fn = mock_client.chat_postMessage
|
|
mock_fn.assert_called_once()
|
|
_, kwargs = mock_fn.call_args
|
|
assert kwargs["icon_url"] == expected_icon
|