a3061bda60
* Refactor ZHA component tests fixtures. * Add tests for ZHA device discovery. * Refactor ZHA registry MatchRule. Allow callables as a matching criteria. Allow sets for model & manufacturer. * Minor ZHA class registry refactoring. Less cluttered strict_matching registrations. * Add entities only if there are any. * Migrate rest of ZHA platforms to ZHA registry. * Pylint fixes.
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""Test zha device discovery."""
|
|
|
|
import asyncio
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.zha.core.channels import EventRelayChannel
|
|
import homeassistant.components.zha.core.const as zha_const
|
|
import homeassistant.components.zha.core.discovery as disc
|
|
import homeassistant.components.zha.core.gateway as core_zha_gw
|
|
|
|
from .common import make_device
|
|
from .zha_devices_list import DEVICES
|
|
|
|
|
|
@pytest.mark.parametrize("device", DEVICES)
|
|
async def test_devices(device, zha_gateway: core_zha_gw.ZHAGateway, hass, config_entry):
|
|
"""Test device discovery."""
|
|
|
|
zigpy_device = make_device(
|
|
device["endpoints"],
|
|
"00:11:22:33:44:55:66:77",
|
|
device["manufacturer"],
|
|
device["model"],
|
|
)
|
|
|
|
with mock.patch(
|
|
"homeassistant.components.zha.core.discovery._async_create_cluster_channel",
|
|
wraps=disc._async_create_cluster_channel,
|
|
) as cr_ch:
|
|
await zha_gateway.async_device_restored(zigpy_device)
|
|
await hass.async_block_till_done()
|
|
tasks = [
|
|
hass.config_entries.async_forward_entry_setup(config_entry, component)
|
|
for component in zha_const.COMPONENTS
|
|
]
|
|
await asyncio.gather(*tasks)
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
entity_ids = hass.states.async_entity_ids()
|
|
await hass.async_block_till_done()
|
|
zha_entities = {
|
|
ent for ent in entity_ids if ent.split(".")[0] in zha_const.COMPONENTS
|
|
}
|
|
|
|
event_channels = {
|
|
arg[0].cluster_id
|
|
for arg, kwarg in cr_ch.call_args_list
|
|
if kwarg.get("channel_class") == EventRelayChannel
|
|
}
|
|
|
|
assert zha_entities == set(device["entities"])
|
|
assert event_channels == set(device["event_channels"])
|