Fan tests: update for FanEntityFeature changes

Issue #2164
make-all · Aug 27, 2024 · 63d7b58 · 63d7b58
1 parent 1837616
commit 63d7b58
Show file tree

Hide file tree

Showing 18 changed files with 59 additions and 13 deletions.
diff --git a/tests/devices/test_anko_fan.py b/tests/devices/test_anko_fan.py
@@ -34,7 +34,9 @@ def test_supported_features(self):
             self.subject.supported_features,
             FanEntityFeature.OSCILLATE
             | FanEntityFeature.PRESET_MODE
-            | FanEntityFeature.SET_SPEED,
+            | FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_preset_mode(self):

diff --git a/tests/devices/test_arlec_fan.py b/tests/devices/test_arlec_fan.py
@@ -44,6 +44,8 @@ def test_supported_features(self):
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

diff --git a/tests/devices/test_arlec_fan_light.py b/tests/devices/test_arlec_fan_light.py
@@ -50,6 +50,8 @@ def test_supported_features(self):
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_ON
+                | FanEntityFeature.TURN_OFF
             ),
         )
 

diff --git a/tests/devices/test_aspen_adv200_fan.py b/tests/devices/test_aspen_adv200_fan.py
@@ -59,6 +59,8 @@ def test_supported_features(self):
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
         self.assertEqual(

diff --git a/tests/devices/test_blitzwolf_bsh2_humidifier.py b/tests/devices/test_blitzwolf_bsh2_humidifier.py
@@ -57,7 +57,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.SET_SPEED,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_deta_fan.py b/tests/devices/test_deta_fan.py
@@ -30,7 +30,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.SET_SPEED,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_eanons_humidifier.py b/tests/devices/test_eanons_humidifier.py
@@ -77,8 +77,16 @@ def setUp(self):
         )
 
     def test_supported_features(self):
-        self.assertEqual(self.subject.supported_features, HumidifierEntityFeature.MODES)
-        self.assertEqual(self.fan.supported_features, FanEntityFeature.SET_SPEED)
+        self.assertEqual(
+            self.subject.supported_features,
+            HumidifierEntityFeature.MODES,
+        )
+        self.assertEqual(
+            self.fan.supported_features,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
+        )
 
     def test_current_humidity(self):
         self.dps[CURRENTHUMID_DPS] = 75

diff --git a/tests/devices/test_goldair_fan.py b/tests/devices/test_goldair_fan.py
@@ -31,6 +31,8 @@ def test_supported_features(self):
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_ON
+                | FanEntityFeature.TURN_OFF
             ),
         )
 

diff --git a/tests/devices/test_himox_h05_purifier.py b/tests/devices/test_himox_h05_purifier.py
@@ -86,7 +86,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.PRESET_MODE,
+            FanEntityFeature.PRESET_MODE
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_preset_modes(self):

diff --git a/tests/devices/test_himox_h06_purifier.py b/tests/devices/test_himox_h06_purifier.py
@@ -97,7 +97,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.SET_SPEED,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_immax_neo_light_vento.py b/tests/devices/test_immax_neo_light_vento.py
@@ -57,7 +57,9 @@ def test_supported_features(self):
             self.fan.supported_features,
             FanEntityFeature.DIRECTION
             | FanEntityFeature.PRESET_MODE
-            | FanEntityFeature.SET_SPEED,
+            | FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
         )
 
     def test_preset_modes(self):

diff --git a/tests/devices/test_lexy_f501_fan.py b/tests/devices/test_lexy_f501_fan.py
@@ -58,6 +58,8 @@ def test_supported_features(self):
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

diff --git a/tests/devices/test_poiema_one_purifier.py b/tests/devices/test_poiema_one_purifier.py
@@ -83,7 +83,10 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.PRESET_MODE | FanEntityFeature.SET_SPEED,
+            FanEntityFeature.PRESET_MODE
+            | FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_renpho_rp_ap001s.py b/tests/devices/test_renpho_rp_ap001s.py
@@ -83,7 +83,12 @@ def setUp(self):
         )
 
     def test_supported_features(self):
-        self.assertEqual(self.subject.supported_features, FanEntityFeature.PRESET_MODE)
+        self.assertEqual(
+            self.subject.supported_features,
+            FanEntityFeature.PRESET_MODE
+            | FanEntityFeature.TURN_OFF
+            | FanEntityFeature.TURN_ON,
+        )
 
     def test_preset_modes(self):
         self.assertCountEqual(

diff --git a/tests/devices/test_stirling_fs140dc_fan.py b/tests/devices/test_stirling_fs140dc_fan.py
@@ -29,6 +29,8 @@ def test_supported_features(self):
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

diff --git a/tests/devices/test_tmwf02_fan.py b/tests/devices/test_tmwf02_fan.py
@@ -32,7 +32,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.SET_SPEED,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_treatlife_ds02f.py b/tests/devices/test_treatlife_ds02f.py
@@ -31,7 +31,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.SET_SPEED,
+            FanEntityFeature.SET_SPEED
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_speed(self):

diff --git a/tests/devices/test_vork_vk6067aw_purifier.py b/tests/devices/test_vork_vk6067aw_purifier.py
@@ -98,7 +98,9 @@ def setUp(self):
     def test_supported_features(self):
         self.assertEqual(
             self.subject.supported_features,
-            FanEntityFeature.PRESET_MODE,
+            FanEntityFeature.PRESET_MODE
+            | FanEntityFeature.TURN_ON
+            | FanEntityFeature.TURN_OFF,
         )
 
     def test_preset_modes(self):