Change uptime sensor to SensorDeviceClass.UPTIME in IronOS integration (#169699)

Author: tr4nt0r

homeassistant/components/iron_os/sensor.py

@@ -2,6 +2,7 @@
 
 from collections.abc import Callable
 from dataclasses import dataclass
+from datetime import datetime, timedelta
 from enum import StrEnum
 
 from pynecil import LiveDataResponse, OperatingMode, PowerSource
@@ -23,6 +24,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType
+from homeassistant.util import dt as dt_util
 
 from . import IronOSConfigEntry
 from .const import OHM
@@ -56,7 +58,7 @@ class PinecilSensor(StrEnum):
 class IronOSSensorEntityDescription(SensorEntityDescription):
     """IronOS sensor entity descriptions."""
 
-    value_fn: Callable[[LiveDataResponse, bool], StateType]
+    value_fn: Callable[[LiveDataResponse, bool], StateType | datetime]
 
 
 PINECIL_SENSOR_DESCRIPTIONS: tuple[IronOSSensorEntityDescription, ...] = (
@@ -116,10 +118,14 @@ class IronOSSensorEntityDescription(SensorEntityDescription):
     IronOSSensorEntityDescription(
         key=PinecilSensor.UPTIME,
         translation_key=PinecilSensor.UPTIME,
-        native_unit_of_measurement=UnitOfTime.SECONDS,
-        device_class=SensorDeviceClass.DURATION,
-        state_class=SensorStateClass.TOTAL_INCREASING,
-        value_fn=lambda data, _: data.uptime,
+        device_class=SensorDeviceClass.UPTIME,
+        value_fn=(
+            lambda data, _: (
+                (dt_util.utcnow() - timedelta(seconds=data.uptime))
+                if data.uptime is not None
+                else None
+            )
+        ),
         entity_category=EntityCategory.DIAGNOSTIC,
     ),
     IronOSSensorEntityDescription(
@@ -200,7 +206,7 @@ class IronOSSensorEntity(IronOSBaseEntity, SensorEntity):
     coordinator: IronOSLiveDataCoordinator
 
     @property
-    def native_value(self) -> StateType:
+    def native_value(self) -> StateType | datetime:
         """Return sensor state."""
         return self.entity_description.value_fn(
             self.coordinator.data, self.coordinator.has_tip

tests/components/iron_os/snapshots/test_sensor.ambr

@@ -723,9 +723,7 @@
       None,
     ]),
     'area_id': None,
-    'capabilities': dict({
-      'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
-    }),
+    'capabilities': None,
     'config_entry_id': <ANY>,
     'config_subentry_id': <ANY>,
     'device_class': None,
@@ -743,11 +741,8 @@
     'name': None,
     'object_id_base': 'Uptime',
     'options': dict({
-      'sensor': dict({
-        'suggested_display_precision': 2,
-      }),
     }),
-    'original_device_class': <SensorDeviceClass.DURATION: 'duration'>,
+    'original_device_class': <SensorDeviceClass.UPTIME: 'uptime'>,
     'original_icon': None,
     'original_name': 'Uptime',
     'platform': 'iron_os',
@@ -756,22 +751,20 @@
     'supported_features': 0,
     'translation_key': <PinecilSensor.UPTIME: 'uptime'>,
     'unique_id': 'c0:ff:ee:c0:ff:ee_uptime',
-    'unit_of_measurement': <UnitOfTime.SECONDS: 's'>,
+    'unit_of_measurement': None,
   })
 # ---
 # name: test_sensors[sensor.pinecil_uptime-state]
   StateSnapshot({
     'attributes': ReadOnlyDict({
-      'device_class': 'duration',
+      'device_class': 'uptime',
       'friendly_name': 'Pinecil Uptime',
-      'state_class': <SensorStateClass.TOTAL_INCREASING: 'total_increasing'>,
-      'unit_of_measurement': <UnitOfTime.SECONDS: 's'>,
     }),
     'context': <ANY>,
     'entity_id': 'sensor.pinecil_uptime',
     'last_changed': <ANY>,
     'last_reported': <ANY>,
     'last_updated': <ANY>,
-    'state': '1671',
+    'state': '2026-05-03T17:53:03+00:00',
   })
 # ---

tests/components/iron_os/test_sensor.py

@@ -28,6 +28,7 @@ async def sensor_only() -> AsyncGenerator[None]:
 
 
 @pytest.mark.usefixtures("entity_registry_enabled_by_default")
+@pytest.mark.freeze_time("2026-05-03T18:20:54+00:00")
 async def test_sensors(
     hass: HomeAssistant,
     config_entry: MockConfigEntry,