Set state class measurement also for Total Energy for AVM Fritz!Smarthome devices (#53707)

This commit is contained in:
Michael
2021-07-29 20:54:51 +02:00
committed by Paulus Schoutsen
parent c6f588fc08
commit b2187022c4
2 changed files with 2 additions and 2 deletions

View File

@@ -96,7 +96,7 @@ async def async_setup_entry(
ATTR_ENTITY_ID: f"{device.ain}_total_energy",
ATTR_UNIT_OF_MEASUREMENT: ENERGY_KILO_WATT_HOUR,
ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY,
ATTR_STATE_CLASS: None,
ATTR_STATE_CLASS: STATE_CLASS_MEASUREMENT,
},
coordinator,
ain,

View File

@@ -76,7 +76,7 @@ async def test_setup(hass: HomeAssistant, fritz: Mock):
assert state.attributes[ATTR_LAST_RESET] == "1970-01-01T00:00:00+00:00"
assert state.attributes[ATTR_FRIENDLY_NAME] == f"{CONF_FAKE_NAME} Total Energy"
assert state.attributes[ATTR_UNIT_OF_MEASUREMENT] == ENERGY_KILO_WATT_HOUR
assert ATTR_STATE_CLASS not in state.attributes
assert state.attributes[ATTR_STATE_CLASS] == STATE_CLASS_MEASUREMENT
async def test_turn_on(hass: HomeAssistant, fritz: Mock):