Add test

Signed-off-by: Daniel Hjelseth Høyer <github@dahoiv.net>

tests/components/mill/test_statistics.py

@@ -0,0 +1,60 @@
+"""Test adding external statistics from Mill."""
+
+from unittest.mock import AsyncMock
+
+from mill import Heater, Mill
+
+from homeassistant.components.mill.const import DOMAIN
+from homeassistant.components.mill.coordinator import MillDataUpdateCoordinator
+from homeassistant.components.recorder import Recorder
+from homeassistant.components.recorder.statistics import statistics_during_period
+from homeassistant.core import HomeAssistant
+from homeassistant.util import dt as dt_util
+
+from tests.components.recorder.common import async_wait_recording_done
+
+
+async def test_async_setup_entry(recorder_mock: Recorder, hass: HomeAssistant) -> None:
+    """Test setup Mill."""
+
+    data = {
+        dt_util.parse_datetime("2024-12-03T00:00:00+01:00"): 2,
+        dt_util.parse_datetime("2024-12-03T01:00:00+01:00"): 3,
+        dt_util.parse_datetime("2024-12-03T02:00:00+01:00"): 4,
+    }
+
+    mill_data_connection = Mill("", "", websession=AsyncMock())
+    mill_data_connection.fetch_heater_and_sensor_data = AsyncMock(return_value=None)
+    mill_data_connection.devices = {"dev_id": Heater(name="heater_name")}
+    mill_data_connection.fetch_historic_energy_usage = AsyncMock(return_value=data)
+
+    statistic_id = f"{DOMAIN}:energy_dev_id"
+
+    coordinator = MillDataUpdateCoordinator(
+        hass, mill_data_connection=mill_data_connection
+    )
+    await coordinator._async_update_data()
+    await async_wait_recording_done(hass)
+    stats = await hass.async_add_executor_job(
+        statistics_during_period,
+        hass,
+        next(iter(data)),
+        None,
+        {statistic_id},
+        "hour",
+        None,
+        {"start", "state", "mean", "min", "max", "last_reset", "sum"},
+    )
+
+    assert len(stats) == 1
+    assert len(stats[statistic_id]) == 3
+    _sum = 0
+    for stat in stats[statistic_id]:
+        start = dt_util.utc_from_timestamp(stat["start"])
+        assert start in data
+        assert stat["state"] == data[start]
+        assert stat["last_reset"] is None
+
+        _sum += data[start]
+        assert stat["sum"] == _sum
+    assert _sum == sum(data.values())