214 lines
6.7 KiB
Python
214 lines
6.7 KiB
Python
"""Tests for AVM Fritz!Box switch component."""
|
|
|
|
from datetime import timedelta
|
|
from unittest.mock import Mock
|
|
|
|
from requests.exceptions import HTTPError
|
|
|
|
from homeassistant.components.fritzbox.const import DOMAIN as FB_DOMAIN
|
|
from homeassistant.components.sensor import (
|
|
ATTR_STATE_CLASS,
|
|
DOMAIN as SENSOR_DOMAIN,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.components.switch import DOMAIN
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
ATTR_FRIENDLY_NAME,
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
CONF_DEVICES,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_ON,
|
|
STATE_UNAVAILABLE,
|
|
EntityCategory,
|
|
UnitOfElectricCurrent,
|
|
UnitOfElectricPotential,
|
|
UnitOfEnergy,
|
|
UnitOfPower,
|
|
UnitOfTemperature,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from . import FritzDeviceSwitchMock, set_devices, setup_config_entry
|
|
from .const import CONF_FAKE_NAME, MOCK_CONFIG
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
ENTITY_ID = f"{DOMAIN}.{CONF_FAKE_NAME}"
|
|
|
|
|
|
async def test_setup(
|
|
hass: HomeAssistant, entity_registry: er.EntityRegistry, fritz: Mock
|
|
) -> None:
|
|
"""Test setup of platform."""
|
|
device = FritzDeviceSwitchMock()
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ID)
|
|
assert state
|
|
assert state.state == STATE_ON
|
|
assert state.attributes[ATTR_FRIENDLY_NAME] == CONF_FAKE_NAME
|
|
assert ATTR_STATE_CLASS not in state.attributes
|
|
|
|
state = hass.states.get(f"{ENTITY_ID}_humidity")
|
|
assert state is None
|
|
|
|
sensors = (
|
|
[
|
|
f"{SENSOR_DOMAIN}.{CONF_FAKE_NAME}_temperature",
|
|
"1.23",
|
|
f"{CONF_FAKE_NAME} Temperature",
|
|
UnitOfTemperature.CELSIUS,
|
|
SensorStateClass.MEASUREMENT,
|
|
EntityCategory.DIAGNOSTIC,
|
|
],
|
|
[
|
|
f"{SENSOR_DOMAIN}.{CONF_FAKE_NAME}_power",
|
|
"5.678",
|
|
f"{CONF_FAKE_NAME} Power",
|
|
UnitOfPower.WATT,
|
|
SensorStateClass.MEASUREMENT,
|
|
None,
|
|
],
|
|
[
|
|
f"{SENSOR_DOMAIN}.{CONF_FAKE_NAME}_energy",
|
|
"1.234",
|
|
f"{CONF_FAKE_NAME} Energy",
|
|
UnitOfEnergy.KILO_WATT_HOUR,
|
|
SensorStateClass.TOTAL_INCREASING,
|
|
None,
|
|
],
|
|
[
|
|
f"{SENSOR_DOMAIN}.{CONF_FAKE_NAME}_voltage",
|
|
"230.0",
|
|
f"{CONF_FAKE_NAME} Voltage",
|
|
UnitOfElectricPotential.VOLT,
|
|
SensorStateClass.MEASUREMENT,
|
|
None,
|
|
],
|
|
[
|
|
f"{SENSOR_DOMAIN}.{CONF_FAKE_NAME}_current",
|
|
"0.025",
|
|
f"{CONF_FAKE_NAME} Current",
|
|
UnitOfElectricCurrent.AMPERE,
|
|
SensorStateClass.MEASUREMENT,
|
|
None,
|
|
],
|
|
)
|
|
|
|
for sensor in sensors:
|
|
state = hass.states.get(sensor[0])
|
|
assert state
|
|
assert state.state == sensor[1]
|
|
assert state.attributes[ATTR_FRIENDLY_NAME] == sensor[2]
|
|
assert state.attributes[ATTR_UNIT_OF_MEASUREMENT] == sensor[3]
|
|
assert state.attributes[ATTR_STATE_CLASS] == sensor[4]
|
|
assert state.attributes[ATTR_STATE_CLASS] == sensor[4]
|
|
entry = entity_registry.async_get(sensor[0])
|
|
assert entry
|
|
assert entry.entity_category is sensor[5]
|
|
|
|
|
|
async def test_turn_on(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test turn device on."""
|
|
device = FritzDeviceSwitchMock()
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: ENTITY_ID}, True
|
|
)
|
|
assert device.set_switch_state_on.call_count == 1
|
|
|
|
|
|
async def test_turn_off(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test turn device off."""
|
|
device = FritzDeviceSwitchMock()
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: ENTITY_ID}, True
|
|
)
|
|
assert device.set_switch_state_off.call_count == 1
|
|
|
|
|
|
async def test_update(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test update without error."""
|
|
device = FritzDeviceSwitchMock()
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
assert fritz().update_devices.call_count == 1
|
|
assert fritz().login.call_count == 1
|
|
|
|
next_update = dt_util.utcnow() + timedelta(seconds=200)
|
|
async_fire_time_changed(hass, next_update)
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
assert fritz().update_devices.call_count == 2
|
|
assert fritz().login.call_count == 1
|
|
|
|
|
|
async def test_update_error(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test update with error."""
|
|
device = FritzDeviceSwitchMock()
|
|
fritz().update_devices.side_effect = HTTPError("Boom")
|
|
assert not await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
assert fritz().update_devices.call_count == 2
|
|
assert fritz().login.call_count == 2
|
|
|
|
next_update = dt_util.utcnow() + timedelta(seconds=200)
|
|
async_fire_time_changed(hass, next_update)
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
assert fritz().update_devices.call_count == 4
|
|
assert fritz().login.call_count == 4
|
|
|
|
|
|
async def test_assume_device_unavailable(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test assume device as unavailable."""
|
|
device = FritzDeviceSwitchMock()
|
|
device.voltage = 0
|
|
device.energy = 0
|
|
device.power = 0
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ID)
|
|
assert state
|
|
assert state.state == STATE_UNAVAILABLE
|
|
|
|
|
|
async def test_discover_new_device(hass: HomeAssistant, fritz: Mock) -> None:
|
|
"""Test adding new discovered devices during runtime."""
|
|
device = FritzDeviceSwitchMock()
|
|
assert await setup_config_entry(
|
|
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
|
|
)
|
|
|
|
state = hass.states.get(ENTITY_ID)
|
|
assert state
|
|
|
|
new_device = FritzDeviceSwitchMock()
|
|
new_device.ain = "7890 1234"
|
|
new_device.name = "new_switch"
|
|
set_devices(fritz, devices=[device, new_device])
|
|
|
|
next_update = dt_util.utcnow() + timedelta(seconds=200)
|
|
async_fire_time_changed(hass, next_update)
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
|
|
state = hass.states.get(f"{DOMAIN}.new_switch")
|
|
assert state
|