Add diagnostics to Switcher (#81146)

homeassistant/components/switcher_kis/diagnostics.py

@@ -0,0 +1,28 @@
+"""Diagnostics support for Switcher."""
+from __future__ import annotations
+
+from dataclasses import asdict
+from typing import Any
+
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+
+from .const import DATA_DEVICE, DOMAIN
+
+TO_REDACT = {"device_id", "ip_address", "mac_address"}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+    devices = hass.data[DOMAIN][DATA_DEVICE]
+
+    return async_redact_data(
+        {
+            "entry": entry.as_dict(),
+            "devices": [asdict(devices[d].data) for d in devices],
+        },
+        TO_REDACT,
+    )

tests/components/switcher_kis/test_diagnostics.py

@@ -0,0 +1,59 @@
+"""Tests for the diagnostics data provided by Switcher."""
+from aiohttp import ClientSession
+
+from homeassistant.components.diagnostics import REDACTED
+from homeassistant.core import HomeAssistant
+
+from . import init_integration
+from .consts import DUMMY_WATER_HEATER_DEVICE
+
+from tests.components.diagnostics import get_diagnostics_for_config_entry
+
+
+async def test_diagnostics(
+    hass: HomeAssistant, hass_client: ClientSession, mock_bridge, monkeypatch
+) -> None:
+    """Test diagnostics."""
+    entry = await init_integration(hass)
+    device = DUMMY_WATER_HEATER_DEVICE
+    monkeypatch.setattr(device, "last_data_update", "2022-09-28T16:42:12.706017")
+    mock_bridge.mock_callbacks([device])
+    await hass.async_block_till_done()
+
+    assert await get_diagnostics_for_config_entry(hass, hass_client, entry) == {
+        "devices": [
+            {
+                "auto_shutdown": "02:00:00",
+                "device_id": REDACTED,
+                "device_state": {
+                    "__type": "<enum 'DeviceState'>",
+                    "repr": "<DeviceState.ON: ('01', 'on')>",
+                },
+                "device_type": {
+                    "__type": "<enum 'DeviceType'>",
+                    "repr": "<DeviceType.V4: ('Switcher V4', '0317', "
+                    "1, <DeviceCategory.WATER_HEATER: 1>)>",
+                },
+                "electric_current": 12.8,
+                "ip_address": REDACTED,
+                "last_data_update": "2022-09-28T16:42:12.706017",
+                "mac_address": REDACTED,
+                "name": "Heater FE12",
+                "power_consumption": 2780,
+                "remaining_time": "01:29:32",
+            }
+        ],
+        "entry": {
+            "entry_id": entry.entry_id,
+            "version": 1,
+            "domain": "switcher_kis",
+            "title": "Mock Title",
+            "data": {},
+            "options": {},
+            "pref_disable_new_entities": False,
+            "pref_disable_polling": False,
+            "source": "user",
+            "unique_id": "switcher_kis",
+            "disabled_by": None,
+        },
+    }