mirror of https://github.com/home-assistant/core
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
"""The tests for the Event automation."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import automation
|
|
from homeassistant.core import CoreState, HomeAssistant
|
|
from homeassistant.helpers.typing import ConfigType
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import async_mock_service
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"hass_config",
|
|
[
|
|
{
|
|
automation.DOMAIN: {
|
|
"alias": "hello",
|
|
"trigger": {"platform": "homeassistant", "event": "start"},
|
|
"action": {
|
|
"service": "test.automation",
|
|
"data_template": {"id": "{{ trigger.id}}"},
|
|
},
|
|
}
|
|
}
|
|
],
|
|
)
|
|
@pytest.mark.usefixtures("mock_hass_config")
|
|
async def test_if_fires_on_hass_start(
|
|
hass: HomeAssistant, hass_config: ConfigType
|
|
) -> None:
|
|
"""Test the firing when Home Assistant starts."""
|
|
calls = async_mock_service(hass, "test", "automation")
|
|
hass.set_state(CoreState.not_running)
|
|
|
|
assert await async_setup_component(hass, automation.DOMAIN, hass_config)
|
|
assert automation.is_on(hass, "automation.hello")
|
|
assert len(calls) == 0
|
|
|
|
await hass.async_start()
|
|
await hass.async_block_till_done()
|
|
assert automation.is_on(hass, "automation.hello")
|
|
assert len(calls) == 1
|
|
|
|
await hass.services.async_call(
|
|
automation.DOMAIN, automation.SERVICE_RELOAD, blocking=True
|
|
)
|
|
|
|
assert automation.is_on(hass, "automation.hello")
|
|
assert len(calls) == 1
|
|
assert calls[0].data["id"] == 0
|
|
|
|
|
|
async def test_if_fires_on_hass_shutdown(hass: HomeAssistant) -> None:
|
|
"""Test the firing when Home Assistant shuts down."""
|
|
calls = async_mock_service(hass, "test", "automation")
|
|
hass.set_state(CoreState.not_running)
|
|
|
|
assert await async_setup_component(
|
|
hass,
|
|
automation.DOMAIN,
|
|
{
|
|
automation.DOMAIN: {
|
|
"alias": "hello",
|
|
"trigger": {"platform": "homeassistant", "event": "shutdown"},
|
|
"action": {
|
|
"service": "test.automation",
|
|
"data_template": {"id": "{{ trigger.id}}"},
|
|
},
|
|
}
|
|
},
|
|
)
|
|
assert automation.is_on(hass, "automation.hello")
|
|
assert len(calls) == 0
|
|
|
|
await hass.async_start()
|
|
assert automation.is_on(hass, "automation.hello")
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 0
|
|
|
|
with patch.object(hass.loop, "stop"):
|
|
await hass.async_stop()
|
|
assert len(calls) == 1
|
|
assert calls[0].data["id"] == 0
|