Clean up async_reproduce_state helper usage (#68617)

This commit is contained in:
Franck Nijhof 2022-03-24 14:40:54 +01:00 committed by GitHub
parent cbf5b5ead5
commit 8aff8d89d2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
22 changed files with 159 additions and 126 deletions

View file

@ -1,5 +1,6 @@
"""Test reproduce state for Input datetime."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -28,7 +29,8 @@ async def test_reproducing_states(hass, caplog):
datetime_calls = async_mock_service(hass, "input_datetime", "set_datetime")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_datetime.entity_datetime", "2010-10-10 01:20:00"),
State("input_datetime.entity_time", "01:20:00"),
@ -39,7 +41,8 @@ async def test_reproducing_states(hass, caplog):
assert len(datetime_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_datetime.entity_datetime", "not_supported"),
State("input_datetime.entity_datetime", "not-valid-date"),
@ -55,7 +58,8 @@ async def test_reproducing_states(hass, caplog):
assert len(datetime_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_datetime.entity_datetime", "2011-10-10 02:20:00"),
State("input_datetime.entity_time", "02:20:00"),