243 lines
8.1 KiB
Python
243 lines
8.1 KiB
Python
"""Common tests for HomematicIP devices."""
|
|
from homematicip.base.enums import EventType
|
|
|
|
from homeassistant.components.homematicip_cloud import DOMAIN as HMIPC_DOMAIN
|
|
from homeassistant.components.homematicip_cloud.hap import HomematicipHAP
|
|
from homeassistant.const import STATE_ON, STATE_UNAVAILABLE
|
|
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
|
|
|
from .helper import (
|
|
HAPID,
|
|
HomeFactory,
|
|
async_manipulate_test_data,
|
|
get_and_check_entity_basics,
|
|
)
|
|
|
|
from tests.async_mock import patch
|
|
|
|
|
|
async def test_hmip_load_all_supported_devices(hass, default_mock_hap_factory):
|
|
"""Ensure that all supported devices could be loaded."""
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=None, test_groups=None
|
|
)
|
|
|
|
assert len(mock_hap.hmip_device_by_entity_id) == 192
|
|
|
|
|
|
async def test_hmip_remove_device(hass, default_mock_hap_factory):
|
|
"""Test Remove of hmip device."""
|
|
entity_id = "light.treppe_ch"
|
|
entity_name = "Treppe CH"
|
|
device_model = "HmIP-BSL"
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=["Treppe"]
|
|
)
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert ha_state.state == STATE_ON
|
|
assert hmip_device
|
|
|
|
device_registry = await dr.async_get_registry(hass)
|
|
entity_registry = await er.async_get_registry(hass)
|
|
|
|
pre_device_count = len(device_registry.devices)
|
|
pre_entity_count = len(entity_registry.entities)
|
|
pre_mapping_count = len(mock_hap.hmip_device_by_entity_id)
|
|
|
|
hmip_device.fire_remove_event()
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(device_registry.devices) == pre_device_count - 1
|
|
assert len(entity_registry.entities) == pre_entity_count - 3
|
|
assert len(mock_hap.hmip_device_by_entity_id) == pre_mapping_count - 3
|
|
|
|
|
|
async def test_hmip_add_device(hass, default_mock_hap_factory, hmip_config_entry):
|
|
"""Test Remove of hmip device."""
|
|
entity_id = "light.treppe_ch"
|
|
entity_name = "Treppe CH"
|
|
device_model = "HmIP-BSL"
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=["Treppe"]
|
|
)
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert ha_state.state == STATE_ON
|
|
assert hmip_device
|
|
|
|
device_registry = await dr.async_get_registry(hass)
|
|
entity_registry = await er.async_get_registry(hass)
|
|
|
|
pre_device_count = len(device_registry.devices)
|
|
pre_entity_count = len(entity_registry.entities)
|
|
pre_mapping_count = len(mock_hap.hmip_device_by_entity_id)
|
|
|
|
hmip_device.fire_remove_event()
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(device_registry.devices) == pre_device_count - 1
|
|
assert len(entity_registry.entities) == pre_entity_count - 3
|
|
assert len(mock_hap.hmip_device_by_entity_id) == pre_mapping_count - 3
|
|
|
|
reloaded_hap = HomematicipHAP(hass, hmip_config_entry)
|
|
with patch(
|
|
"homeassistant.components.homematicip_cloud.HomematicipHAP",
|
|
return_value=reloaded_hap,
|
|
), patch.object(reloaded_hap, "async_connect"), patch.object(
|
|
reloaded_hap, "get_hap", return_value=mock_hap.home
|
|
), patch(
|
|
"homeassistant.components.homematicip_cloud.hap.asyncio.sleep"
|
|
):
|
|
mock_hap.home.fire_create_event(event_type=EventType.DEVICE_ADDED)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(device_registry.devices) == pre_device_count
|
|
assert len(entity_registry.entities) == pre_entity_count
|
|
new_hap = hass.data[HMIPC_DOMAIN][HAPID]
|
|
assert len(new_hap.hmip_device_by_entity_id) == pre_mapping_count
|
|
|
|
|
|
async def test_hmip_remove_group(hass, default_mock_hap_factory):
|
|
"""Test Remove of hmip group."""
|
|
entity_id = "switch.strom_group"
|
|
entity_name = "Strom Group"
|
|
device_model = None
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(test_groups=["Strom"])
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert ha_state.state == STATE_ON
|
|
assert hmip_device
|
|
|
|
device_registry = await dr.async_get_registry(hass)
|
|
entity_registry = await er.async_get_registry(hass)
|
|
|
|
pre_device_count = len(device_registry.devices)
|
|
pre_entity_count = len(entity_registry.entities)
|
|
pre_mapping_count = len(mock_hap.hmip_device_by_entity_id)
|
|
|
|
hmip_device.fire_remove_event()
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(device_registry.devices) == pre_device_count
|
|
assert len(entity_registry.entities) == pre_entity_count - 1
|
|
assert len(mock_hap.hmip_device_by_entity_id) == pre_mapping_count - 1
|
|
|
|
|
|
async def test_all_devices_unavailable_when_hap_not_connected(
|
|
hass, default_mock_hap_factory
|
|
):
|
|
"""Test make all devices unavaulable when hap is not connected."""
|
|
entity_id = "light.treppe_ch"
|
|
entity_name = "Treppe CH"
|
|
device_model = "HmIP-BSL"
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=["Treppe"]
|
|
)
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert ha_state.state == STATE_ON
|
|
assert hmip_device
|
|
|
|
assert mock_hap.home.connected
|
|
|
|
await async_manipulate_test_data(hass, mock_hap.home, "connected", False)
|
|
|
|
ha_state = hass.states.get(entity_id)
|
|
assert ha_state.state == STATE_UNAVAILABLE
|
|
|
|
|
|
async def test_hap_reconnected(hass, default_mock_hap_factory):
|
|
"""Test reconnect hap."""
|
|
entity_id = "light.treppe_ch"
|
|
entity_name = "Treppe CH"
|
|
device_model = "HmIP-BSL"
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=["Treppe"]
|
|
)
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert ha_state.state == STATE_ON
|
|
assert hmip_device
|
|
|
|
assert mock_hap.home.connected
|
|
|
|
await async_manipulate_test_data(hass, mock_hap.home, "connected", False)
|
|
|
|
ha_state = hass.states.get(entity_id)
|
|
assert ha_state.state == STATE_UNAVAILABLE
|
|
|
|
mock_hap._accesspoint_connected = False # pylint: disable=protected-access
|
|
await async_manipulate_test_data(hass, mock_hap.home, "connected", True)
|
|
await hass.async_block_till_done()
|
|
ha_state = hass.states.get(entity_id)
|
|
assert ha_state.state == STATE_ON
|
|
|
|
|
|
async def test_hap_with_name(hass, mock_connection, hmip_config_entry):
|
|
"""Test hap with name."""
|
|
home_name = "TestName"
|
|
entity_id = f"light.{home_name.lower()}_treppe_ch"
|
|
entity_name = f"{home_name} Treppe CH"
|
|
device_model = "HmIP-BSL"
|
|
|
|
hmip_config_entry.data = {**hmip_config_entry.data, "name": home_name}
|
|
mock_hap = await HomeFactory(
|
|
hass, mock_connection, hmip_config_entry
|
|
).async_get_mock_hap(test_devices=["Treppe"])
|
|
assert mock_hap
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
|
|
assert hmip_device
|
|
assert ha_state.state == STATE_ON
|
|
assert ha_state.attributes["friendly_name"] == entity_name
|
|
|
|
|
|
async def test_hmip_reset_energy_counter_services(hass, default_mock_hap_factory):
|
|
"""Test reset_energy_counter service."""
|
|
entity_id = "switch.pc"
|
|
entity_name = "Pc"
|
|
device_model = "HMIP-PSM"
|
|
mock_hap = await default_mock_hap_factory.async_get_mock_hap(
|
|
test_devices=[entity_name]
|
|
)
|
|
|
|
ha_state, hmip_device = get_and_check_entity_basics(
|
|
hass, mock_hap, entity_id, entity_name, device_model
|
|
)
|
|
assert ha_state
|
|
|
|
await hass.services.async_call(
|
|
"homematicip_cloud",
|
|
"reset_energy_counter",
|
|
{"entity_id": "switch.pc"},
|
|
blocking=True,
|
|
)
|
|
assert hmip_device.mock_calls[-1][0] == "reset_energy_counter"
|
|
assert len(hmip_device._connection.mock_calls) == 2 # pylint: disable=W0212
|
|
|
|
await hass.services.async_call(
|
|
"homematicip_cloud", "reset_energy_counter", {"entity_id": "all"}, blocking=True
|
|
)
|
|
assert hmip_device.mock_calls[-1][0] == "reset_energy_counter"
|
|
assert len(hmip_device._connection.mock_calls) == 4 # pylint: disable=W0212
|