236 lines
8.0 KiB
Python
236 lines
8.0 KiB
Python
"""Test ZHA Gateway."""
|
|
import asyncio
|
|
import time
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import zigpy.profiles.zha as zha
|
|
import zigpy.zcl.clusters.general as general
|
|
import zigpy.zcl.clusters.lighting as lighting
|
|
|
|
from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN
|
|
from homeassistant.components.zha.core.group import GroupMember
|
|
from homeassistant.components.zha.core.store import TOMBSTONE_LIFETIME
|
|
|
|
from .common import async_enable_traffic, async_find_group_entity_id, get_zha_gateway
|
|
|
|
IEEE_GROUPABLE_DEVICE = "01:2d:6f:00:0a:90:69:e8"
|
|
IEEE_GROUPABLE_DEVICE2 = "02:2d:6f:00:0a:90:69:e8"
|
|
|
|
|
|
@pytest.fixture
|
|
def zigpy_dev_basic(zigpy_device_mock):
|
|
"""Zigpy device with just a basic cluster."""
|
|
return zigpy_device_mock(
|
|
{
|
|
1: {
|
|
"in_clusters": [general.Basic.cluster_id],
|
|
"out_clusters": [],
|
|
"device_type": zha.DeviceType.ON_OFF_SWITCH,
|
|
}
|
|
}
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
async def zha_dev_basic(hass, zha_device_restored, zigpy_dev_basic):
|
|
"""ZHA device with just a basic cluster."""
|
|
|
|
zha_device = await zha_device_restored(zigpy_dev_basic)
|
|
return zha_device
|
|
|
|
|
|
@pytest.fixture
|
|
async def coordinator(hass, zigpy_device_mock, zha_device_joined):
|
|
"""Test zha light platform."""
|
|
|
|
zigpy_device = zigpy_device_mock(
|
|
{
|
|
1: {
|
|
"in_clusters": [],
|
|
"out_clusters": [],
|
|
"device_type": zha.DeviceType.COLOR_DIMMABLE_LIGHT,
|
|
}
|
|
},
|
|
ieee="00:15:8d:00:02:32:4f:32",
|
|
nwk=0x0000,
|
|
node_descriptor=b"\xf8\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff",
|
|
)
|
|
zha_device = await zha_device_joined(zigpy_device)
|
|
zha_device.available = True
|
|
return zha_device
|
|
|
|
|
|
@pytest.fixture
|
|
async def device_light_1(hass, zigpy_device_mock, zha_device_joined):
|
|
"""Test zha light platform."""
|
|
|
|
zigpy_device = zigpy_device_mock(
|
|
{
|
|
1: {
|
|
"in_clusters": [
|
|
general.OnOff.cluster_id,
|
|
general.LevelControl.cluster_id,
|
|
lighting.Color.cluster_id,
|
|
general.Groups.cluster_id,
|
|
],
|
|
"out_clusters": [],
|
|
"device_type": zha.DeviceType.COLOR_DIMMABLE_LIGHT,
|
|
}
|
|
},
|
|
ieee=IEEE_GROUPABLE_DEVICE,
|
|
)
|
|
zha_device = await zha_device_joined(zigpy_device)
|
|
zha_device.available = True
|
|
return zha_device
|
|
|
|
|
|
@pytest.fixture
|
|
async def device_light_2(hass, zigpy_device_mock, zha_device_joined):
|
|
"""Test zha light platform."""
|
|
|
|
zigpy_device = zigpy_device_mock(
|
|
{
|
|
1: {
|
|
"in_clusters": [
|
|
general.OnOff.cluster_id,
|
|
general.LevelControl.cluster_id,
|
|
lighting.Color.cluster_id,
|
|
general.Groups.cluster_id,
|
|
],
|
|
"out_clusters": [],
|
|
"device_type": zha.DeviceType.COLOR_DIMMABLE_LIGHT,
|
|
}
|
|
},
|
|
ieee=IEEE_GROUPABLE_DEVICE2,
|
|
)
|
|
zha_device = await zha_device_joined(zigpy_device)
|
|
zha_device.available = True
|
|
return zha_device
|
|
|
|
|
|
async def test_device_left(hass, zigpy_dev_basic, zha_dev_basic):
|
|
"""Device leaving the network should become unavailable."""
|
|
|
|
assert zha_dev_basic.available is True
|
|
|
|
get_zha_gateway(hass).device_left(zigpy_dev_basic)
|
|
await hass.async_block_till_done()
|
|
assert zha_dev_basic.available is False
|
|
|
|
|
|
async def test_gateway_group_methods(hass, device_light_1, device_light_2, coordinator):
|
|
"""Test creating a group with 2 members."""
|
|
zha_gateway = get_zha_gateway(hass)
|
|
assert zha_gateway is not None
|
|
zha_gateway.coordinator_zha_device = coordinator
|
|
coordinator._zha_gateway = zha_gateway
|
|
device_light_1._zha_gateway = zha_gateway
|
|
device_light_2._zha_gateway = zha_gateway
|
|
member_ieee_addresses = [device_light_1.ieee, device_light_2.ieee]
|
|
members = [GroupMember(device_light_1.ieee, 1), GroupMember(device_light_2.ieee, 1)]
|
|
|
|
# test creating a group with 2 members
|
|
zha_group = await zha_gateway.async_create_zigpy_group("Test Group", members)
|
|
await hass.async_block_till_done()
|
|
|
|
assert zha_group is not None
|
|
assert len(zha_group.members) == 2
|
|
for member in zha_group.members:
|
|
assert member.device.ieee in member_ieee_addresses
|
|
|
|
entity_id = async_find_group_entity_id(hass, LIGHT_DOMAIN, zha_group)
|
|
assert hass.states.get(entity_id) is not None
|
|
|
|
# test get group by name
|
|
assert zha_group == zha_gateway.async_get_group_by_name(zha_group.name)
|
|
|
|
# test removing a group
|
|
await zha_gateway.async_remove_zigpy_group(zha_group.group_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# we shouldn't have the group anymore
|
|
assert zha_gateway.async_get_group_by_name(zha_group.name) is None
|
|
|
|
# the group entity should be cleaned up
|
|
assert entity_id not in hass.states.async_entity_ids(LIGHT_DOMAIN)
|
|
|
|
# test creating a group with 1 member
|
|
zha_group = await zha_gateway.async_create_zigpy_group(
|
|
"Test Group", [GroupMember(device_light_1.ieee, 1)]
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert zha_group is not None
|
|
assert len(zha_group.members) == 1
|
|
for member in zha_group.members:
|
|
assert member.device.ieee in [device_light_1.ieee]
|
|
|
|
# the group entity should not have been cleaned up
|
|
assert entity_id not in hass.states.async_entity_ids(LIGHT_DOMAIN)
|
|
|
|
with patch("zigpy.zcl.Cluster.request", side_effect=asyncio.TimeoutError):
|
|
await zha_group.members[0].async_remove_from_group()
|
|
assert len(zha_group.members) == 1
|
|
for member in zha_group.members:
|
|
assert member.device.ieee in [device_light_1.ieee]
|
|
|
|
|
|
async def test_updating_device_store(hass, zigpy_dev_basic, zha_dev_basic):
|
|
"""Test saving data after a delay."""
|
|
zha_gateway = get_zha_gateway(hass)
|
|
assert zha_gateway is not None
|
|
await async_enable_traffic(hass, [zha_dev_basic])
|
|
|
|
assert zha_dev_basic.last_seen is not None
|
|
entry = zha_gateway.zha_storage.async_get_or_create_device(zha_dev_basic)
|
|
assert entry.last_seen == zha_dev_basic.last_seen
|
|
|
|
assert zha_dev_basic.last_seen is not None
|
|
last_seen = zha_dev_basic.last_seen
|
|
|
|
# test that we can't set None as last seen any more
|
|
zha_dev_basic.async_update_last_seen(None)
|
|
assert last_seen == zha_dev_basic.last_seen
|
|
|
|
# test that we won't put None in storage
|
|
zigpy_dev_basic.last_seen = None
|
|
assert zha_dev_basic.last_seen is None
|
|
await zha_gateway.async_update_device_storage()
|
|
await hass.async_block_till_done()
|
|
entry = zha_gateway.zha_storage.async_get_or_create_device(zha_dev_basic)
|
|
assert entry.last_seen == last_seen
|
|
|
|
# test that we can still set a good last_seen
|
|
last_seen = time.time()
|
|
zha_dev_basic.async_update_last_seen(last_seen)
|
|
assert last_seen == zha_dev_basic.last_seen
|
|
|
|
# test that we still put good values in storage
|
|
await zha_gateway.async_update_device_storage()
|
|
await hass.async_block_till_done()
|
|
entry = zha_gateway.zha_storage.async_get_or_create_device(zha_dev_basic)
|
|
assert entry.last_seen == last_seen
|
|
|
|
|
|
async def test_cleaning_up_storage(hass, zigpy_dev_basic, zha_dev_basic, hass_storage):
|
|
"""Test cleaning up zha storage and remove stale devices."""
|
|
zha_gateway = get_zha_gateway(hass)
|
|
assert zha_gateway is not None
|
|
await async_enable_traffic(hass, [zha_dev_basic])
|
|
|
|
assert zha_dev_basic.last_seen is not None
|
|
await zha_gateway.zha_storage.async_save()
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass_storage["zha.storage"]["data"]["devices"]
|
|
device = hass_storage["zha.storage"]["data"]["devices"][0]
|
|
assert device["ieee"] == str(zha_dev_basic.ieee)
|
|
|
|
zha_dev_basic.device.last_seen = time.time() - TOMBSTONE_LIFETIME - 1
|
|
await zha_gateway.async_update_device_storage()
|
|
await hass.async_block_till_done()
|
|
await zha_gateway.zha_storage.async_save()
|
|
await hass.async_block_till_done()
|
|
assert not hass_storage["zha.storage"]["data"]["devices"]
|