From 0c901435bdb300551b08291b8ce1f01bb25db2b1 Mon Sep 17 00:00:00 2001
From: Bouwe Westerdijk <11290930+bouwew@users.noreply.github.com>
Date: Thu, 12 Oct 2023 19:57:15 +0200
Subject: [PATCH] Fix state_class of huisbaasje sensors (#101892)

---
 homeassistant/components/huisbaasje/sensor.py | 8 ++++----
 tests/components/huisbaasje/test_sensor.py    | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/huisbaasje/sensor.py b/homeassistant/components/huisbaasje/sensor.py
index 8bc86d423a1..6e3f5eaee33 100644
--- a/homeassistant/components/huisbaasje/sensor.py
+++ b/homeassistant/components/huisbaasje/sensor.py
@@ -146,7 +146,7 @@ SENSORS_INFO = [
         translation_key="energy_today",
         device_class=SensorDeviceClass.ENERGY,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
-        state_class=SensorStateClass.MEASUREMENT,
+        state_class=SensorStateClass.TOTAL_INCREASING,
         key=SOURCE_TYPE_ELECTRICITY,
         sensor_type=SENSOR_TYPE_THIS_DAY,
         precision=1,
@@ -156,7 +156,7 @@ SENSORS_INFO = [
         translation_key="energy_week",
         device_class=SensorDeviceClass.ENERGY,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
-        state_class=SensorStateClass.MEASUREMENT,
+        state_class=SensorStateClass.TOTAL_INCREASING,
         key=SOURCE_TYPE_ELECTRICITY,
         sensor_type=SENSOR_TYPE_THIS_WEEK,
         precision=1,
@@ -166,7 +166,7 @@ SENSORS_INFO = [
         translation_key="energy_month",
         device_class=SensorDeviceClass.ENERGY,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
-        state_class=SensorStateClass.MEASUREMENT,
+        state_class=SensorStateClass.TOTAL_INCREASING,
         key=SOURCE_TYPE_ELECTRICITY,
         sensor_type=SENSOR_TYPE_THIS_MONTH,
         precision=1,
@@ -176,7 +176,7 @@ SENSORS_INFO = [
         translation_key="energy_year",
         device_class=SensorDeviceClass.ENERGY,
         native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
-        state_class=SensorStateClass.MEASUREMENT,
+        state_class=SensorStateClass.TOTAL_INCREASING,
         key=SOURCE_TYPE_ELECTRICITY,
         sensor_type=SENSOR_TYPE_THIS_YEAR,
         precision=1,
diff --git a/tests/components/huisbaasje/test_sensor.py b/tests/components/huisbaasje/test_sensor.py
index b324a5be970..484dc8bac48 100644
--- a/tests/components/huisbaasje/test_sensor.py
+++ b/tests/components/huisbaasje/test_sensor.py
@@ -224,7 +224,7 @@ async def test_setup_entry(hass: HomeAssistant) -> None:
         assert energy_today.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
         assert (
             energy_today.attributes.get(ATTR_STATE_CLASS)
-            is SensorStateClass.MEASUREMENT
+            is SensorStateClass.TOTAL_INCREASING
         )
         assert (
             energy_today.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@@ -240,7 +240,7 @@ async def test_setup_entry(hass: HomeAssistant) -> None:
         assert energy_this_week.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
         assert (
             energy_this_week.attributes.get(ATTR_STATE_CLASS)
-            is SensorStateClass.MEASUREMENT
+            is SensorStateClass.TOTAL_INCREASING
         )
         assert (
             energy_this_week.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@@ -256,7 +256,7 @@ async def test_setup_entry(hass: HomeAssistant) -> None:
         assert energy_this_month.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
         assert (
             energy_this_month.attributes.get(ATTR_STATE_CLASS)
-            is SensorStateClass.MEASUREMENT
+            is SensorStateClass.TOTAL_INCREASING
         )
         assert (
             energy_this_month.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
@@ -272,7 +272,7 @@ async def test_setup_entry(hass: HomeAssistant) -> None:
         assert energy_this_year.attributes.get(ATTR_ICON) == "mdi:lightning-bolt"
         assert (
             energy_this_year.attributes.get(ATTR_STATE_CLASS)
-            is SensorStateClass.MEASUREMENT
+            is SensorStateClass.TOTAL_INCREASING
         )
         assert (
             energy_this_year.attributes.get(ATTR_UNIT_OF_MEASUREMENT)