mirror of https://github.com/home-assistant/core
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""Test reproduce state for Automation."""
|
|
|
|
import pytest
|
|
|
|
from homeassistant.core import HomeAssistant, State
|
|
from homeassistant.helpers.state import async_reproduce_state
|
|
|
|
from tests.common import async_mock_service
|
|
|
|
|
|
async def test_reproducing_states(
|
|
hass: HomeAssistant, caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""Test reproducing Automation states."""
|
|
hass.states.async_set("automation.entity_off", "off", {})
|
|
hass.states.async_set("automation.entity_on", "on", {})
|
|
|
|
turn_on_calls = async_mock_service(hass, "automation", "turn_on")
|
|
turn_off_calls = async_mock_service(hass, "automation", "turn_off")
|
|
|
|
# These calls should do nothing as entities already in desired state
|
|
await async_reproduce_state(
|
|
hass,
|
|
[State("automation.entity_off", "off"), State("automation.entity_on", "on")],
|
|
)
|
|
|
|
assert len(turn_on_calls) == 0
|
|
assert len(turn_off_calls) == 0
|
|
|
|
# Test invalid state is handled
|
|
await async_reproduce_state(hass, [State("automation.entity_off", "not_supported")])
|
|
|
|
assert "not_supported" in caplog.text
|
|
assert len(turn_on_calls) == 0
|
|
assert len(turn_off_calls) == 0
|
|
|
|
# Make sure correct services are called
|
|
await async_reproduce_state(
|
|
hass,
|
|
[
|
|
State("automation.entity_on", "off"),
|
|
State("automation.entity_off", "on"),
|
|
# Should not raise
|
|
State("automation.non_existing", "on"),
|
|
],
|
|
)
|
|
|
|
assert len(turn_on_calls) == 1
|
|
assert turn_on_calls[0].domain == "automation"
|
|
assert turn_on_calls[0].data == {"entity_id": "automation.entity_off"}
|
|
|
|
assert len(turn_off_calls) == 1
|
|
assert turn_off_calls[0].domain == "automation"
|
|
assert turn_off_calls[0].data == {"entity_id": "automation.entity_on"}
|