142 lines
7.1 KiB
Python
142 lines
7.1 KiB
Python
"""The tests for Efergy sensor platform."""
|
|
from datetime import timedelta
|
|
|
|
from homeassistant.components.efergy.sensor import SENSOR_TYPES
|
|
from homeassistant.components.sensor import (
|
|
ATTR_STATE_CLASS,
|
|
DOMAIN as SENSOR_DOMAIN,
|
|
SensorDeviceClass,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_DEVICE_CLASS,
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
ENERGY_KILO_WATT_HOUR,
|
|
POWER_WATT,
|
|
STATE_UNAVAILABLE,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
from homeassistant.helpers.entity_registry import EntityRegistry
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from . import MULTI_SENSOR_TOKEN, mock_responses, setup_platform
|
|
|
|
from tests.common import async_fire_time_changed
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
|
|
async def test_sensor_readings(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
|
):
|
|
"""Test for successfully setting up the Efergy platform."""
|
|
for description in SENSOR_TYPES:
|
|
description.entity_registry_enabled_default = True
|
|
entry = await setup_platform(hass, aioclient_mock, SENSOR_DOMAIN)
|
|
ent_reg: EntityRegistry = er.async_get(hass)
|
|
|
|
state = hass.states.get("sensor.power_usage")
|
|
assert state.state == "1580"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get("sensor.energy_budget")
|
|
assert state.state == "ok"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) is None
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get("sensor.daily_consumption")
|
|
assert state.state == "38.21"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.weekly_consumption")
|
|
assert state.state == "267.47"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.monthly_consumption")
|
|
assert state.state == "1069.88"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.yearly_consumption")
|
|
assert state.state == "13373.50"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_KILO_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.daily_energy_cost")
|
|
assert state.state == "5.27"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.MONETARY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "EUR"
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.weekly_energy_cost")
|
|
assert state.state == "36.89"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.MONETARY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "EUR"
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.monthly_energy_cost")
|
|
assert state.state == "147.56"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.MONETARY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "EUR"
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get("sensor.yearly_energy_cost")
|
|
assert state.state == "1844.50"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.MONETARY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "EUR"
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
entity = ent_reg.async_get("sensor.power_usage_728386")
|
|
assert entity.disabled_by is er.RegistryEntryDisabler.INTEGRATION
|
|
ent_reg.async_update_entity(entity.entity_id, **{"disabled_by": None})
|
|
await hass.config_entries.async_reload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("sensor.power_usage_728386")
|
|
assert state.state == "1628"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
|
|
|
|
async def test_multi_sensor_readings(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
|
):
|
|
"""Test for multiple sensors in one household."""
|
|
for description in SENSOR_TYPES:
|
|
description.entity_registry_enabled_default = True
|
|
await setup_platform(hass, aioclient_mock, SENSOR_DOMAIN, MULTI_SENSOR_TOKEN)
|
|
state = hass.states.get("sensor.power_usage_728386")
|
|
assert state.state == "218"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get("sensor.power_usage_0")
|
|
assert state.state == "1808"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get("sensor.power_usage_728387")
|
|
assert state.state == "312"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
|
|
|
|
async def test_failed_update_and_reconnection(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
|
):
|
|
"""Test failed update and reconnection."""
|
|
await setup_platform(hass, aioclient_mock, SENSOR_DOMAIN)
|
|
assert hass.states.get("sensor.power_usage").state == "1580"
|
|
aioclient_mock.clear_requests()
|
|
await mock_responses(hass, aioclient_mock, error=True)
|
|
next_update = dt_util.utcnow() + timedelta(seconds=30)
|
|
async_fire_time_changed(hass, next_update)
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get("sensor.power_usage").state == STATE_UNAVAILABLE
|
|
aioclient_mock.clear_requests()
|
|
await mock_responses(hass, aioclient_mock)
|
|
next_update = dt_util.utcnow() + timedelta(seconds=30)
|
|
async_fire_time_changed(hass, next_update)
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get("sensor.power_usage").state == "1580"
|