From bf6a412be096bdbd602845dbc17ee49c3802c296 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Fri, 11 Jun 2021 23:22:18 +0200 Subject: [PATCH] Tweak device action scaffold, fix typo (#51751) --- .../integration/device_trigger.py | 26 ++++++------------- .../components/climate/test_device_action.py | 2 +- 2 files changed, 9 insertions(+), 19 deletions(-) diff --git a/script/scaffold/templates/device_trigger/integration/device_trigger.py b/script/scaffold/templates/device_trigger/integration/device_trigger.py index 5a060bbfaec..e070bc43f57 100644 --- a/script/scaffold/templates/device_trigger/integration/device_trigger.py +++ b/script/scaffold/templates/device_trigger/integration/device_trigger.py @@ -51,24 +51,14 @@ async def async_get_triggers(hass: HomeAssistant, device_id: str) -> list[dict]: # Add triggers for each entity that belongs to this integration # TODO add your own triggers. - triggers.append( - { - CONF_PLATFORM: "device", - CONF_DEVICE_ID: device_id, - CONF_DOMAIN: DOMAIN, - CONF_ENTITY_ID: entry.entity_id, - CONF_TYPE: "turned_on", - } - ) - triggers.append( - { - CONF_PLATFORM: "device", - CONF_DEVICE_ID: device_id, - CONF_DOMAIN: DOMAIN, - CONF_ENTITY_ID: entry.entity_id, - CONF_TYPE: "turned_off", - } - ) + base_trigger = { + CONF_PLATFORM: "device", + CONF_DEVICE_ID: device_id, + CONF_DOMAIN: DOMAIN, + CONF_ENTITY_ID: entry.entity_id, + } + triggers.append({**base_trigger, CONF_TYPE: "turned_on"}) + triggers.append({**base_trigger, CONF_TYPE: "turned_off"}) return triggers diff --git a/tests/components/climate/test_device_action.py b/tests/components/climate/test_device_action.py index fd5c5548916..3f9b1148443 100644 --- a/tests/components/climate/test_device_action.py +++ b/tests/components/climate/test_device_action.py @@ -259,7 +259,7 @@ async def test_capabilities( "action,capability_name", [("set_hvac_mode", "hvac_mode"), ("set_preset_mode", "preset_mode")], ) -async def test_capabilities_mising_entity( +async def test_capabilities_missing_entity( hass, device_reg, entity_reg, action, capability_name ): """Test getting capabilities."""