From 6279979d506f1c55f7614e333a79c7c226dc8563 Mon Sep 17 00:00:00 2001
From: YogevBokobza <yogevbokobza12@gmail.com>
Date: Fri, 15 Nov 2024 21:03:20 +0200
Subject: [PATCH] Switcher add current current temperature sensor (#130653)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
---
 .../components/switcher_kis/icons.json          |  3 +++
 homeassistant/components/switcher_kis/sensor.py | 12 ++++++++++++
 .../components/switcher_kis/strings.json        |  3 +++
 tests/components/switcher_kis/consts.py         |  6 ++++++
 tests/components/switcher_kis/test_sensor.py    | 17 ++++++++++++++---
 5 files changed, 38 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/switcher_kis/icons.json b/homeassistant/components/switcher_kis/icons.json
index 6ca8e0e8351..bd770d3e656 100644
--- a/homeassistant/components/switcher_kis/icons.json
+++ b/homeassistant/components/switcher_kis/icons.json
@@ -20,6 +20,9 @@
       },
       "auto_shutdown": {
         "default": "mdi:progress-clock"
+      },
+      "temperature": {
+        "default": "mdi:thermometer"
       }
     }
   },
diff --git a/homeassistant/components/switcher_kis/sensor.py b/homeassistant/components/switcher_kis/sensor.py
index 9ff3d6dfaae..0ed60e5a721 100644
--- a/homeassistant/components/switcher_kis/sensor.py
+++ b/homeassistant/components/switcher_kis/sensor.py
@@ -46,9 +46,16 @@ TIME_SENSORS: list[SensorEntityDescription] = [
         entity_registry_enabled_default=False,
     ),
 ]
+TEMPERATURE_SENSORS: list[SensorEntityDescription] = [
+    SensorEntityDescription(
+        key="temperature",
+        translation_key="temperature",
+    ),
+]
 
 POWER_PLUG_SENSORS = POWER_SENSORS
 WATER_HEATER_SENSORS = [*POWER_SENSORS, *TIME_SENSORS]
+THERMOSTAT_SENSORS = TEMPERATURE_SENSORS
 
 
 async def async_setup_entry(
@@ -71,6 +78,11 @@ async def async_setup_entry(
                 SwitcherSensorEntity(coordinator, description)
                 for description in WATER_HEATER_SENSORS
             )
+        elif coordinator.data.device_type.category == DeviceCategory.THERMOSTAT:
+            async_add_entities(
+                SwitcherSensorEntity(coordinator, description)
+                for description in THERMOSTAT_SENSORS
+            )
 
     config_entry.async_on_unload(
         async_dispatcher_connect(hass, SIGNAL_DEVICE_ADD, async_add_sensors)
diff --git a/homeassistant/components/switcher_kis/strings.json b/homeassistant/components/switcher_kis/strings.json
index 798a43c981c..844cbb4ca98 100644
--- a/homeassistant/components/switcher_kis/strings.json
+++ b/homeassistant/components/switcher_kis/strings.json
@@ -59,6 +59,9 @@
       },
       "auto_shutdown": {
         "name": "Auto shutdown"
+      },
+      "temperature": {
+        "name": "Current temperature"
       }
     }
   },
diff --git a/tests/components/switcher_kis/consts.py b/tests/components/switcher_kis/consts.py
index fe77ee0236b..e9d96673e24 100644
--- a/tests/components/switcher_kis/consts.py
+++ b/tests/components/switcher_kis/consts.py
@@ -219,3 +219,9 @@ DUMMY_TRIPLE_LIGHT_DEVICE = SwitcherLight(
 )
 
 DUMMY_SWITCHER_DEVICES = [DUMMY_PLUG_DEVICE, DUMMY_WATER_HEATER_DEVICE]
+
+DUMMY_SWITCHER_SENSORS_DEVICES = [
+    DUMMY_PLUG_DEVICE,
+    DUMMY_WATER_HEATER_DEVICE,
+    DUMMY_THERMOSTAT_DEVICE,
+]
diff --git a/tests/components/switcher_kis/test_sensor.py b/tests/components/switcher_kis/test_sensor.py
index 8ccc33f2d37..f99d91bd9a3 100644
--- a/tests/components/switcher_kis/test_sensor.py
+++ b/tests/components/switcher_kis/test_sensor.py
@@ -7,7 +7,12 @@ from homeassistant.helpers import entity_registry as er
 from homeassistant.util import slugify
 
 from . import init_integration
-from .consts import DUMMY_PLUG_DEVICE, DUMMY_SWITCHER_DEVICES, DUMMY_WATER_HEATER_DEVICE
+from .consts import (
+    DUMMY_PLUG_DEVICE,
+    DUMMY_SWITCHER_SENSORS_DEVICES,
+    DUMMY_THERMOSTAT_DEVICE,
+    DUMMY_WATER_HEATER_DEVICE,
+)
 
 DEVICE_SENSORS_TUPLE = (
     (
@@ -25,17 +30,23 @@ DEVICE_SENSORS_TUPLE = (
             ("remaining_time", "remaining_time"),
         ],
     ),
+    (
+        DUMMY_THERMOSTAT_DEVICE,
+        [
+            ("current_temperature", "temperature"),
+        ],
+    ),
 )
 
 
-@pytest.mark.parametrize("mock_bridge", [DUMMY_SWITCHER_DEVICES], indirect=True)
+@pytest.mark.parametrize("mock_bridge", [DUMMY_SWITCHER_SENSORS_DEVICES], indirect=True)
 async def test_sensor_platform(hass: HomeAssistant, mock_bridge) -> None:
     """Test sensor platform."""
     entry = await init_integration(hass)
     assert mock_bridge
 
     assert mock_bridge.is_running is True
-    assert len(entry.runtime_data) == 2
+    assert len(entry.runtime_data) == 3
 
     for device, sensors in DEVICE_SENSORS_TUPLE:
         for sensor, field in sensors: