Merge tests

This commit is contained in:
abmantis 2025-07-22 15:56:12 +01:00
parent 09328aabc0
commit 35b54b7e04

View File

@ -96,7 +96,7 @@ async def target_lights(hass: HomeAssistant) -> None:
], ],
) )
@pytest.mark.parametrize("state", [STATE_ON, STATE_OFF]) @pytest.mark.parametrize("state", [STATE_ON, STATE_OFF])
async def test_light_state_trigger( async def test_light_state_trigger_behavior_any(
hass: HomeAssistant, hass: HomeAssistant,
service_calls: list[ServiceCall], service_calls: list[ServiceCall],
trigger_target_config: dict, trigger_target_config: dict,
@ -113,78 +113,42 @@ async def test_light_state_trigger(
hass, hass,
automation.DOMAIN, automation.DOMAIN,
{ {
automation.DOMAIN: { automation.DOMAIN: [
"trigger": { {
CONF_PLATFORM: "light.state", "alias": "Trigger when state changes to specific state",
CONF_STATE: state, "trigger": {
**trigger_target_config, CONF_PLATFORM: "light.state",
CONF_STATE: state,
**trigger_target_config,
},
"action": {
"service": "test.automation",
"data_template": {CONF_ENTITY_ID: f"{entity_id}"},
},
}, },
"action": { {
"service": "test.automation", "alias": "Trigger when state changes to any state",
"data_template": {CONF_ENTITY_ID: f"{entity_id}"}, "trigger": {
CONF_PLATFORM: "light.state",
**trigger_target_config,
},
"action": {
"service": "test.automation",
"data_template": {CONF_ENTITY_ID: f"{entity_id}"},
},
}, },
} ]
}, },
) )
hass.states.async_set(entity_id, state) hass.states.async_set(entity_id, state)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(service_calls) == 1 assert len(service_calls) == 2
assert service_calls[0].data[CONF_ENTITY_ID] == entity_id assert service_calls[0].data[CONF_ENTITY_ID] == entity_id
assert service_calls[1].data[CONF_ENTITY_ID] == entity_id
service_calls.clear()
hass.states.async_set(entity_id, reverse_state) hass.states.async_set(entity_id, reverse_state)
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(service_calls) == 1 assert len(service_calls) == 1
@pytest.mark.usefixtures("target_lights")
@pytest.mark.parametrize(
("trigger_target_config", "entity_id"),
[
({CONF_ENTITY_ID: "light.standalone_light"}, "light.standalone_light"),
({ATTR_LABEL_ID: "test_label"}, "light.label_light"),
({ATTR_AREA_ID: "test_area"}, "light.area_light"),
({ATTR_FLOOR_ID: "test_floor"}, "light.area_light"),
({ATTR_LABEL_ID: "test_label"}, "light.device_light"),
({ATTR_AREA_ID: "test_area"}, "light.device_light"),
({ATTR_FLOOR_ID: "test_floor"}, "light.device_light"),
({ATTR_DEVICE_ID: "test_device"}, "light.device_light"),
],
)
async def test_light_state_trigger_any_state(
hass: HomeAssistant,
service_calls: list[ServiceCall],
trigger_target_config: dict,
entity_id: str,
) -> None:
"""Test that the light state trigger fires when light state changes to any state."""
await async_setup_component(hass, "light", {})
hass.states.async_set(entity_id, STATE_OFF)
await async_setup_component(
hass,
automation.DOMAIN,
{
automation.DOMAIN: {
"trigger": {
CONF_PLATFORM: "light.state",
**trigger_target_config,
},
"action": {
"service": "test.automation",
"data_template": {CONF_ENTITY_ID: f"{entity_id}"},
},
}
},
)
hass.states.async_set(entity_id, STATE_ON)
await hass.async_block_till_done()
assert len(service_calls) == 1
assert service_calls[0].data[CONF_ENTITY_ID] == entity_id assert service_calls[0].data[CONF_ENTITY_ID] == entity_id
hass.states.async_set(entity_id, STATE_OFF)
await hass.async_block_till_done()
assert len(service_calls) == 2
assert service_calls[1].data[CONF_ENTITY_ID] == entity_id