From 384d10a51d298f7bddbd6c16dfc07bc3c4bef165 Mon Sep 17 00:00:00 2001
From: mkmer <mike.j.kasper@gmail.com>
Date: Tue, 2 Apr 2024 03:41:40 -0400
Subject: [PATCH] Add diagnostic platform to Whirlpool (#114578)

* Add diagnostic platform and tests

* lowercase variable

* Correc doc string
---
 .../components/whirlpool/diagnostics.py       | 49 +++++++++++++++++++
 .../whirlpool/snapshots/test_diagnostics.ambr | 44 +++++++++++++++++
 .../components/whirlpool/test_diagnostics.py  | 32 ++++++++++++
 3 files changed, 125 insertions(+)
 create mode 100644 homeassistant/components/whirlpool/diagnostics.py
 create mode 100644 tests/components/whirlpool/snapshots/test_diagnostics.ambr
 create mode 100644 tests/components/whirlpool/test_diagnostics.py

diff --git a/homeassistant/components/whirlpool/diagnostics.py b/homeassistant/components/whirlpool/diagnostics.py
new file mode 100644
index 00000000000..9b1dd00e7bd
--- /dev/null
+++ b/homeassistant/components/whirlpool/diagnostics.py
@@ -0,0 +1,49 @@
+"""Diagnostics support for Whirlpool."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components.diagnostics import async_redact_data
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+
+from . import WhirlpoolData
+from .const import DOMAIN
+
+TO_REDACT = {
+    "SERIAL_NUMBER",
+    "macaddress",
+    "username",
+    "password",
+    "token",
+    "unique_id",
+    "SAID",
+}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+
+    whirlpool: WhirlpoolData = hass.data[DOMAIN][config_entry.entry_id]
+    diagnostics_data = {
+        "Washer_dryers": {
+            wd["NAME"]: dict(wd.items())
+            for wd in whirlpool.appliances_manager.washer_dryers
+        },
+        "aircons": {
+            ac["NAME"]: dict(ac.items()) for ac in whirlpool.appliances_manager.aircons
+        },
+        "ovens": {
+            oven["NAME"]: dict(oven.items())
+            for oven in whirlpool.appliances_manager.ovens
+        },
+    }
+
+    return {
+        "config_entry": async_redact_data(config_entry.as_dict(), TO_REDACT),
+        "appliances": async_redact_data(diagnostics_data, TO_REDACT),
+    }
diff --git a/tests/components/whirlpool/snapshots/test_diagnostics.ambr b/tests/components/whirlpool/snapshots/test_diagnostics.ambr
new file mode 100644
index 00000000000..5a0beb112e6
--- /dev/null
+++ b/tests/components/whirlpool/snapshots/test_diagnostics.ambr
@@ -0,0 +1,44 @@
+# serializer version: 1
+# name: test_entry_diagnostics
+  dict({
+    'appliances': dict({
+      'Washer_dryers': dict({
+        'dryer': dict({
+          'NAME': 'dryer',
+          'SAID': '**REDACTED**',
+        }),
+        'washer': dict({
+          'NAME': 'washer',
+          'SAID': '**REDACTED**',
+        }),
+      }),
+      'aircons': dict({
+        'TestZone': dict({
+          'NAME': 'TestZone',
+          'SAID': '**REDACTED**',
+        }),
+      }),
+      'ovens': dict({
+      }),
+    }),
+    'config_entry': dict({
+      'data': dict({
+        'brand': 'Whirlpool',
+        'password': '**REDACTED**',
+        'region': 'EU',
+        'username': '**REDACTED**',
+      }),
+      'disabled_by': None,
+      'domain': 'whirlpool',
+      'minor_version': 1,
+      'options': dict({
+      }),
+      'pref_disable_new_entities': False,
+      'pref_disable_polling': False,
+      'source': 'user',
+      'title': 'Mock Title',
+      'unique_id': None,
+      'version': 1,
+    }),
+  })
+# ---
diff --git a/tests/components/whirlpool/test_diagnostics.py b/tests/components/whirlpool/test_diagnostics.py
new file mode 100644
index 00000000000..6cfc1b76e38
--- /dev/null
+++ b/tests/components/whirlpool/test_diagnostics.py
@@ -0,0 +1,32 @@
+"""Test Blink diagnostics."""
+
+from unittest.mock import MagicMock
+
+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.typing import ClientSessionGenerator
+
+YAML_CONFIG = {"username": "test-user", "password": "test-password"}
+
+
+async def test_entry_diagnostics(
+    hass: HomeAssistant,
+    hass_client: ClientSessionGenerator,
+    snapshot: SnapshotAssertion,
+    mock_appliances_manager_api: MagicMock,
+    mock_aircon1_api: MagicMock,
+    mock_aircon_api_instances: MagicMock,
+) -> None:
+    """Test config entry diagnostics."""
+
+    mock_entry = await init_integration(hass)
+
+    result = await get_diagnostics_for_config_entry(hass, hass_client, mock_entry)
+
+    assert result == snapshot(exclude=props("entry_id"))