Use snapshot assertion in switcher_kis diagnostics test (home-assistant#155273)

Author: epenet

tests/components/switcher_kis/__init__.py

@@ -17,7 +17,12 @@ async def init_integration(
     if token is not None:
         data[CONF_TOKEN] = token
 
-    entry = MockConfigEntry(domain=DOMAIN, data=data, unique_id=DOMAIN)
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        data=data,
+        entry_id="01K8K1GVZQW5RF1SMCEF1D55NE",
+        unique_id=DOMAIN,
+    )
     entry.add_to_hass(hass)
 
     await hass.config_entries.async_setup(entry.entry_id)

tests/components/switcher_kis/snapshots/test_diagnostics.ambr

@@ -0,0 +1,48 @@
+# serializer version: 1
+# name: test_diagnostics
+  dict({
+    'devices': list([
+      dict({
+        'auto_shutdown': '02:00:00',
+        'device_id': '**REDACTED**',
+        'device_key': '**REDACTED**',
+        'device_state': dict({
+          '__type': "<enum 'DeviceState'>",
+          'repr': "<DeviceState.ON: ('01', 'on')>",
+        }),
+        'device_type': dict({
+          '__type': "<enum 'DeviceType'>",
+          'repr': "<DeviceType.V4: ('Switcher V4', '0317', 1, <DeviceCategory.WATER_HEATER: 1>, False)>",
+        }),
+        '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',
+        'token_needed': False,
+      }),
+    ]),
+    'entry': dict({
+      'data': dict({
+      }),
+      'disabled_by': None,
+      'discovery_keys': dict({
+      }),
+      'domain': 'switcher_kis',
+      'entry_id': '01K8K1GVZQW5RF1SMCEF1D55NE',
+      'minor_version': 1,
+      'options': dict({
+      }),
+      'pref_disable_new_entities': False,
+      'pref_disable_polling': False,
+      'source': 'user',
+      'subentries': list([
+      ]),
+      'title': 'Mock Title',
+      'unique_id': 'switcher_kis',
+      'version': 1,
+    }),
+  })
+# ---

tests/components/switcher_kis/test_diagnostics.py

@@ -1,14 +1,14 @@
 """Tests for the diagnostics data provided by Switcher."""
 
 import pytest
+from syrupy.assertion import SnapshotAssertion
+from syrupy.filters import props
 
-from homeassistant.components.diagnostics import REDACTED
 from homeassistant.core import HomeAssistant
 
 from . import init_integration
 from .consts import DUMMY_WATER_HEATER_DEVICE
 
-from tests.common import ANY
 from tests.components.diagnostics import get_diagnostics_for_config_entry
 from tests.typing import ClientSessionGenerator
 
@@ -18,6 +18,7 @@ async def test_diagnostics(
     hass_client: ClientSessionGenerator,
     mock_bridge,
     monkeypatch: pytest.MonkeyPatch,
+    snapshot: SnapshotAssertion,
 ) -> None:
     """Test diagnostics."""
     entry = await init_integration(hass)
@@ -26,49 +27,6 @@ async def test_diagnostics(
     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_key": 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>, False)>"
-                    ),
-                },
-                "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",
-                "token_needed": False,
-            }
-        ],
-        "entry": {
-            "entry_id": entry.entry_id,
-            "version": 1,
-            "minor_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,
-            "created_at": ANY,
-            "modified_at": ANY,
-            "discovery_keys": {},
-            "subentries": [],
-        },
-    }
+    assert await get_diagnostics_for_config_entry(hass, hass_client, entry) == snapshot(
+        exclude=props("created_at", "modified_at")
+    )