Selaa lähdekoodia

RGB lights: update tests for scene/music renaming.

Jason Rumney 3 vuotta sitten
vanhempi
commit
28611df8bf

+ 4 - 6
tests/devices/test_digoo_dgsp01_dual_nightlight_switch.py

@@ -2,8 +2,6 @@
 from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.components.light import (
     ColorMode,
-    EFFECT_COLORLOOP,
-    EFFECT_RANDOM,
     LightEntityFeature,
 )
 from ..const import DIGOO_DGSP01_SOCKET_PAYLOAD
@@ -78,8 +76,8 @@ class TestDigooNightlightSwitch(BasicSwitchTests, TuyaDeviceTestCase):
         self.assertCountEqual(
             self.light.effect_list,
             [
-                EFFECT_COLORLOOP,
-                EFFECT_RANDOM,
+                "Scene",
+                "Music",
                 "Scene 1",
                 "Scene 2",
                 "Scene 3",
@@ -89,9 +87,9 @@ class TestDigooNightlightSwitch(BasicSwitchTests, TuyaDeviceTestCase):
 
     def test_light_effect(self):
         self.dps[COLORMODE_DPS] = "scene"
-        self.assertEqual(self.light.effect, EFFECT_COLORLOOP)
+        self.assertEqual(self.light.effect, "Scene")
         self.dps[COLORMODE_DPS] = "music"
-        self.assertEqual(self.light.effect, EFFECT_RANDOM)
+        self.assertEqual(self.light.effect, "Music")
         self.dps[COLORMODE_DPS] = "scene_1"
         self.assertEqual(self.light.effect, "Scene 1")
         self.dps[COLORMODE_DPS] = "scene_2"

+ 2 - 3
tests/devices/test_moes_rgb_socket.py

@@ -1,7 +1,6 @@
 """Tests for the MoesHouse RGBW smart socket."""
 from homeassistant.components.light import (
     ColorMode,
-    EFFECT_COLORLOOP,
     LightEntityFeature,
 )
 from homeassistant.components.sensor import SensorDeviceClass
@@ -137,7 +136,7 @@ class TestMoesRGBWSocket(
         self.assertCountEqual(
             self.light.effect_list,
             [
-                EFFECT_COLORLOOP,
+                "Scene",
                 "Scene 1",
                 "Scene 2",
                 "Scene 3",
@@ -147,7 +146,7 @@ class TestMoesRGBWSocket(
 
     def test_light_effect(self):
         self.dps[MODE_DPS] = "scene"
-        self.assertEqual(self.light.effect, EFFECT_COLORLOOP)
+        self.assertEqual(self.light.effect, "Scene")
         self.dps[MODE_DPS] = "scene_1"
         self.assertEqual(self.light.effect, "Scene 1")
         self.dps[MODE_DPS] = "scene_2"

+ 3 - 5
tests/devices/test_rgbcw_lightbulb.py

@@ -1,8 +1,6 @@
 from homeassistant.components.light import (
     ColorMode,
     LightEntityFeature,
-    EFFECT_COLORLOOP,
-    EFFECT_RANDOM,
 )
 from homeassistant.const import UnitOfTime
 
@@ -77,14 +75,14 @@ class TestRGBCWLightbulb(BasicNumberTests, TuyaDeviceTestCase):
     def test_effect_list(self):
         self.assertCountEqual(
             self.subject.effect_list,
-            [EFFECT_COLORLOOP, EFFECT_RANDOM],
+            ["Scene", "Music"],
         )
 
     def test_effect(self):
         self.dps[MODE_DPS] = "scene"
-        self.assertEqual(self.subject.effect, EFFECT_COLORLOOP)
+        self.assertEqual(self.subject.effect, "Scene")
         self.dps[MODE_DPS] = "music"
-        self.assertEqual(self.subject.effect, EFFECT_RANDOM)
+        self.assertEqual(self.subject.effect, "Music")
         self.dps[MODE_DPS] = "white"
         self.assertIsNone(self.subject.effect)
         self.dps[MODE_DPS] = "colour"