From fe0120b65a5e685b1aed06e8bd3cf10b561a710b Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 26 Apr 2022 14:31:25 +0200
Subject: [PATCH] Use light enums in zwave_js (#70791)

---
 homeassistant/components/zwave_js/light.py | 4 ++--
 tests/components/zwave_js/test_light.py    | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/zwave_js/light.py b/homeassistant/components/zwave_js/light.py
index 478268ed528..ca51bc83986 100644
--- a/homeassistant/components/zwave_js/light.py
+++ b/homeassistant/components/zwave_js/light.py
@@ -31,9 +31,9 @@ from homeassistant.components.light import (
     ATTR_RGBW_COLOR,
     ATTR_TRANSITION,
     DOMAIN as LIGHT_DOMAIN,
-    SUPPORT_TRANSITION,
     ColorMode,
     LightEntity,
+    LightEntityFeature,
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
@@ -160,7 +160,7 @@ class ZwaveLight(ZWaveBaseEntity, LightEntity):
         )
 
         if self.supports_brightness_transition or self.supports_color_transition:
-            self._attr_supported_features |= SUPPORT_TRANSITION
+            self._attr_supported_features |= LightEntityFeature.TRANSITION
 
     @callback
     def on_value_update(self) -> None:
diff --git a/tests/components/zwave_js/test_light.py b/tests/components/zwave_js/test_light.py
index 01de5a70692..d2434f01963 100644
--- a/tests/components/zwave_js/test_light.py
+++ b/tests/components/zwave_js/test_light.py
@@ -14,7 +14,7 @@ from homeassistant.components.light import (
     ATTR_SUPPORTED_COLOR_MODES,
     ATTR_TRANSITION,
     DOMAIN as LIGHT_DOMAIN,
-    SUPPORT_TRANSITION,
+    LightEntityFeature,
 )
 from homeassistant.const import (
     ATTR_ENTITY_ID,
@@ -44,7 +44,7 @@ async def test_light(hass, client, bulb_6_multi_color, integration):
     assert state.state == STATE_OFF
     assert state.attributes[ATTR_MIN_MIREDS] == 153
     assert state.attributes[ATTR_MAX_MIREDS] == 370
-    assert state.attributes[ATTR_SUPPORTED_FEATURES] == SUPPORT_TRANSITION
+    assert state.attributes[ATTR_SUPPORTED_FEATURES] == LightEntityFeature.TRANSITION
     assert state.attributes[ATTR_SUPPORTED_COLOR_MODES] == ["color_temp", "hs"]
 
     # Test turning on
@@ -478,7 +478,7 @@ async def test_rgbw_light(hass, client, zen_31, integration):
 
     assert state
     assert state.state == STATE_ON
-    assert state.attributes[ATTR_SUPPORTED_FEATURES] == SUPPORT_TRANSITION
+    assert state.attributes[ATTR_SUPPORTED_FEATURES] == LightEntityFeature.TRANSITION
 
     # Test turning on
     await hass.services.async_call(
@@ -543,7 +543,7 @@ async def test_light_none_color_value(hass, light_color_null_values, integration
 
     assert state
     assert state.state == STATE_ON
-    assert state.attributes[ATTR_SUPPORTED_FEATURES] == SUPPORT_TRANSITION
+    assert state.attributes[ATTR_SUPPORTED_FEATURES] == LightEntityFeature.TRANSITION
     assert state.attributes[ATTR_SUPPORTED_COLOR_MODES] == ["hs"]