Add diagnostic platform to utility_meter (#114967)

* Add diagnostic to identify next_reset

* Add test

* add next_reset attr

* Trigger CI

* set as _unrecorded_attributes

tests/components/utility_meter/snapshots/test_diagnostics.ambr

@@ -0,0 +1,65 @@
+# serializer version: 1
+# name: test_diagnostics
+  dict({
+    'config_entry': dict({
+      'data': dict({
+      }),
+      'disabled_by': None,
+      'domain': 'utility_meter',
+      'minor_version': 1,
+      'options': dict({
+        'cycle': 'monthly',
+        'delta_values': False,
+        'name': 'Energy Bill',
+        'net_consumption': False,
+        'offset': 0,
+        'periodically_resetting': True,
+        'source': 'sensor.input1',
+        'tariffs': list([
+          'tariff0',
+          'tariff1',
+        ]),
+      }),
+      'pref_disable_new_entities': False,
+      'pref_disable_polling': False,
+      'source': 'user',
+      'title': 'Energy Bill',
+      'unique_id': None,
+      'version': 2,
+    }),
+    'tariff_sensors': list([
+      dict({
+        'entity_id': 'sensor.energy_bill_tariff0',
+        'extra_attributes': dict({
+          'cron pattern': '0 0 1 * *',
+          'last_period': '0',
+          'last_reset': '2024-04-05T00:00:00+00:00',
+          'last_valid_state': 'None',
+          'meter_period': 'monthly',
+          'next_reset': '2024-05-01T00:00:00-07:00',
+          'source': 'sensor.input1',
+          'status': 'collecting',
+          'tariff': 'tariff0',
+        }),
+        'last_sensor_data': None,
+        'name': 'Energy Bill tariff0',
+      }),
+      dict({
+        'entity_id': 'sensor.energy_bill_tariff1',
+        'extra_attributes': dict({
+          'cron pattern': '0 0 1 * *',
+          'last_period': '0',
+          'last_reset': '2024-04-05T00:00:00+00:00',
+          'last_valid_state': 'None',
+          'meter_period': 'monthly',
+          'next_reset': '2024-05-01T00:00:00-07:00',
+          'source': 'sensor.input1',
+          'status': 'paused',
+          'tariff': 'tariff1',
+        }),
+        'last_sensor_data': None,
+        'name': 'Energy Bill tariff1',
+      }),
+    ]),
+  })
+# ---

tests/components/utility_meter/test_diagnostics.py

@@ -0,0 +1,127 @@
+"""Test Utility Meter diagnostics."""
+
+from aiohttp.test_utils import TestClient
+from freezegun import freeze_time
+from syrupy import SnapshotAssertion
+
+from homeassistant.auth.models import Credentials
+from homeassistant.components.utility_meter.const import DOMAIN
+from homeassistant.components.utility_meter.sensor import ATTR_LAST_RESET
+from homeassistant.core import HomeAssistant, State
+
+from tests.common import (
+    CLIENT_ID,
+    MockConfigEntry,
+    MockUser,
+    mock_restore_cache_with_extra_data,
+)
+from tests.components.diagnostics import get_diagnostics_for_config_entry
+from tests.typing import ClientSessionGenerator
+
+
+async def generate_new_hass_access_token(
+    hass: HomeAssistant, hass_admin_user: MockUser, hass_admin_credential: Credentials
+) -> str:
+    """Return an access token to access Home Assistant."""
+    await hass.auth.async_link_user(hass_admin_user, hass_admin_credential)
+
+    refresh_token = await hass.auth.async_create_refresh_token(
+        hass_admin_user, CLIENT_ID, credential=hass_admin_credential
+    )
+    return hass.auth.async_create_access_token(refresh_token)
+
+
+def _get_test_client_generator(
+    hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, new_token: str
+):
+    """Return a test client generator.""."""
+
+    async def auth_client() -> TestClient:
+        return await aiohttp_client(
+            hass.http.app, headers={"Authorization": f"Bearer {new_token}"}
+        )
+
+    return auth_client
+
+
+def limit_diagnostic_attrs(prop, path) -> bool:
+    """Mark attributes to exclude from diagnostic snapshot."""
+    return prop in {"entry_id"}
+
+
+@freeze_time("2024-04-06 00:00:00+00:00")
+async def test_diagnostics(
+    hass: HomeAssistant,
+    aiohttp_client: ClientSessionGenerator,
+    hass_admin_user: MockUser,
+    hass_admin_credential: Credentials,
+    socket_enabled: None,
+    snapshot: SnapshotAssertion,
+) -> None:
+    """Test generating diagnostics for a config entry."""
+
+    # Setup the config entry
+    config_entry = MockConfigEntry(
+        data={},
+        domain=DOMAIN,
+        options={
+            "cycle": "monthly",
+            "delta_values": False,
+            "name": "Energy Bill",
+            "net_consumption": False,
+            "offset": 0,
+            "periodically_resetting": True,
+            "source": "sensor.input1",
+            "tariffs": [
+                "tariff0",
+                "tariff1",
+            ],
+        },
+        title="Energy Bill",
+    )
+
+    last_reset = "2024-04-05T00:00:00+00:00"
+
+    # Set up the sensors restore data
+    mock_restore_cache_with_extra_data(
+        hass,
+        [
+            (
+                State(
+                    "sensor.energy_bill_tariff0",
+                    "3",
+                    attributes={
+                        ATTR_LAST_RESET: last_reset,
+                    },
+                ),
+                {},
+            ),
+            (
+                State(
+                    "sensor.energy_bill_tariff1",
+                    "7",
+                    attributes={
+                        ATTR_LAST_RESET: last_reset,
+                    },
+                ),
+                {},
+            ),
+        ],
+    )
+
+    config_entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    # Since we are freezing time only when we enter this test, we need to
+    # manually create a new token and clients since the token created by
+    # the fixtures would not be valid.
+    new_token = await generate_new_hass_access_token(
+        hass, hass_admin_user, hass_admin_credential
+    )
+
+    diag = await get_diagnostics_for_config_entry(
+        hass, _get_test_client_generator(hass, aiohttp_client, new_token), config_entry
+    )
+
+    assert diag == snapshot(exclude=limit_diagnostic_attrs)