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

Set state class to total for Integration sensors (#55872)

Browse Source
This commit is contained in:
Erik Montnemery 2021-09-07 00:27:31 +02:00 committed by GitHub
parent 6895081595
commit b1dbdec2ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,7 +8,7 @@ from homeassistant.components.sensor import (
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
PLATFORM_SCHEMA,
STATE_CLASS_TOTAL_INCREASING,
STATE_CLASS_TOTAL,
SensorEntity,
)
from homeassistant.const import (
@ -116,7 +116,7 @@ class IntegrationSensor(RestoreEntity, SensorEntity):
self._unit_of_measurement = unit_of_measurement
self._unit_prefix = UNIT_PREFIXES[unit_prefix]
self._unit_time = UNIT_TIME[unit_time]
self._attr_state_class = STATE_CLASS_TOTAL_INCREASING
self._attr_state_class = STATE_CLASS_TOTAL
async def async_added_to_hass(self):
"""Handle entity which will be added."""

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

@ -2,7 +2,7 @@
from datetime import timedelta
from unittest.mock import patch
from homeassistant.components.sensor import STATE_CLASS_TOTAL_INCREASING
from homeassistant.components.sensor import STATE_CLASS_TOTAL
from homeassistant.const import (
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
@ -39,7 +39,7 @@ async def test_state(hass) -> None:
state = hass.states.get("sensor.integration")
assert state is not None
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
assert "device_class" not in state.attributes
future_now = dt_util.utcnow() + timedelta(seconds=3600)
@ -57,7 +57,7 @@ async def test_state(hass) -> None:
assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
assert state.attributes.get("device_class") == DEVICE_CLASS_ENERGY
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
async def test_restore_state(hass: HomeAssistant) -> None:
@ -104,7 +104,9 @@ async def test_restore_state_failed(hass: HomeAssistant) -> None:
State(
"sensor.integration",
"INVALID",
{},
{
"last_reset": "2019-10-06T21:00:00.000000",
},
),
),
)
@ -125,7 +127,7 @@ async def test_restore_state_failed(hass: HomeAssistant) -> None:
assert state
assert state.state == "0"
assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL_INCREASING
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL
assert "device_class" not in state.attributes