evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Clean up async_reproduce_state helper usage (#68617)

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

11
script/scaffold/templates/reproduce_state/tests/test_reproduce_state.py
View file

@ -2,6 +2,7 @@
import pytest
from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -17,7 +18,8 @@ async def test_reproducing_states(
turn_off_calls = async_mock_service(hass, "NEW_DOMAIN", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("NEW_DOMAIN.entity_off", "off"),
State("NEW_DOMAIN.entity_on", "on", {"color": "red"}),
@ -29,8 +31,8 @@ async def test_reproducing_states(
assert len(turn_off_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("NEW_DOMAIN.entity_off", "not_supported")], blocking=True
await async_reproduce_state(
hass, [State("NEW_DOMAIN.entity_off", "not_supported")], blocking=True
)
assert "not_supported" in caplog.text
@ -38,7 +40,8 @@ async def test_reproducing_states(
assert len(turn_off_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("NEW_DOMAIN.entity_on", "off"),
State("NEW_DOMAIN.entity_off", "on", {"color": "red"}),

15
tests/components/alarm_control_panel/test_reproduce_state.py
View file

@ -16,6 +16,7 @@ from homeassistant.const import (
STATE_ALARM_TRIGGERED,
)
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -67,7 +68,8 @@ async def test_reproducing_states(hass, caplog):
)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("alarm_control_panel.entity_armed_away", STATE_ALARM_ARMED_AWAY),
State(
@ -81,7 +83,7 @@ async def test_reproducing_states(hass, caplog):
),
State("alarm_control_panel.entity_disarmed", STATE_ALARM_DISARMED),
State("alarm_control_panel.entity_triggered", STATE_ALARM_TRIGGERED),
]
],
)
assert len(arm_away_calls) == 0
@ -93,8 +95,8 @@ async def test_reproducing_states(hass, caplog):
assert len(trigger_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("alarm_control_panel.entity_triggered", "not_supported")]
await async_reproduce_state(
hass, [State("alarm_control_panel.entity_triggered", "not_supported")]
)
assert "not_supported" in caplog.text
@ -107,7 +109,8 @@ async def test_reproducing_states(hass, caplog):
assert len(trigger_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("alarm_control_panel.entity_armed_away", STATE_ALARM_TRIGGERED),
State(
@ -122,7 +125,7 @@ async def test_reproducing_states(hass, caplog):
State("alarm_control_panel.entity_triggered", STATE_ALARM_DISARMED),
# Should not raise
State("alarm_control_panel.non_existing", "on"),
]
],
)
assert len(arm_away_calls) == 1

14
tests/components/alert/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Alert."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -13,30 +14,29 @@ async def test_reproducing_states(hass, caplog):
turn_off_calls = async_mock_service(hass, "alert", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
[State("alert.entity_off", "off"), State("alert.entity_on", "on")]
await async_reproduce_state(
hass, [State("alert.entity_off", "off"), State("alert.entity_on", "on")]
)
assert len(turn_on_calls) == 0
assert len(turn_off_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("alert.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("alert.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 hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("alert.entity_on", "off"),
State("alert.entity_off", "on"),
# Should not raise
State("alert.non_existing", "on"),
]
],
)
assert len(turn_on_calls) == 1

15
tests/components/automation/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Automation."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -13,30 +14,30 @@ async def test_reproducing_states(hass, caplog):
turn_off_calls = async_mock_service(hass, "automation", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
[State("automation.entity_off", "off"), State("automation.entity_on", "on")]
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 hass.helpers.state.async_reproduce_state(
[State("automation.entity_off", "not_supported")]
)
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 hass.helpers.state.async_reproduce_state(
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

15
tests/components/counter/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Counter."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -16,7 +17,8 @@ async def test_reproducing_states(hass, caplog):
configure_calls = async_mock_service(hass, "counter", "configure")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("counter.entity", "5"),
State(
@ -24,21 +26,20 @@ async def test_reproducing_states(hass, caplog):
"8",
{"initial": 12, "minimum": 5, "maximum": 15, "step": 3},
),
]
],
)
assert len(configure_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("counter.entity", "not_supported")]
)
await async_reproduce_state(hass, [State("counter.entity", "not_supported")])
assert "not_supported" in caplog.text
assert len(configure_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("counter.entity", "2"),
State(
@ -48,7 +49,7 @@ async def test_reproducing_states(hass, caplog):
),
# Should not raise
State("counter.non_existing", "6"),
]
],
)
valid_calls = [

13
tests/components/cover/test_reproduce_state.py
View file

@ -16,6 +16,7 @@ from homeassistant.const import (
STATE_OPEN,
)
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -61,7 +62,8 @@ async def test_reproducing_states(hass, caplog):
)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("cover.entity_close", STATE_CLOSED),
State(
@ -93,7 +95,7 @@ async def test_reproducing_states(hass, caplog):
STATE_OPEN,
{ATTR_CURRENT_POSITION: 100, ATTR_CURRENT_TILT_POSITION: 100},
),
]
],
)
assert len(close_calls) == 0
@ -104,9 +106,7 @@ async def test_reproducing_states(hass, caplog):
assert len(position_tilt_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("cover.entity_close", "not_supported")]
)
await async_reproduce_state(hass, [State("cover.entity_close", "not_supported")])
assert "not_supported" in caplog.text
assert len(close_calls) == 0
@ -117,7 +117,8 @@ async def test_reproducing_states(hass, caplog):
assert len(position_tilt_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("cover.entity_close", STATE_OPEN),
State(

11
tests/components/fan/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Fan."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -19,7 +20,8 @@ async def test_reproducing_states(hass, caplog):
set_percentage_calls = async_mock_service(hass, "fan", "set_percentage")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("fan.entity_off", "off"),
State("fan.entity_on", "on"),
@ -35,9 +37,7 @@ async def test_reproducing_states(hass, caplog):
assert len(oscillate_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("fan.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("fan.entity_off", "not_supported")])
assert "not_supported" in caplog.text
assert len(turn_on_calls) == 0
@ -47,7 +47,8 @@ async def test_reproducing_states(hass, caplog):
assert len(set_percentage_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("fan.entity_on", "off"),
State("fan.entity_off", "on"),

11
tests/components/humidifier/test_reproduce_state.py
View file

@ -20,6 +20,7 @@ from homeassistant.const import (
STATE_ON,
)
from homeassistant.core import Context, State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -38,7 +39,8 @@ async def test_reproducing_on_off_states(hass, caplog):
humidity_calls = async_mock_service(hass, DOMAIN, SERVICE_SET_HUMIDITY)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State(ENTITY_1, "off", {ATTR_MODE: MODE_NORMAL, ATTR_HUMIDITY: 45}),
State(ENTITY_2, "on", {ATTR_MODE: MODE_NORMAL, ATTR_HUMIDITY: 45}),
@ -51,7 +53,7 @@ async def test_reproducing_on_off_states(hass, caplog):
assert len(humidity_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state([State(ENTITY_1, "not_supported")])
await async_reproduce_state(hass, [State(ENTITY_1, "not_supported")])
assert "not_supported" in caplog.text
assert len(turn_on_calls) == 0
@ -60,13 +62,14 @@ async def test_reproducing_on_off_states(hass, caplog):
assert len(humidity_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State(ENTITY_2, "off"),
State(ENTITY_1, "on", {}),
# Should not raise
State("humidifier.non_existing", "on"),
]
],
)
assert len(turn_on_calls) == 1

7
tests/components/input_boolean/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for input boolean."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from homeassistant.setup import async_setup_component
@ -15,7 +16,8 @@ async def test_reproducing_states(hass):
}
},
)
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_boolean.initial_on", "off"),
State("input_boolean.initial_off", "on"),
@ -26,7 +28,8 @@ async def test_reproducing_states(hass):
assert hass.states.get("input_boolean.initial_off").state == "on"
assert hass.states.get("input_boolean.initial_on").state == "off"
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
# Test invalid state
State("input_boolean.initial_on", "invalid_state"),

10
tests/components/input_datetime/test_reproduce_state.py
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"),

14
tests/components/input_number/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Input number."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from homeassistant.setup import async_setup_component
VALID_NUMBER1 = "19.0"
@ -20,7 +21,8 @@ async def test_reproducing_states(hass, caplog):
)
# 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_number.test_number", VALID_NUMBER1),
# Should not raise
@ -31,7 +33,8 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get("input_number.test_number").state == VALID_NUMBER1
# Test reproducing with different state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_number.test_number", VALID_NUMBER2),
# Should not raise
@ -42,14 +45,13 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get("input_number.test_number").state == VALID_NUMBER2
# Test setting state to number out of range
await hass.helpers.state.async_reproduce_state(
[State("input_number.test_number", "150")]
)
await async_reproduce_state(hass, [State("input_number.test_number", "150")])
# The entity states should be unchanged after trying to set them to out-of-range number
assert hass.states.get("input_number.test_number").state == VALID_NUMBER2
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
# Test invalid state
State("input_number.test_number", "invalid_state"),

