From c306ebed49163564e55edffa0ed17c8c1fc19f34 Mon Sep 17 00:00:00 2001
From: Paul Bottein <paul.bottein@gmail.com>
Date: Wed, 2 Oct 2024 19:04:36 +0200
Subject: [PATCH] Fix device id support for alarm control panel template
(#127340)
---
.../template/alarm_control_panel.py | 6 ++-
.../template/test_alarm_control_panel.py | 43 +++++++++++++++++++
2 files changed, 48 insertions(+), 1 deletion(-)
diff --git a/homeassistant/components/template/alarm_control_panel.py b/homeassistant/components/template/alarm_control_panel.py
index 0d9e5ebc8ce..6c8a70b328e 100644
--- a/homeassistant/components/template/alarm_control_panel.py
+++ b/homeassistant/components/template/alarm_control_panel.py
@@ -38,6 +38,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import TemplateError
from homeassistant.helpers import selector
import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.device import async_device_info_to_link_from_device_id
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
@@ -233,7 +234,10 @@ class AlarmControlPanelTemplate(TemplateEntity, AlarmControlPanelEntity, Restore
self._trigger_script = Script(hass, trigger_action, name, DOMAIN)
self._state: str | None = None
-
+ self._attr_device_info = async_device_info_to_link_from_device_id(
+ hass,
+ config.get(CONF_DEVICE_ID),
+ )
supported_features = AlarmControlPanelEntityFeature(0)
if self._arm_night_script is not None:
supported_features = (
diff --git a/tests/components/template/test_alarm_control_panel.py b/tests/components/template/test_alarm_control_panel.py
index 263563fe752..666dfe744a2 100644
--- a/tests/components/template/test_alarm_control_panel.py
+++ b/tests/components/template/test_alarm_control_panel.py
@@ -23,6 +23,7 @@ from homeassistant.const import (
STATE_UNKNOWN,
)
from homeassistant.core import Event, HomeAssistant, State, callback
+from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.setup import async_setup_component
from tests.common import MockConfigEntry, assert_setup_component, mock_restore_cache
@@ -508,3 +509,45 @@ async def test_restore_state(
state = hass.states.get("alarm_control_panel.test_template_panel")
assert state.state == initial_state
+
+
+async def test_device_id(
+ hass: HomeAssistant,
+ device_registry: dr.DeviceRegistry,
+ entity_registry: er.EntityRegistry,
+) -> None:
+ """Test for device for button template."""
+
+ device_config_entry = MockConfigEntry()
+ device_config_entry.add_to_hass(hass)
+ device_entry = device_registry.async_get_or_create(
+ config_entry_id=device_config_entry.entry_id,
+ identifiers={("test", "identifier_test")},
+ connections={("mac", "30:31:32:33:34:35")},
+ )
+ await hass.async_block_till_done()
+ assert device_entry is not None
+ assert device_entry.id is not None
+
+ template_config_entry = MockConfigEntry(
+ data={},
+ domain=template.DOMAIN,
+ options={
+ "name": "My template",
+ "value_template": "disarmed",
+ "template_type": "alarm_control_panel",
+ "code_arm_required": True,
+ "code_format": "number",
+ "device_id": device_entry.id,
+ },
+ title="My template",
+ )
+
+ template_config_entry.add_to_hass(hass)
+
+ assert await hass.config_entries.async_setup(template_config_entry.entry_id)
+ await hass.async_block_till_done()
+
+ template_entity = entity_registry.async_get("alarm_control_panel.my_template")
+ assert template_entity is not None
+ assert template_entity.device_id == device_entry.id
--
GitLab