From a539d851ccc5ac49e7df5bd6a586893f7ec60156 Mon Sep 17 00:00:00 2001
From: Joost Lekkerkerker <joostlek@outlook.com>
Date: Thu, 24 Aug 2023 00:57:55 +0200
Subject: [PATCH] Use snapshot assertion for Enphase Envoy diagnostics test
 (#98910)

---
 tests/components/enphase_envoy/conftest.py    |  1 +
 .../snapshots/test_diagnostics.ambr           | 28 +++++++++++++++++
 .../enphase_envoy/test_diagnostics.py         | 30 +++++--------------
 3 files changed, 36 insertions(+), 23 deletions(-)
 create mode 100644 tests/components/enphase_envoy/snapshots/test_diagnostics.ambr

diff --git a/tests/components/enphase_envoy/conftest.py b/tests/components/enphase_envoy/conftest.py
index 355c247b182..41cbb239129 100644
--- a/tests/components/enphase_envoy/conftest.py
+++ b/tests/components/enphase_envoy/conftest.py
@@ -24,6 +24,7 @@ def config_entry_fixture(hass: HomeAssistant, config, serial_number):
     """Define a config entry fixture."""
     entry = MockConfigEntry(
         domain=DOMAIN,
+        entry_id="45a36e55aaddb2007c5f6602e0c38e72",
         title=f"Envoy {serial_number}" if serial_number else "Envoy",
         unique_id=serial_number,
         data=config,
diff --git a/tests/components/enphase_envoy/snapshots/test_diagnostics.ambr b/tests/components/enphase_envoy/snapshots/test_diagnostics.ambr
new file mode 100644
index 00000000000..098fc4ee37e
--- /dev/null
+++ b/tests/components/enphase_envoy/snapshots/test_diagnostics.ambr
@@ -0,0 +1,28 @@
+# serializer version: 1
+# name: test_entry_diagnostics
+  dict({
+    'data': dict({
+      'varies_by': 'firmware_version',
+    }),
+    'entry': dict({
+      'data': dict({
+        'host': '1.1.1.1',
+        'name': '**REDACTED**',
+        'password': '**REDACTED**',
+        'token': '**REDACTED**',
+        'username': '**REDACTED**',
+      }),
+      'disabled_by': None,
+      'domain': 'enphase_envoy',
+      'entry_id': '45a36e55aaddb2007c5f6602e0c38e72',
+      'options': dict({
+      }),
+      'pref_disable_new_entities': False,
+      'pref_disable_polling': False,
+      'source': 'user',
+      'title': '**REDACTED**',
+      'unique_id': '**REDACTED**',
+      'version': 1,
+    }),
+  })
+# ---
diff --git a/tests/components/enphase_envoy/test_diagnostics.py b/tests/components/enphase_envoy/test_diagnostics.py
index fb1a54dc522..c3659b2a9bb 100644
--- a/tests/components/enphase_envoy/test_diagnostics.py
+++ b/tests/components/enphase_envoy/test_diagnostics.py
@@ -1,5 +1,6 @@
 """Test Enphase Envoy diagnostics."""
-from homeassistant.components.diagnostics import REDACTED
+from syrupy import SnapshotAssertion
+
 from homeassistant.core import HomeAssistant
 
 from tests.components.diagnostics import get_diagnostics_for_config_entry
@@ -11,27 +12,10 @@ async def test_entry_diagnostics(
     config_entry,
     hass_client: ClientSessionGenerator,
     setup_enphase_envoy,
+    snapshot: SnapshotAssertion,
 ) -> None:
     """Test config entry diagnostics."""
-    assert await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == {
-        "entry": {
-            "entry_id": config_entry.entry_id,
-            "version": 1,
-            "domain": "enphase_envoy",
-            "title": REDACTED,
-            "data": {
-                "host": "1.1.1.1",
-                "name": REDACTED,
-                "username": REDACTED,
-                "password": REDACTED,
-                "token": REDACTED,
-            },
-            "options": {},
-            "pref_disable_new_entities": False,
-            "pref_disable_polling": False,
-            "source": "user",
-            "unique_id": REDACTED,
-            "disabled_by": None,
-        },
-        "data": {"varies_by": "firmware_version"},
-    }
+    assert (
+        await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
+        == snapshot
+    )