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

Use ENERGY_KILO_WATT_HOUR constant (#33962)

Browse source
This commit is contained in:
springstan 2020-04-10 20:21:29 +02:00 • committed by GitHub
parent c3c4752fa5
commit 02c9e47db8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
13 changed files with 132 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/components/utility_meter/test_init.py
View file

@ -11,7 +11,11 @@ from homeassistant.components.utility_meter.const import (
SERVICE_SELECT_NEXT_TARIFF,
SERVICE_SELECT_TARIFF,
)
from homeassistant.const import ATTR_ENTITY_ID, EVENT_HOMEASSISTANT_START
from homeassistant.const import (
ATTR_ENTITY_ID,
ENERGY_KILO_WATT_HOUR,
EVENT_HOMEASSISTANT_START,
)
from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util
@ -36,13 +40,16 @@ async def test_services(hass):
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
entity_id = config[DOMAIN]["energy_bill"]["source"]
hass.states.async_set(entity_id, 1, {"unit_of_measurement": "kWh"})
hass.states.async_set(entity_id, 1, {"unit_of_measurement": ENERGY_KILO_WATT_HOUR})
await hass.async_block_till_done()
now = dt_util.utcnow() + timedelta(seconds=10)
with patch("homeassistant.util.dt.utcnow", return_value=now):
hass.states.async_set(
entity_id, 3, {"unit_of_measurement": "kWh"}, force_update=True
entity_id,
3,
{"unit_of_measurement": ENERGY_KILO_WATT_HOUR},
force_update=True,
)
await hass.async_block_till_done()
@ -60,7 +67,10 @@ async def test_services(hass):
now += timedelta(seconds=10)
with patch("homeassistant.util.dt.utcnow", return_value=now):
hass.states.async_set(
entity_id, 4, {"unit_of_measurement": "kWh"}, force_update=True
entity_id,
4,
{"unit_of_measurement": ENERGY_KILO_WATT_HOUR},
force_update=True,
)
await hass.async_block_till_done()
@ -78,7 +88,10 @@ async def test_services(hass):
now += timedelta(seconds=10)
with patch("homeassistant.util.dt.utcnow", return_value=now):
hass.states.async_set(
entity_id, 5, {"unit_of_measurement": "kWh"}, force_update=True
entity_id,
5,
{"unit_of_measurement": ENERGY_KILO_WATT_HOUR},
force_update=True,
)
await hass.async_block_till_done()