13
tests/components/input_select/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Input select."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from homeassistant.setup import async_setup_component
VALID_OPTION1 = "Option A"
@ -29,7 +30,8 @@ async def test_reproducing_states(hass, caplog):
)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State(ENTITY, VALID_OPTION1),
# Should not raise
@ -41,7 +43,8 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get(ENTITY).state == VALID_OPTION1
# Try reproducing with different state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State(ENTITY, VALID_OPTION3),
# Should not raise
@ -53,14 +56,14 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get(ENTITY).state == VALID_OPTION3
# Test setting state to invalid state
await hass.helpers.state.async_reproduce_state([State(ENTITY, INVALID_OPTION)])
await async_reproduce_state(hass, [State(ENTITY, INVALID_OPTION)])
# The entity state should be unchanged
assert hass.states.get(ENTITY).state == VALID_OPTION3
# Test setting a different option set
await hass.helpers.state.async_reproduce_state(
[State(ENTITY, VALID_OPTION5, {"options": VALID_OPTION_SET2})]
await async_reproduce_state(
hass, [State(ENTITY, VALID_OPTION5, {"options": VALID_OPTION_SET2})]
)
# These should fail if options weren't changed to VALID_OPTION_SET2

15
tests/components/input_text/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Input text."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from homeassistant.setup import async_setup_component
VALID_TEXT1 = "Test text"
@ -23,7 +24,8 @@ async def test_reproducing_states(hass, caplog):
)
# 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_text.test_text", VALID_TEXT1),
# Should not raise
@ -35,7 +37,8 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get("input_text.test_text").state == VALID_TEXT1
# Try reproducing with different state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("input_text.test_text", VALID_TEXT2),
# Should not raise
@ -47,17 +50,13 @@ async def test_reproducing_states(hass, caplog):
assert hass.states.get("input_text.test_text").state == VALID_TEXT2
# Test setting state to invalid state (length too long)
await hass.helpers.state.async_reproduce_state(
[State("input_text.test_text", INVALID_TEXT1)]
)
await async_reproduce_state(hass, [State("input_text.test_text", INVALID_TEXT1)])
# The entity state should be unchanged
assert hass.states.get("input_text.test_text").state == VALID_TEXT2
# Test setting state to invalid state (length too short)
await hass.helpers.state.async_reproduce_state(
[State("input_text.test_text", INVALID_TEXT2)]
)
await async_reproduce_state(hass, [State("input_text.test_text", INVALID_TEXT2)])
# The entity state should be unchanged
assert hass.states.get("input_text.test_text").state == VALID_TEXT2

