"""Tests for the ntfy notify platform.""" from collections.abc import AsyncGenerator from unittest.mock import patch from aiontfy import Message from aiontfy.exceptions import NtfyException, NtfyHTTPError from freezegun.api import freeze_time import pytest from syrupy.assertion import SnapshotAssertion from homeassistant.components.notify import ( ATTR_MESSAGE, ATTR_TITLE, DOMAIN as NOTIFY_DOMAIN, SERVICE_SEND_MESSAGE, ) from homeassistant.config_entries import ConfigEntryState from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN, Platform from homeassistant.core import HomeAssistant from homeassistant.exceptions import HomeAssistantError from homeassistant.helpers import entity_registry as er from tests.common import AsyncMock, MockConfigEntry, snapshot_platform @pytest.fixture(autouse=True) async def notify_only() -> AsyncGenerator[None]: """Enable only the notify platform.""" with patch( "homeassistant.components.ntfy.PLATFORMS", [Platform.NOTIFY], ): yield @pytest.mark.usefixtures("mock_aiontfy") async def test_notify_platform( hass: HomeAssistant, config_entry: MockConfigEntry, snapshot: SnapshotAssertion, entity_registry: er.EntityRegistry, ) -> None: """Test setup of the ntfy notify platform.""" config_entry.add_to_hass(hass) await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() assert config_entry.state is ConfigEntryState.LOADED await snapshot_platform(hass, entity_registry, snapshot, config_entry.entry_id) @freeze_time("2025-01-09T12:00:00+00:00") async def test_send_message( hass: HomeAssistant, config_entry: MockConfigEntry, mock_aiontfy: AsyncMock, ) -> None: """Test publishing ntfy message.""" config_entry.add_to_hass(hass) await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() assert config_entry.state is ConfigEntryState.LOADED state = hass.states.get("notify.mytopic") assert state assert state.state == STATE_UNKNOWN await hass.services.async_call( NOTIFY_DOMAIN, SERVICE_SEND_MESSAGE, { ATTR_ENTITY_ID: "notify.mytopic", ATTR_MESSAGE: "triggered", ATTR_TITLE: "test", }, blocking=True, ) state = hass.states.get("notify.mytopic") assert state assert state.state == "2025-01-09T12:00:00+00:00" mock_aiontfy.publish.assert_called_once_with( Message(topic="mytopic", message="triggered", title="test") ) @pytest.mark.parametrize( ("exception", "error_msg"), [ ( NtfyHTTPError(41801, 418, "I'm a teapot", ""), "Failed to publish notification: I'm a teapot", ), ( NtfyException, "Failed to publish notification due to a connection error", ), ], ) async def test_send_message_exception( hass: HomeAssistant, config_entry: MockConfigEntry, mock_aiontfy: AsyncMock, exception: Exception, error_msg: str, ) -> None: """Test publish message exceptions.""" config_entry.add_to_hass(hass) await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() assert config_entry.state is ConfigEntryState.LOADED mock_aiontfy.publish.side_effect = exception with pytest.raises(HomeAssistantError, match=error_msg): await hass.services.async_call( NOTIFY_DOMAIN, SERVICE_SEND_MESSAGE, { ATTR_ENTITY_ID: "notify.mytopic", ATTR_MESSAGE: "triggered", ATTR_TITLE: "test", }, blocking=True, ) mock_aiontfy.publish.assert_called_once_with( Message(topic="mytopic", message="triggered", title="test") )