evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use DeviceClass Enums in integration tests (#62511)

Browse source
This commit is contained in:
Dave T 2021-12-21 18:03:39 +00:00 • committed by GitHub
parent 1f62371f45
commit 498720f3c9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/components/integration/test_sensor.py
View file

@ -2,14 +2,8 @@
from datetime import timedelta from datetime import timedelta
from unittest.mock import patch from unittest.mock import patch
from homeassistant.components.sensor import STATE_CLASS_TOTAL from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
from homeassistant.const import ( from homeassistant.const import ENERGY_KILO_WATT_HOUR, POWER_WATT, TIME_SECONDS
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
ENERGY_KILO_WATT_HOUR,
POWER_WATT,
TIME_SECONDS,
)
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -39,13 +33,13 @@ async def test_state(hass) -> None:
state = hass.states.get("sensor.integration") state = hass.states.get("sensor.integration")
assert state is not None assert state is not None
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL assert state.attributes.get("state_class") is SensorStateClass.TOTAL
assert "device_class" not in state.attributes assert "device_class" not in state.attributes
future_now = dt_util.utcnow() + timedelta(seconds=3600) future_now = dt_util.utcnow() + timedelta(seconds=3600)
with patch("homeassistant.util.dt.utcnow", return_value=future_now): with patch("homeassistant.util.dt.utcnow", return_value=future_now):
hass.states.async_set( hass.states.async_set(
entity_id, 1, {"device_class": DEVICE_CLASS_POWER}, force_update=True entity_id, 1, {"device_class": SensorDeviceClass.POWER}, force_update=True
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -56,8 +50,8 @@ async def test_state(hass) -> None:
assert round(float(state.state), config["sensor"]["round"]) == 1.0 assert round(float(state.state), config["sensor"]["round"]) == 1.0
assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
assert state.attributes.get("device_class") == DEVICE_CLASS_ENERGY assert state.attributes.get("device_class") == SensorDeviceClass.ENERGY
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL assert state.attributes.get("state_class") is SensorStateClass.TOTAL
async def test_restore_state(hass: HomeAssistant) -> None: async def test_restore_state(hass: HomeAssistant) -> None:
@ -69,7 +63,7 @@ async def test_restore_state(hass: HomeAssistant) -> None:
"sensor.integration", "sensor.integration",
"100.0", "100.0",
{ {
"device_class": DEVICE_CLASS_ENERGY, "device_class": SensorDeviceClass.ENERGY,
"unit_of_measurement": ENERGY_KILO_WATT_HOUR, "unit_of_measurement": ENERGY_KILO_WATT_HOUR,
}, },
), ),
@ -92,7 +86,7 @@ async def test_restore_state(hass: HomeAssistant) -> None:
assert state assert state
assert state.state == "100.00" assert state.state == "100.00"
assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR assert state.attributes.get("unit_of_measurement") == ENERGY_KILO_WATT_HOUR
assert state.attributes.get("device_class") == DEVICE_CLASS_ENERGY assert state.attributes.get("device_class") == SensorDeviceClass.ENERGY
async def test_restore_state_failed(hass: HomeAssistant) -> None: async def test_restore_state_failed(hass: HomeAssistant) -> None:
@ -125,7 +119,7 @@ async def test_restore_state_failed(hass: HomeAssistant) -> None:
assert state assert state
assert state.state == "unknown" assert state.state == "unknown"
assert state.attributes.get("unit_of_measurement") is None assert state.attributes.get("unit_of_measurement") is None
assert state.attributes.get("state_class") == STATE_CLASS_TOTAL assert state.attributes.get("state_class") is SensorStateClass.TOTAL
assert "device_class" not in state.attributes assert "device_class" not in state.attributes