jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 09:17:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add state_class to Tradfri battery and air quality sensors (#84185)

Browse Source
This commit is contained in:
Jelte Zeilstra 2022-12-18 16:52:08 +01:00 committed by GitHub
parent 10a6c56fec
commit f99d26231f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/tradfri/sensor.py
View File

@ -78,6 +78,7 @@ SENSOR_DESCRIPTIONS_BATTERY: tuple[TradfriSensorEntityDescription, ...] = (
TradfriSensorEntityDescription( TradfriSensorEntityDescription(
key="battery_level", key="battery_level",
device_class=SensorDeviceClass.BATTERY, device_class=SensorDeviceClass.BATTERY,
state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
value=lambda device: cast(int, device.device_info.battery_level), value=lambda device: cast(int, device.device_info.battery_level),
), ),
@ -89,6 +90,7 @@ SENSOR_DESCRIPTIONS_FAN: tuple[TradfriSensorEntityDescription, ...] = (
key="aqi", key="aqi",
name="air quality", name="air quality",
device_class=SensorDeviceClass.AQI, device_class=SensorDeviceClass.AQI,
state_class=SensorStateClass.MEASUREMENT,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
value=_get_air_quality, value=_get_air_quality,
), ),

2
tests/components/tradfri/test_sensor.py
View File

@ -71,6 +71,7 @@ async def test_cover_battery_sensor(hass, mock_gateway, mock_api_factory):
assert sensor_1.state == "42" assert sensor_1.state == "42"
assert sensor_1.attributes["unit_of_measurement"] == "%" assert sensor_1.attributes["unit_of_measurement"] == "%"
assert sensor_1.attributes["device_class"] == "battery" 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): 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.state == "42"
assert sensor_1.attributes["unit_of_measurement"] == "µg/m³" assert sensor_1.attributes["unit_of_measurement"] == "µg/m³"
assert sensor_1.attributes["device_class"] == "aqi" 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): async def test_filter_time_left_sensor(hass, mock_gateway, mock_api_factory):