38270ee823
* Add the SiLabs flasher as a dependency
* Create a repair if the wrong firmware is detected on an EZSP device
* Update homeassistant/components/zha/strings.json
Co-authored-by: c0ffeeca7 <38767475+c0ffeeca7@users.noreply.github.com>
* Provide the ZHA config entry as a reusable fixture
* Create a separate repair when using non-Nabu Casa hardware
* Add unit tests
* Drop extraneous `config_entry.add_to_hass` added in 021def44
* Fully unit test all edge cases
* Move `socket://`-ignoring logic into repair function
* Open a repair from ZHA flows when the wrong firmware is running
* Fix existing unit tests
* Link to the flashing section in the documentation
* Reduce repair severity to `ERROR`
* Make issue persistent
* Add unit tests for new radio probing states
* Add unit tests for new config flow steps
* Handle probing failure raising an exception
* Implement review suggestions
* Address review comments
---------
Co-authored-by: c0ffeeca7 <38767475+c0ffeeca7@users.noreply.github.com>
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
"""Tests for the diagnostics data provided by the ESPHome integration."""
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import zigpy.profiles.zha as zha
|
|
import zigpy.zcl.clusters.security as security
|
|
|
|
from homeassistant.components.diagnostics import REDACTED
|
|
from homeassistant.components.zha.core.const import DATA_ZHA, DATA_ZHA_GATEWAY
|
|
from homeassistant.components.zha.core.device import ZHADevice
|
|
from homeassistant.components.zha.diagnostics import KEYS_TO_REDACT
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.device_registry import async_get
|
|
|
|
from .conftest import SIG_EP_INPUT, SIG_EP_OUTPUT, SIG_EP_PROFILE, SIG_EP_TYPE
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.diagnostics import (
|
|
get_diagnostics_for_config_entry,
|
|
get_diagnostics_for_device,
|
|
)
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
CONFIG_ENTRY_DIAGNOSTICS_KEYS = [
|
|
"config",
|
|
"config_entry",
|
|
"application_state",
|
|
"versions",
|
|
]
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def required_platforms_only():
|
|
"""Only set up the required platform and required base platforms to speed up tests."""
|
|
with patch(
|
|
"homeassistant.components.zha.PLATFORMS", (Platform.ALARM_CONTROL_PANEL,)
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def zigpy_device(zigpy_device_mock):
|
|
"""Device tracker zigpy device."""
|
|
endpoints = {
|
|
1: {
|
|
SIG_EP_INPUT: [security.IasAce.cluster_id],
|
|
SIG_EP_OUTPUT: [],
|
|
SIG_EP_TYPE: zha.DeviceType.IAS_ANCILLARY_CONTROL,
|
|
SIG_EP_PROFILE: zha.PROFILE_ID,
|
|
}
|
|
}
|
|
return zigpy_device_mock(
|
|
endpoints, node_descriptor=b"\x02@\x8c\x02\x10RR\x00\x00\x00R\x00\x00"
|
|
)
|
|
|
|
|
|
async def test_diagnostics_for_config_entry(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
config_entry: MockConfigEntry,
|
|
zha_device_joined,
|
|
zigpy_device,
|
|
) -> None:
|
|
"""Test diagnostics for config entry."""
|
|
await zha_device_joined(zigpy_device)
|
|
|
|
gateway = hass.data[DATA_ZHA][DATA_ZHA_GATEWAY]
|
|
scan = {c: c for c in range(11, 26 + 1)}
|
|
|
|
with patch.object(gateway.application_controller, "energy_scan", return_value=scan):
|
|
diagnostics_data = await get_diagnostics_for_config_entry(
|
|
hass, hass_client, config_entry
|
|
)
|
|
|
|
for key in CONFIG_ENTRY_DIAGNOSTICS_KEYS:
|
|
assert key in diagnostics_data
|
|
assert diagnostics_data[key] is not None
|
|
|
|
# Energy scan results are presented as a percentage. JSON object keys also must be
|
|
# strings, not integers.
|
|
assert diagnostics_data["energy_scan"] == {
|
|
str(k): 100 * v / 255 for k, v in scan.items()
|
|
}
|
|
|
|
|
|
async def test_diagnostics_for_device(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
config_entry: MockConfigEntry,
|
|
zha_device_joined,
|
|
zigpy_device,
|
|
) -> None:
|
|
"""Test diagnostics for device."""
|
|
zha_device: ZHADevice = await zha_device_joined(zigpy_device)
|
|
dev_reg = async_get(hass)
|
|
device = dev_reg.async_get_device(identifiers={("zha", str(zha_device.ieee))})
|
|
assert device
|
|
diagnostics_data = await get_diagnostics_for_device(
|
|
hass, hass_client, config_entry, device
|
|
)
|
|
assert diagnostics_data
|
|
device_info: dict = zha_device.zha_device_info
|
|
for key in device_info:
|
|
assert key in diagnostics_data
|
|
if key not in KEYS_TO_REDACT:
|
|
assert key in diagnostics_data
|
|
else:
|
|
assert diagnostics_data[key] == REDACTED
|