From 8ff12eacd41aa17b73a9aec5d9d345082999ea1d Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Sat, 1 Oct 2022 14:33:11 +0000 Subject: [PATCH] Do not use AQI device class for CAQI sensor in Airly integration (#79402) --- homeassistant/components/airly/sensor.py | 2 +- tests/components/airly/test_sensor.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/airly/sensor.py b/homeassistant/components/airly/sensor.py index bbb501ae47b..122990adecc 100644 --- a/homeassistant/components/airly/sensor.py +++ b/homeassistant/components/airly/sensor.py @@ -68,7 +68,7 @@ class AirlySensorEntityDescription(SensorEntityDescription): SENSOR_TYPES: tuple[AirlySensorEntityDescription, ...] = ( AirlySensorEntityDescription( key=ATTR_API_CAQI, - device_class=SensorDeviceClass.AQI, + icon="mdi:air-filter", name=ATTR_API_CAQI, native_unit_of_measurement="CAQI", ), diff --git a/tests/components/airly/test_sensor.py b/tests/components/airly/test_sensor.py index 9ac10f20fc3..c95d2d895fd 100644 --- a/tests/components/airly/test_sensor.py +++ b/tests/components/airly/test_sensor.py @@ -11,6 +11,7 @@ from homeassistant.const import ( ATTR_ATTRIBUTION, ATTR_DEVICE_CLASS, ATTR_ENTITY_ID, + ATTR_ICON, ATTR_UNIT_OF_MEASUREMENT, CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, PERCENTAGE, @@ -37,7 +38,7 @@ async def test_sensor(hass, aioclient_mock): assert state.state == "7" assert state.attributes.get(ATTR_ATTRIBUTION) == ATTRIBUTION assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "CAQI" - assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.AQI + assert state.attributes.get(ATTR_ICON) == "mdi:air-filter" entry = registry.async_get("sensor.home_caqi") assert entry