hass-core/tests/components/deconz/test_diagnostics.py

"""Test deCONZ diagnostics."""

from pydeconz.websocket import STATE_RUNNING

from homeassistant.components.deconz.const import CONF_MASTER_GATEWAY
from homeassistant.components.diagnostics import REDACTED
from homeassistant.const import CONF_API_KEY, CONF_HOST, CONF_PORT, Platform

from .test_gateway import HOST, PORT, setup_deconz_integration

from tests.components.diagnostics import get_diagnostics_for_config_entry


async def test_entry_diagnostics(
    hass, hass_client, aioclient_mock, mock_deconz_websocket
):
    """Test config entry diagnostics."""
    config_entry = await setup_deconz_integration(hass, aioclient_mock)

    await mock_deconz_websocket(state=STATE_RUNNING)
    await hass.async_block_till_done()

    assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
        "config": {CONF_API_KEY: REDACTED, CONF_HOST: HOST, CONF_PORT: PORT},
        "options": {CONF_MASTER_GATEWAY: True},
        "deconz_config": {
            "bridgeid": REDACTED,
            "ipaddress": HOST,
            "mac": REDACTED,
            "modelid": "deCONZ",
            "name": "deCONZ mock gateway",
            "sw_version": "2.05.69",
            "uuid": "1234",
            "websocketport": 1234,
        },
        "websocket_state": STATE_RUNNING,
        "deconz_ids": {},
        "entities": {
            str(Platform.ALARM_CONTROL_PANEL): [],
            str(Platform.BINARY_SENSOR): [],
            str(Platform.CLIMATE): [],
            str(Platform.COVER): [],
            str(Platform.FAN): [],
            str(Platform.LIGHT): [],
            str(Platform.LOCK): [],
            str(Platform.NUMBER): [],
            str(Platform.SENSOR): [],
            str(Platform.SIREN): [],
            str(Platform.SWITCH): [],
        },
        "events": {},
        "alarm_systems": {},
        "groups": {},
        "lights": {},
        "scenes": {},
        "sensors": {},
    }