From 56f6522c00be1b2a8ab6ef831745b6aaa2fe616d Mon Sep 17 00:00:00 2001 From: Erik Date: Mon, 6 Dec 2021 12:05:33 +0100 Subject: [PATCH] Adjust tests --- tests/components/device_automation/test_init.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/components/device_automation/test_init.py b/tests/components/device_automation/test_init.py index 563611b99a..44a2c57488 100644 --- a/tests/components/device_automation/test_init.py +++ b/tests/components/device_automation/test_init.py @@ -140,6 +140,13 @@ async def test_websocket_get_triggers(hass, hass_ws_client, device_reg, entity_r ) entity_reg.async_get_or_create("light", "test", "5678", device_id=device_entry.id) expected_triggers = [ + { + "platform": "device", + "domain": "light", + "type": "toggled", + "device_id": device_entry.id, + "entity_id": "light.test_5678", + }, { "platform": "device", "domain": "light", @@ -395,7 +402,7 @@ async def test_async_get_device_automations_single_device_trigger( hass, "trigger", [device_entry.id] ) assert device_entry.id in result - assert len(result[device_entry.id]) == 2 + assert len(result[device_entry.id]) == 3 # toggled, turned_on, turned_off async def test_async_get_device_automations_all_devices_trigger( @@ -412,7 +419,7 @@ async def test_async_get_device_automations_all_devices_trigger( entity_reg.async_get_or_create("light", "test", "5678", device_id=device_entry.id) result = await device_automation.async_get_device_automations(hass, "trigger") assert device_entry.id in result - assert len(result[device_entry.id]) == 2 + assert len(result[device_entry.id]) == 3 # toggled, turned_on, turned_off async def test_async_get_device_automations_all_devices_condition( @@ -505,7 +512,7 @@ async def test_websocket_get_trigger_capabilities( triggers = msg["result"] id = 2 - assert len(triggers) == 2 + assert len(triggers) == 3 # toggled, turned_on, turned_off for trigger in triggers: await client.send_json( {