From a6f508c624d3a264fff98804d07af490f844a006 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= Date: Mon, 17 Mar 2025 21:22:37 +0100 Subject: [PATCH] mill MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Daniel Hjelseth Høyer --- homeassistant/components/mill/coordinator.py | 5 +++-- .../mill/{test_statistics.py => test_coordinator.py} | 0 2 files changed, 3 insertions(+), 2 deletions(-) rename tests/components/mill/{test_statistics.py => test_coordinator.py} (100%) diff --git a/homeassistant/components/mill/coordinator.py b/homeassistant/components/mill/coordinator.py index b0166cc3f47..b49df121b30 100644 --- a/homeassistant/components/mill/coordinator.py +++ b/homeassistant/components/mill/coordinator.py @@ -79,12 +79,13 @@ class MillHistoricDataUpdateCoordinator(DataUpdateCoordinator): timedelta(hours=1) + now.replace(minute=0, second=0) - now ) + recoder_instance = get_instance(self.hass) for dev_id, heater in self.mill_data_connection.devices.items(): if not isinstance(heater, Heater): continue statistic_id = f"{DOMAIN}:energy_{slugify(dev_id)}" - last_stats = await get_instance(self.hass).async_add_executor_job( + last_stats = await recoder_instance.async_add_executor_job( get_last_statistics, self.hass, 1, statistic_id, True, set() ) @@ -115,7 +116,7 @@ class MillHistoricDataUpdateCoordinator(DataUpdateCoordinator): hourly_data = dict(sorted(hourly_data.items(), key=lambda x: x[0])) start_time = next(iter(hourly_data)) - stats = await get_instance(self.hass).async_add_executor_job( + stats = await recoder_instance.async_add_executor_job( statistics_during_period, self.hass, start_time, diff --git a/tests/components/mill/test_statistics.py b/tests/components/mill/test_coordinator.py similarity index 100% rename from tests/components/mill/test_statistics.py rename to tests/components/mill/test_coordinator.py