25
tests/components/light/test_reproduce_state.py
View file

@ -4,6 +4,7 @@ import pytest
from homeassistant.components import light
from homeassistant.components.light.reproduce_state import DEPRECATION_WARNING
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -43,7 +44,8 @@ async def test_reproducing_states(hass, caplog):
turn_off_calls = async_mock_service(hass, "light", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("light.entity_off", "off"),
State("light.entity_bright", "on", VALID_BRIGHTNESS),
@ -58,23 +60,22 @@ async def test_reproducing_states(hass, caplog):
State("light.entity_profile", "on", VALID_PROFILE),
State("light.entity_rgb", "on", VALID_RGB_COLOR),
State("light.entity_xy", "on", VALID_XY_COLOR),
]
],
)
assert len(turn_on_calls) == 0
assert len(turn_off_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("light.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("light.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 hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("light.entity_xy", "off"),
State("light.entity_off", "on", VALID_BRIGHTNESS),
@ -194,8 +195,8 @@ async def test_filter_color_modes(hass, caplog, color_mode):
turn_on_calls = async_mock_service(hass, "light", "turn_on")
await hass.helpers.state.async_reproduce_state(
[State("light.entity", "on", {**all_colors, "color_mode": color_mode})]
await async_reproduce_state(
hass, [State("light.entity", "on", {**all_colors, "color_mode": color_mode})]
)
expected_map = {
@ -225,8 +226,8 @@ async def test_filter_color_modes(hass, caplog, color_mode):
# This should do nothing, the light is already in the desired state
hass.states.async_set("light.entity", "on", {"color_mode": color_mode, **expected})
await hass.helpers.state.async_reproduce_state(
[State("light.entity", "on", {**expected, "color_mode": color_mode})]
await async_reproduce_state(
hass, [State("light.entity", "on", {**expected, "color_mode": color_mode})]
)
assert len(turn_on_calls) == 1
@ -235,8 +236,8 @@ async def test_deprecation_warning(hass, caplog):
"""Test deprecation warning."""
hass.states.async_set("light.entity_off", "off", {})
turn_on_calls = async_mock_service(hass, "light", "turn_on")
await hass.helpers.state.async_reproduce_state(
[State("light.entity_off", "on", {"brightness_pct": 80})]
await async_reproduce_state(
hass, [State("light.entity_off", "on", {"brightness_pct": 80})]
)
assert len(turn_on_calls) == 1
assert DEPRECATION_WARNING % ["brightness_pct"] in caplog.text

11
tests/components/lock/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Lock."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -13,7 +14,8 @@ async def test_reproducing_states(hass, caplog):
unlock_calls = async_mock_service(hass, "lock", "unlock")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("lock.entity_locked", "locked"),
State("lock.entity_unlocked", "unlocked", {}),
@ -24,16 +26,15 @@ async def test_reproducing_states(hass, caplog):
assert len(unlock_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("lock.entity_locked", "not_supported")]
)
await async_reproduce_state(hass, [State("lock.entity_locked", "not_supported")])
assert "not_supported" in caplog.text
assert len(lock_calls) == 0
assert len(unlock_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("lock.entity_locked", "unlocked"),
State("lock.entity_unlocked", "locked"),

11
tests/components/number/test_reproduce_state.py
View file

@ -6,6 +6,7 @@ from homeassistant.components.number.const import (
SERVICE_SET_VALUE,
)
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -21,7 +22,8 @@ async def test_reproducing_states(hass, caplog):
)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("number.test_number", VALID_NUMBER1),
# Should not raise
@ -33,7 +35,8 @@ async def test_reproducing_states(hass, caplog):
# Test reproducing with different state
calls = async_mock_service(hass, DOMAIN, SERVICE_SET_VALUE)
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("number.test_number", VALID_NUMBER2),
# Should not raise
@ -46,8 +49,6 @@ async def test_reproducing_states(hass, caplog):
assert calls[0].data == {"entity_id": "number.test_number", "value": VALID_NUMBER2}
# Test invalid state
await hass.helpers.state.async_reproduce_state(
[State("number.test_number", "invalid_state")]
)
await async_reproduce_state(hass, [State("number.test_number", "invalid_state")])
assert len(calls) == 1

11
tests/components/remote/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Remote."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -13,7 +14,8 @@ async def test_reproducing_states(hass, caplog):
turn_off_calls = async_mock_service(hass, "remote", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[State("remote.entity_off", "off"), State("remote.entity_on", "on")],
)
@ -21,16 +23,15 @@ async def test_reproducing_states(hass, caplog):
assert len(turn_off_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("remote.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("remote.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 hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("remote.entity_on", "off"),
State("remote.entity_off", "on", {}),

15
tests/components/select/test_reproduce_state.py
View file

@ -9,6 +9,7 @@ from homeassistant.components.select.const import (
)
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -24,7 +25,8 @@ async def test_reproducing_states(
{ATTR_OPTIONS: ["option_one", "option_two", "option_three"]},
)
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("select.test", "option_two"),
],
@ -35,7 +37,8 @@ async def test_reproducing_states(
assert calls[0].data == {ATTR_ENTITY_ID: "select.test", ATTR_OPTION: "option_two"}
# Calling it again should not do anything
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("select.test", "option_one"),
],
@ -43,15 +46,11 @@ async def test_reproducing_states(
assert len(calls) == 1
# Restoring an invalid state should not work either
await hass.helpers.state.async_reproduce_state(
[State("select.test", "option_four")]
)
await async_reproduce_state(hass, [State("select.test", "option_four")])
assert len(calls) == 1
assert "Invalid state specified" in caplog.text
# Restoring an state for an invalid entity ID logs a warning
await hass.helpers.state.async_reproduce_state(
[State("select.non_existing", "option_three")]
)
await async_reproduce_state(hass, [State("select.non_existing", "option_three")])
assert len(calls) == 1
assert "Unable to find entity" in caplog.text

13
tests/components/switch/test_reproduce_state.py
View file

@ -1,5 +1,6 @@
"""Test reproduce state for Switch."""
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -13,7 +14,8 @@ async def test_reproducing_states(hass, caplog):
turn_off_calls = async_mock_service(hass, "switch", "turn_off")
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[State("switch.entity_off", "off"), State("switch.entity_on", "on", {})],
)
@ -21,22 +23,21 @@ async def test_reproducing_states(hass, caplog):
assert len(turn_off_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("switch.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("switch.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 hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("switch.entity_on", "off"),
State("switch.entity_off", "on", {}),
# Should not raise
State("switch.non_existing", "on"),
]
],
)
assert len(turn_on_calls) == 1

11
tests/components/timer/test_reproduce_state.py
View file

@ -9,6 +9,7 @@ from homeassistant.components.timer import (
STATUS_PAUSED,
)
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -27,7 +28,8 @@ async def test_reproducing_states(hass, caplog):
cancel_calls = async_mock_service(hass, "timer", SERVICE_CANCEL)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("timer.entity_idle", STATUS_IDLE),
State("timer.entity_paused", STATUS_PAUSED),
@ -43,9 +45,7 @@ async def test_reproducing_states(hass, caplog):
assert len(cancel_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("timer.entity_idle", "not_supported")]
)
await async_reproduce_state(hass, [State("timer.entity_idle", "not_supported")])
assert "not_supported" in caplog.text
assert len(start_calls) == 0
@ -53,7 +53,8 @@ async def test_reproducing_states(hass, caplog):
assert len(cancel_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("timer.entity_idle", STATUS_ACTIVE, {ATTR_DURATION: "00:01:00"}),
State("timer.entity_paused", STATUS_ACTIVE),

11
tests/components/vacuum/test_reproduce_state.py
View file

@ -19,6 +19,7 @@ from homeassistant.const import (
STATE_PAUSED,
)
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -48,7 +49,8 @@ async def test_reproducing_states(hass, caplog):
fan_speed_calls = async_mock_service(hass, "vacuum", SERVICE_SET_FAN_SPEED)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("vacuum.entity_off", STATE_OFF),
State("vacuum.entity_on", STATE_ON),
@ -70,9 +72,7 @@ async def test_reproducing_states(hass, caplog):
assert len(fan_speed_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("vacuum.entity_off", "not_supported")]
)
await async_reproduce_state(hass, [State("vacuum.entity_off", "not_supported")])
assert "not_supported" in caplog.text
assert len(turn_on_calls) == 0
@ -84,7 +84,8 @@ async def test_reproducing_states(hass, caplog):
assert len(fan_speed_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("vacuum.entity_off", STATE_ON),
State("vacuum.entity_on", STATE_OFF),

13
tests/components/water_heater/test_reproduce_state.py
View file

@ -11,6 +11,7 @@ from homeassistant.components.water_heater import (
)
from homeassistant.const import SERVICE_TURN_OFF, SERVICE_TURN_ON, STATE_OFF, STATE_ON
from homeassistant.core import State
from homeassistant.helpers.state import async_reproduce_state
from tests.common import async_mock_service
@ -34,7 +35,8 @@ async def test_reproducing_states(hass, caplog):
set_away_calls = async_mock_service(hass, "water_heater", SERVICE_SET_AWAY_MODE)
# These calls should do nothing as entities already in desired state
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("water_heater.entity_off", STATE_OFF),
State("water_heater.entity_on", STATE_ON, {ATTR_TEMPERATURE: 45}),
@ -45,7 +47,7 @@ async def test_reproducing_states(hass, caplog):
STATE_ECO,
{ATTR_AWAY_MODE: True, ATTR_TEMPERATURE: 45},
),
]
],
)
assert len(turn_on_calls) == 0
@ -55,8 +57,8 @@ async def test_reproducing_states(hass, caplog):
assert len(set_away_calls) == 0
# Test invalid state is handled
await hass.helpers.state.async_reproduce_state(
[State("water_heater.entity_off", "not_supported")]
await async_reproduce_state(
hass, [State("water_heater.entity_off", "not_supported")]
)
assert "not_supported" in caplog.text
@ -67,7 +69,8 @@ async def test_reproducing_states(hass, caplog):
assert len(set_away_calls) == 0
# Make sure correct services are called
await hass.helpers.state.async_reproduce_state(
await async_reproduce_state(
hass,
[
State("water_heater.entity_on", STATE_OFF),
State("water_heater.entity_off", STATE_ON, {ATTR_TEMPERATURE: 45}),