From edefa9f4f4647ee1ab63094b0fb065266f2d849c Mon Sep 17 00:00:00 2001 From: Robert Svensson Date: Wed, 20 Oct 2021 09:11:17 +0200 Subject: [PATCH] Assign entity category diagnostics to deCONZ battery sensors (#58077) --- homeassistant/components/deconz/sensor.py | 2 ++ tests/components/deconz/test_sensor.py | 7 +++++++ 2 files changed, 9 insertions(+) diff --git a/homeassistant/components/deconz/sensor.py b/homeassistant/components/deconz/sensor.py index aed4d2df7ee..f9d177523f9 100644 --- a/homeassistant/components/deconz/sensor.py +++ b/homeassistant/components/deconz/sensor.py @@ -33,6 +33,7 @@ from homeassistant.const import ( DEVICE_CLASS_PRESSURE, DEVICE_CLASS_TEMPERATURE, ENERGY_KILO_WATT_HOUR, + ENTITY_CATEGORY_DIAGNOSTIC, LIGHT_LUX, PERCENTAGE, POWER_WATT, @@ -73,6 +74,7 @@ ENTITY_DESCRIPTIONS = { device_class=DEVICE_CLASS_BATTERY, state_class=STATE_CLASS_MEASUREMENT, native_unit_of_measurement=PERCENTAGE, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, ), Consumption: SensorEntityDescription( key="consumption", diff --git a/tests/components/deconz/test_sensor.py b/tests/components/deconz/test_sensor.py index 624a1bec7ff..5f9762c339b 100644 --- a/tests/components/deconz/test_sensor.py +++ b/tests/components/deconz/test_sensor.py @@ -14,6 +14,7 @@ from homeassistant.const import ( DEVICE_CLASS_ILLUMINANCE, DEVICE_CLASS_POWER, DEVICE_CLASS_TEMPERATURE, + ENTITY_CATEGORY_DIAGNOSTIC, STATE_UNAVAILABLE, ) from homeassistant.helpers import entity_registry as er @@ -92,6 +93,8 @@ async def test_sensors(hass, aioclient_mock, mock_deconz_websocket): assert len(hass.states.async_all()) == 6 + ent_reg = er.async_get(hass) + light_level_sensor = hass.states.get("sensor.light_level_sensor") assert light_level_sensor.state == "999.8" assert light_level_sensor.attributes[ATTR_DEVICE_CLASS] == DEVICE_CLASS_ILLUMINANCE @@ -109,6 +112,10 @@ async def test_sensors(hass, aioclient_mock, mock_deconz_websocket): switch_2_battery_level = hass.states.get("sensor.switch_2_battery_level") assert switch_2_battery_level.state == "100" assert switch_2_battery_level.attributes[ATTR_DEVICE_CLASS] == DEVICE_CLASS_BATTERY + assert ( + ent_reg.async_get("sensor.switch_2_battery_level").entity_category + == ENTITY_CATEGORY_DIAGNOSTIC + ) assert not hass.states.get("sensor.daylight_sensor")