Refresh rate sensors on time changes

Author: javaDevJT

custom_components/dte_rates/sensor.py

@@ -5,11 +5,13 @@
 from homeassistant.components.sensor import SensorDeviceClass, SensorEntity, SensorStateClass
 from homeassistant.components import persistent_notification
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.device_registry import DeviceEntryType
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.event import async_track_time_interval
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.util import dt as dt_util
+from datetime import timedelta
 
 from .const import (
     ATTR_CARD_EFFECTIVE_DATE,
@@ -66,6 +68,18 @@ def __init__(self, coordinator, entry: ConfigEntry) -> None:
         self._entry = entry
         self._notification_id = f"dte_rates_missing_rate_{entry.entry_id}"
 
+    async def async_added_to_hass(self) -> None:
+        remove_listener = async_track_time_interval(
+            self.hass,
+            self._handle_time_interval,
+            timedelta(minutes=1),
+        )
+        self.async_on_remove(remove_listener)
+
+    @callback
+    def _handle_time_interval(self, _now) -> None:
+        self.async_write_ha_state()
+
     def _selected_rate(self) -> RatePlan | None:
         selected = self._entry.data[CONF_SELECTED_RATE]
         return self.coordinator.data.rates.get(selected)

tests/conftest.py

@@ -64,6 +64,15 @@ class CoordinatorEntity:
         def __init__(self, coordinator):
             self.coordinator = coordinator
 
+        async def async_added_to_hass(self):
+            return None
+
+        def async_on_remove(self, _remove_callback):
+            return None
+
+        def async_write_ha_state(self):
+            return None
+
     class SensorEntity:
         @property
         def native_unit_of_measurement(self):
@@ -77,6 +86,9 @@ class SensorDeviceClass:
     class HomeAssistant:
         pass
 
+    def callback(func):
+        return func
+
     class DeviceEntryType:
         SERVICE = "service"
 
@@ -99,7 +111,7 @@ def __init__(self, url_path, path, cache_headers=True):
             FlowResult=FlowResult,
         ),
         "homeassistant.data_entry_flow": _make_module("homeassistant.data_entry_flow", FlowResult=FlowResult),
-        "homeassistant.core": _make_module("homeassistant.core", HomeAssistant=HomeAssistant),
+        "homeassistant.core": _make_module("homeassistant.core", HomeAssistant=HomeAssistant, callback=callback),
         "homeassistant.helpers": _make_module("homeassistant.helpers"),
         "homeassistant.helpers.update_coordinator": _make_module(
             "homeassistant.helpers.update_coordinator",
@@ -108,6 +120,10 @@ def __init__(self, url_path, path, cache_headers=True):
             CoordinatorEntity=CoordinatorEntity,
         ),
         "homeassistant.helpers.entity_platform": _make_module("homeassistant.helpers.entity_platform", AddEntitiesCallback=MagicMock()),
+        "homeassistant.helpers.event": _make_module(
+            "homeassistant.helpers.event",
+            async_track_time_interval=MagicMock(return_value=lambda: None),
+        ),
         "homeassistant.helpers.aiohttp_client": _make_module("homeassistant.helpers.aiohttp_client", async_get_clientsession=MagicMock()),
         "homeassistant.helpers.device_registry": _make_module("homeassistant.helpers.device_registry", DeviceEntryType=DeviceEntryType),
         "homeassistant.components": _make_module("homeassistant.components"),

tests/test_sensor.py

@@ -4,6 +4,7 @@
 from decimal import Decimal
 from types import SimpleNamespace
 from unittest.mock import MagicMock
+import pytest
 
 from homeassistant.components import persistent_notification
 from custom_components.dte_rates.const import CONF_NET_METERING, CONF_SELECTED_RATE
@@ -206,3 +207,33 @@ def test_available_rate_dismisses_existing_notification(monkeypatch):
 
     create_mock.assert_not_called()
     dismiss_mock.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_sensor_registers_time_interval_refresh(monkeypatch):
+    monkeypatch.setattr("custom_components.dte_rates.sensor.dt_util.now", lambda: datetime(2026, 3, 1, 12, 0))
+    track_mock = MagicMock(return_value=lambda: None)
+    monkeypatch.setattr("custom_components.dte_rates.sensor.async_track_time_interval", track_mock)
+
+    coordinator = _coordinator_with_rate()
+    entry = SimpleNamespace(entry_id="entry_11", data={CONF_SELECTED_RATE: "D1.11", CONF_NET_METERING: False})
+    sensor = DteImportRateSensor(coordinator, entry)
+    sensor.hass = SimpleNamespace()
+    sensor.async_on_remove = MagicMock()
+
+    await sensor.async_added_to_hass()
+
+    track_mock.assert_called_once()
+    sensor.async_on_remove.assert_called_once()
+
+
+def test_time_interval_callback_writes_state(monkeypatch):
+    monkeypatch.setattr("custom_components.dte_rates.sensor.dt_util.now", lambda: datetime(2026, 3, 1, 12, 0))
+    coordinator = _coordinator_with_rate()
+    entry = SimpleNamespace(entry_id="entry_12", data={CONF_SELECTED_RATE: "D1.11", CONF_NET_METERING: False})
+    sensor = DteImportRateSensor(coordinator, entry)
+    sensor.async_write_ha_state = MagicMock()
+
+    sensor._handle_time_interval(None)
+
+    sensor.async_write_ha_state.assert_called_once()