From 9b9d4d7dab31a211cea93ea5b7b29176e6535bc6 Mon Sep 17 00:00:00 2001
From: Jason Mahdjoub <jason.mahdjoub@distri-mind.fr>
Date: Mon, 26 May 2025 21:13:47 +0200
Subject: [PATCH] Set correct state_class for battery_stored and increase
 timeout to prevent Imeon integration disconnections (#144925)

---
 homeassistant/components/imeon_inverter/const.py           | 2 +-
 homeassistant/components/imeon_inverter/sensor.py          | 2 +-
 tests/components/imeon_inverter/snapshots/test_sensor.ambr | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/imeon_inverter/const.py b/homeassistant/components/imeon_inverter/const.py
index c71a8c72d11..fd08955c038 100644
--- a/homeassistant/components/imeon_inverter/const.py
+++ b/homeassistant/components/imeon_inverter/const.py
@@ -3,7 +3,7 @@
 from homeassistant.const import Platform
 
 DOMAIN = "imeon_inverter"
-TIMEOUT = 20
+TIMEOUT = 30
 PLATFORMS = [
     Platform.SENSOR,
 ]
diff --git a/homeassistant/components/imeon_inverter/sensor.py b/homeassistant/components/imeon_inverter/sensor.py
index b7a01c3cf17..a2f6ded5ab3 100644
--- a/homeassistant/components/imeon_inverter/sensor.py
+++ b/homeassistant/components/imeon_inverter/sensor.py
@@ -69,7 +69,7 @@ ENTITY_DESCRIPTIONS = (
         translation_key="battery_stored",
         native_unit_of_measurement=UnitOfEnergy.WATT_HOUR,
         device_class=SensorDeviceClass.ENERGY_STORAGE,
-        state_class=SensorStateClass.TOTAL,
+        state_class=SensorStateClass.MEASUREMENT,
     ),
     # Grid
     SensorEntityDescription(
diff --git a/tests/components/imeon_inverter/snapshots/test_sensor.ambr b/tests/components/imeon_inverter/snapshots/test_sensor.ambr
index d3ae33a6c8b..8816889f049 100644
--- a/tests/components/imeon_inverter/snapshots/test_sensor.ambr
+++ b/tests/components/imeon_inverter/snapshots/test_sensor.ambr
@@ -282,7 +282,7 @@
     }),
     'area_id': None,
     'capabilities': dict({
-      'state_class': <SensorStateClass.TOTAL: 'total'>,
+      'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
     }),
     'config_entry_id': <ANY>,
     'config_subentry_id': <ANY>,
@@ -321,7 +321,7 @@
     'attributes': ReadOnlyDict({
       'device_class': 'energy_storage',
       'friendly_name': 'Imeon inverter Battery stored',
-      'state_class': <SensorStateClass.TOTAL: 'total'>,
+      'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
       'unit_of_measurement': <UnitOfEnergy.WATT_HOUR: 'Wh'>,
     }),
     'context': <ANY>,