diff --git a/homeassistant/components/tradfri/sensor.py b/homeassistant/components/tradfri/sensor.py index acda0bec06d..6295cf09ab2 100644 --- a/homeassistant/components/tradfri/sensor.py +++ b/homeassistant/components/tradfri/sensor.py @@ -78,6 +78,7 @@ SENSOR_DESCRIPTIONS_BATTERY: tuple[TradfriSensorEntityDescription, ...] = ( TradfriSensorEntityDescription( key="battery_level", device_class=SensorDeviceClass.BATTERY, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=PERCENTAGE, value=lambda device: cast(int, device.device_info.battery_level), ), @@ -89,6 +90,7 @@ SENSOR_DESCRIPTIONS_FAN: tuple[TradfriSensorEntityDescription, ...] = ( key="aqi", name="air quality", device_class=SensorDeviceClass.AQI, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, value=_get_air_quality, ), diff --git a/tests/components/tradfri/test_sensor.py b/tests/components/tradfri/test_sensor.py index 1cd7a3672bc..6408613f4e3 100644 --- a/tests/components/tradfri/test_sensor.py +++ b/tests/components/tradfri/test_sensor.py @@ -71,6 +71,7 @@ async def test_cover_battery_sensor(hass, mock_gateway, mock_api_factory): assert sensor_1.state == "42" assert sensor_1.attributes["unit_of_measurement"] == "%" assert sensor_1.attributes["device_class"] == "battery" + assert sensor_1.attributes["state_class"] == "measurement" async def test_air_quality_sensor(hass, mock_gateway, mock_api_factory): @@ -91,6 +92,7 @@ async def test_air_quality_sensor(hass, mock_gateway, mock_api_factory): assert sensor_1.state == "42" assert sensor_1.attributes["unit_of_measurement"] == "µg/m³" assert sensor_1.attributes["device_class"] == "aqi" + assert sensor_1.attributes["state_class"] == "measurement" async def test_filter_time_left_sensor(hass, mock_gateway, mock_api_factory):