"""The tests for the notify.persistent_notification service."""
from homeassistant.components import notify
import homeassistant.components.persistent_notification as pn
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component

from tests.common import async_get_persistent_notifications


async def test_async_send_message(hass: HomeAssistant) -> None:
    """Test sending a message to notify.persistent_notification service."""
    await async_setup_component(hass, pn.DOMAIN, {"core": {}})
    await async_setup_component(hass, notify.DOMAIN, {})
    await hass.async_block_till_done()

    message = {"message": "Hello", "title": "Test notification"}
    await hass.services.async_call(
        notify.DOMAIN, notify.SERVICE_PERSISTENT_NOTIFICATION, message
    )
    await hass.async_block_till_done()

    notifications = async_get_persistent_notifications(hass)
    assert len(notifications) == 1
    notification = notifications[list(notifications)[0]]

    assert notification["message"] == "Hello"
    assert notification["title"] == "Test notification"


async def test_async_supports_notification_id(hass: HomeAssistant) -> None:
    """Test that notify.persistent_notification supports notification_id."""
    await async_setup_component(hass, pn.DOMAIN, {"core": {}})
    await async_setup_component(hass, notify.DOMAIN, {})
    await hass.async_block_till_done()

    message = {
        "message": "Hello",
        "title": "Test notification",
        "data": {"notification_id": "my_id"},
    }
    await hass.services.async_call(
        notify.DOMAIN, notify.SERVICE_PERSISTENT_NOTIFICATION, message
    )
    await hass.async_block_till_done()

    notifications = async_get_persistent_notifications(hass)
    assert len(notifications) == 1

    # Send second message with same ID

    message = {
        "message": "Goodbye",
        "title": "Notification was updated",
        "data": {"notification_id": "my_id"},
    }
    await hass.services.async_call(
        notify.DOMAIN, notify.SERVICE_PERSISTENT_NOTIFICATION, message
    )
    await hass.async_block_till_done()

    notifications = async_get_persistent_notifications(hass)
    assert len(notifications) == 1

    notification = notifications[list(notifications)[0]]
    assert notification["message"] == "Goodbye"
    assert notification["title"] == "Notification was updated"