fix: add light device sensor

Author: muhlba91

custom_components/hella_onyx/sensor.py

@@ -47,6 +47,7 @@ async def async_setup_entry(
             lambda item: item[1].device_type is not None
             and (
                 item[1].device_type.is_shutter()
+                or item[1].device_type.is_light()
                 or item[1].device_type == DeviceType.WEATHER
             ),
             api.devices.items(),

tests/test_sensor.py

@@ -5,6 +5,7 @@
 import pytest
 from homeassistant.config_entries import ConfigEntry
 from onyx_client.data.device_mode import DeviceMode
+from onyx_client.device.device import Device
 from onyx_client.device.light import Light
 from onyx_client.device.weather import Weather
 from onyx_client.device.shutter import Shutter
@@ -67,22 +68,25 @@ async def test_async_setup_entry(mock_hass):
 
     await async_setup_entry(mock_hass, config_entry, async_add_entries.call)
     assert async_add_entries.called_async_add_entities
-    assert len(async_add_entries.data) == 8
+    assert len(async_add_entries.data) == 9
     assert async_add_entries.data[0]._uuid == "shutter"
-    assert async_add_entries.data[1]._uuid == "weather"
-    assert async_add_entries.data[1].unique_id == "weather/DeviceType"
+    assert async_add_entries.data[0].unique_id == "shutter/DeviceType"
+    assert async_add_entries.data[1]._uuid == "light"
+    assert async_add_entries.data[1].unique_id == "light/DeviceType"
     assert async_add_entries.data[2]._uuid == "weather"
-    assert async_add_entries.data[2].unique_id == "weather/Humidity"
+    assert async_add_entries.data[2].unique_id == "weather/DeviceType"
     assert async_add_entries.data[3]._uuid == "weather"
-    assert async_add_entries.data[3].unique_id == "weather/Temperature"
+    assert async_add_entries.data[3].unique_id == "weather/Humidity"
     assert async_add_entries.data[4]._uuid == "weather"
-    assert async_add_entries.data[4].unique_id == "weather/AirPressure"
+    assert async_add_entries.data[4].unique_id == "weather/Temperature"
     assert async_add_entries.data[5]._uuid == "weather"
-    assert async_add_entries.data[5].unique_id == "weather/WindPeak"
+    assert async_add_entries.data[5].unique_id == "weather/AirPressure"
     assert async_add_entries.data[6]._uuid == "weather"
-    assert async_add_entries.data[6].unique_id == "weather/SunBrightnessPeak"
+    assert async_add_entries.data[6].unique_id == "weather/WindPeak"
     assert async_add_entries.data[7]._uuid == "weather"
-    assert async_add_entries.data[7].unique_id == "weather/SunBrightnessSink"
+    assert async_add_entries.data[7].unique_id == "weather/SunBrightnessPeak"
+    assert async_add_entries.data[8]._uuid == "weather"
+    assert async_add_entries.data[8].unique_id == "weather/SunBrightnessSink"
 
 
 @patch("homeassistant.core.HomeAssistant")
@@ -91,11 +95,11 @@ async def test_async_setup_entry_filter_all(mock_hass):
     config_entry = ConfigEntry(1, DOMAIN, "entry", {}, "source", "POLL", {})
     api = MagicMock()
     api.devices = {
-        "light": Light(
-            "light",
+        "click": Device(
+            "click",
             "name",
-            DeviceType.BASIC_LIGHT,
-            DeviceMode(DeviceType.BASIC_LIGHT),
+            DeviceType.CLICK,
+            DeviceMode(DeviceType.CLICK),
             list(),
         )
     }