From 5cd4c8e896ac557e88c5249264daafeba2268ca3 Mon Sep 17 00:00:00 2001
From: Olivier Douville <olivier.douville@yahoo.fr>
Date: Mon, 28 Apr 2025 07:55:29 +0200
Subject: [PATCH] Add missing state class in sfr-box (#143773)

* Update sensor.py - Add MEASUREMENT state class on alimvoltage and temperature sensors

This will allow state values to be stored in LTS (long term statistics)

* Update tests accordingly to previous changes in sensors

* Update tests accordingly to previous changes in sensors
---
 homeassistant/components/sfr_box/sensor.py          |  2 ++
 tests/components/sfr_box/snapshots/test_sensor.ambr | 10 ++++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/sfr_box/sensor.py b/homeassistant/components/sfr_box/sensor.py
index 8b495da56c3..ca064d137b7 100644
--- a/homeassistant/components/sfr_box/sensor.py
+++ b/homeassistant/components/sfr_box/sensor.py
@@ -174,6 +174,7 @@ SYSTEM_SENSOR_TYPES: tuple[SFRBoxSensorEntityDescription[SystemInfo], ...] = (
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         native_unit_of_measurement=UnitOfElectricPotential.MILLIVOLT,
+        state_class=SensorStateClass.MEASUREMENT,
         value_fn=lambda x: x.alimvoltage,
     ),
     SFRBoxSensorEntityDescription[SystemInfo](
@@ -182,6 +183,7 @@ SYSTEM_SENSOR_TYPES: tuple[SFRBoxSensorEntityDescription[SystemInfo], ...] = (
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
+        state_class=SensorStateClass.MEASUREMENT,
         value_fn=lambda x: _get_temperature(x.temperature),
     ),
 )
diff --git a/tests/components/sfr_box/snapshots/test_sensor.ambr b/tests/components/sfr_box/snapshots/test_sensor.ambr
index 56745c8be8e..3ad7395caad 100644
--- a/tests/components/sfr_box/snapshots/test_sensor.ambr
+++ b/tests/components/sfr_box/snapshots/test_sensor.ambr
@@ -79,7 +79,9 @@
       'aliases': set({
       }),
       'area_id': None,
-      'capabilities': None,
+      'capabilities': dict({
+        'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
+      }),
       'config_entry_id': <ANY>,
       'config_subentry_id': <ANY>,
       'device_class': None,
@@ -111,7 +113,9 @@
       'aliases': set({
       }),
       'area_id': None,
-      'capabilities': None,
+      'capabilities': dict({
+        'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
+      }),
       'config_entry_id': <ANY>,
       'config_subentry_id': <ANY>,
       'device_class': None,
@@ -591,6 +595,7 @@
       'attributes': ReadOnlyDict({
         'device_class': 'voltage',
         'friendly_name': 'SFR Box Voltage',
+        'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
         'unit_of_measurement': <UnitOfElectricPotential.MILLIVOLT: 'mV'>,
       }),
       'context': <ANY>,
@@ -604,6 +609,7 @@
       'attributes': ReadOnlyDict({
         'device_class': 'temperature',
         'friendly_name': 'SFR Box Temperature',
+        'state_class': <SensorStateClass.MEASUREMENT: 'measurement'>,
         'unit_of_measurement': <UnitOfTemperature.CELSIUS: '°C'>,
       }),
       'context': <ANY>,