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

Restores unit_of_measurement (#54335)

Browse Source
This commit is contained in:
Diogo Gomes 2021-08-09 18:48:01 +01:00 committed by GitHub
parent 188919f079
commit b88f0adbe9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/integration/sensor.py
View File

@ -145,6 +145,10 @@ class IntegrationSensor(RestoreEntity, SensorEntity):
) )
self._attr_device_class = state.attributes.get(ATTR_DEVICE_CLASS) self._attr_device_class = state.attributes.get(ATTR_DEVICE_CLASS)
self._unit_of_measurement = state.attributes.get(
ATTR_UNIT_OF_MEASUREMENT
)
@callback @callback
def calc_integration(event): def calc_integration(event):
"""Handle the sensor state changes.""" """Handle the sensor state changes."""

1
tests/components/integration/test_sensor.py
View File

@ -73,6 +73,7 @@ async def test_restore_state(hass: HomeAssistant) -> None:
{ {
"last_reset": "2019-10-06T21:00:00", "last_reset": "2019-10-06T21:00:00",
"device_class": DEVICE_CLASS_ENERGY, "device_class": DEVICE_CLASS_ENERGY,
"unit_of_measurement": ENERGY_KILO_WATT_HOUR,
}, },
), ),
), ),