fix: fix light supported_features return value

muhlba91 · Dec 12, 2023 · f3af412 · f3af412
1 parent 891a124
commit f3af412
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/custom_components/hella_onyx/sensors/light.py b/custom_components/hella_onyx/sensors/light.py
@@ -7,6 +7,7 @@
 from homeassistant.components.light import (
     LightEntity,
     ColorMode,
+    LightEntityFeature,
     ATTR_BRIGHTNESS,
 )
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@@ -55,7 +56,7 @@ def unique_id(self) -> str:
     @property
     def supported_features(self):
         """Flag supported features."""
-        return []
+        return LightEntityFeature(0)
 
     @property
     def color_mode(self) -> ColorMode | str | None:

diff --git a/tests/sensors/test_light.py b/tests/sensors/test_light.py
@@ -63,7 +63,7 @@ def test_unique_id(self, entity):
         assert entity.unique_id == "uuid/Light"
 
     def test_supported_features(self, entity):
-        assert len(entity.supported_features) == 0
+        assert entity.supported_features == 0
 
     def test_color_mode(self, api, entity, device):
         device.device_type = DeviceType.BASIC_LIGHT