From 07e251d488b66b644eea48af90606a0851c74e41 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Sun, 1 Sep 2024 22:04:29 +0200
Subject: [PATCH] Add diagnostics platform to modern forms (#125032)

---
 .../components/modern_forms/diagnostics.py    | 36 +++++++++++++
 .../snapshots/test_diagnostics.ambr           | 50 +++++++++++++++++++
 .../modern_forms/test_diagnostics.py          | 26 ++++++++++
 3 files changed, 112 insertions(+)
 create mode 100644 homeassistant/components/modern_forms/diagnostics.py
 create mode 100644 tests/components/modern_forms/snapshots/test_diagnostics.ambr
 create mode 100644 tests/components/modern_forms/test_diagnostics.py

diff --git a/homeassistant/components/modern_forms/diagnostics.py b/homeassistant/components/modern_forms/diagnostics.py
new file mode 100644
index 00000000000..0011a7c3bab
--- /dev/null
+++ b/homeassistant/components/modern_forms/diagnostics.py
@@ -0,0 +1,36 @@
+"""Diagnostics support for Modern Forms."""
+
+from __future__ import annotations
+
+from dataclasses import asdict
+from typing import TYPE_CHECKING, Any
+
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_MAC
+from homeassistant.core import HomeAssistant
+
+from .const import DOMAIN
+from .coordinator import ModernFormsDataUpdateCoordinator
+
+REDACT_CONFIG = {CONF_MAC}
+REDACT_DEVICE_INFO = {"mac_address", "owner"}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+    coordinator: ModernFormsDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
+    if TYPE_CHECKING:
+        assert coordinator is not None
+
+    return {
+        "config_entry": async_redact_data(entry.as_dict(), REDACT_CONFIG),
+        "device": {
+            "info": async_redact_data(
+                asdict(coordinator.modern_forms.info), REDACT_DEVICE_INFO
+            ),
+            "status": asdict(coordinator.modern_forms.status),
+        },
+    }
diff --git a/tests/components/modern_forms/snapshots/test_diagnostics.ambr b/tests/components/modern_forms/snapshots/test_diagnostics.ambr
new file mode 100644
index 00000000000..56e299aa12a
--- /dev/null
+++ b/tests/components/modern_forms/snapshots/test_diagnostics.ambr
@@ -0,0 +1,50 @@
+# serializer version: 1
+# name: test_entry_diagnostics
+  dict({
+    'config_entry': dict({
+      'data': dict({
+        'host': '192.168.1.123',
+        'mac': '**REDACTED**',
+      }),
+      'disabled_by': None,
+      'domain': 'modern_forms',
+      'minor_version': 1,
+      'options': dict({
+      }),
+      'pref_disable_new_entities': False,
+      'pref_disable_polling': False,
+      'source': 'user',
+      'title': 'Mock Title',
+      'unique_id': None,
+      'version': 1,
+    }),
+    'device': dict({
+      'info': dict({
+        'client_id': 'MF_000000000000',
+        'device_name': 'ModernFormsFan',
+        'fan_motor_type': 'DC125X25',
+        'fan_type': '1818-56',
+        'federated_identity': 'us-east-1:f3da237b-c19c-4f61-b387-0e6dde2e470b',
+        'firmware_url': '',
+        'firmware_version': '01.03.0025',
+        'light_type': 'F6IN-120V-R1-30',
+        'mac_address': '**REDACTED**',
+        'main_mcu_firmware_version': '01.03.3008',
+        'owner': '**REDACTED**',
+        'product_sku': '',
+        'production_lot_number': '',
+      }),
+      'status': dict({
+        'adaptive_learning_enabled': False,
+        'away_mode_enabled': False,
+        'fan_direction': 'forward',
+        'fan_on': True,
+        'fan_sleep_timer': 0,
+        'fan_speed': 3,
+        'light_brightness': 50,
+        'light_on': True,
+        'light_sleep_timer': 0,
+      }),
+    }),
+  })
+# ---
diff --git a/tests/components/modern_forms/test_diagnostics.py b/tests/components/modern_forms/test_diagnostics.py
new file mode 100644
index 00000000000..9eb2e4efa94
--- /dev/null
+++ b/tests/components/modern_forms/test_diagnostics.py
@@ -0,0 +1,26 @@
+"""Tests for the Modern Forms diagnostics platform."""
+
+from syrupy import SnapshotAssertion
+from syrupy.filters import props
+
+from homeassistant.core import HomeAssistant
+
+from . import init_integration
+
+from tests.components.diagnostics import get_diagnostics_for_config_entry
+from tests.test_util.aiohttp import AiohttpClientMocker
+from tests.typing import ClientSessionGenerator
+
+
+async def test_entry_diagnostics(
+    hass: HomeAssistant,
+    aioclient_mock: AiohttpClientMocker,
+    hass_client: ClientSessionGenerator,
+    snapshot: SnapshotAssertion,
+) -> None:
+    """Test the creation and values of the Modern Forms fans."""
+    entry = await init_integration(hass, aioclient_mock)
+
+    result = await get_diagnostics_for_config_entry(hass, hass_client, entry)
+
+    assert result == snapshot(exclude=props("created_at", "modified_at", "entry_id"))