From 43fe30f6eece36d6921836fff28b9619ff507527 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 27 Jun 2023 17:42:13 +0200 Subject: [PATCH] Use identify device class in ZHA identify button entity (#95373) --- homeassistant/components/zha/button.py | 2 +- tests/components/zha/test_button.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/zha/button.py b/homeassistant/components/zha/button.py index 6564f3bc39a..010c0f63e27 100644 --- a/homeassistant/components/zha/button.py +++ b/homeassistant/components/zha/button.py @@ -105,7 +105,7 @@ class ZHAIdentifyButton(ZHAButton): return None return cls(unique_id, zha_device, cluster_handlers, **kwargs) - _attr_device_class: ButtonDeviceClass = ButtonDeviceClass.UPDATE + _attr_device_class = ButtonDeviceClass.IDENTIFY _attr_entity_category = EntityCategory.DIAGNOSTIC _attr_name = "Identify" _command_name = "identify" diff --git a/tests/components/zha/test_button.py b/tests/components/zha/test_button.py index f3ee3d6dca7..2a2fbc92ace 100644 --- a/tests/components/zha/test_button.py +++ b/tests/components/zha/test_button.py @@ -143,7 +143,7 @@ async def test_button(hass: HomeAssistant, contact_sensor) -> None: state = hass.states.get(entity_id) assert state assert state.state == STATE_UNKNOWN - assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.UPDATE + assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.IDENTIFY entry = entity_registry.async_get(entity_id) assert entry @@ -168,7 +168,7 @@ async def test_button(hass: HomeAssistant, contact_sensor) -> None: state = hass.states.get(entity_id) assert state assert state.state == "2021-11-04T16:37:00+00:00" - assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.UPDATE + assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.IDENTIFY async def test_frost_unlock(hass: HomeAssistant, tuya_water_valve) -> None: