Fan tests: update for FanEntityFeature changes

Issue #2164

tests/devices/test_anko_fan.py

@@ -34,7 +34,9 @@ class TestAnkoFan(SwitchableTests, BasicNumberTests, TuyaDeviceTestCase):
             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):

tests/devices/test_arlec_fan.py

@@ -44,6 +44,8 @@ class TestArlecFan(SwitchableTests, BasicSelectTests, TuyaDeviceTestCase):
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

tests/devices/test_arlec_fan_light.py

@@ -50,6 +50,8 @@ class TestArlecFan(SwitchableTests, BasicSelectTests, TuyaDeviceTestCase):
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_ON
+                | FanEntityFeature.TURN_OFF
             ),
         )
 

tests/devices/test_aspen_adv200_fan.py

@@ -59,6 +59,8 @@ class TestAspenASP200Fan(
                 FanEntityFeature.DIRECTION
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
         self.assertEqual(

tests/devices/test_blitzwolf_bsh2_humidifier.py

@@ -57,7 +57,9 @@ class TestBlitzwolfSH2Humidifier(MultiSelectTests, TuyaDeviceTestCase):
     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):

tests/devices/test_deta_fan.py

@@ -30,7 +30,9 @@ class TestDetaFan(
     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):

tests/devices/test_eanons_humidifier.py

@@ -77,8 +77,16 @@ class TestEanonsHumidifier(
         )
 
     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

tests/devices/test_goldair_fan.py

@@ -31,6 +31,8 @@ class TestGoldairFan(BasicLightTests, SwitchableTests, TuyaDeviceTestCase):
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_ON
+                | FanEntityFeature.TURN_OFF
             ),
         )
 

tests/devices/test_himox_h05_purifier.py

@@ -86,7 +86,9 @@ class TestHimoxH05Purifier(
     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):

tests/devices/test_himox_h06_purifier.py

@@ -97,7 +97,9 @@ class TestHimoxH06Purifier(
     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):

tests/devices/test_immax_neo_light_vento.py

@@ -57,7 +57,9 @@ class TestImmaxNeoLightVento(
             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):

tests/devices/test_lexy_f501_fan.py

@@ -58,6 +58,8 @@ class TestLexyF501Fan(
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

tests/devices/test_poiema_one_purifier.py

@@ -83,7 +83,10 @@ class TestPoeimaOnePurifier(
     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):

tests/devices/test_renpho_rp_ap001s.py

@@ -83,7 +83,12 @@ class TestRenphoPurifier(
         )
 
     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(

tests/devices/test_stirling_fs140dc_fan.py

@@ -29,6 +29,8 @@ class TestStirlingFS1Fan(SwitchableTests, TuyaDeviceTestCase):
                 FanEntityFeature.OSCILLATE
                 | FanEntityFeature.PRESET_MODE
                 | FanEntityFeature.SET_SPEED
+                | FanEntityFeature.TURN_OFF
+                | FanEntityFeature.TURN_ON
             ),
         )
 

tests/devices/test_tmwf02_fan.py

@@ -32,7 +32,9 @@ class TestTMWF02Fan(BasicNumberTests, SwitchableTests, TuyaDeviceTestCase):
     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):

tests/devices/test_treatlife_ds02f.py

@@ -31,7 +31,9 @@ class TestTreatlifeFan(SwitchableTests, BasicNumberTests, TuyaDeviceTestCase):
     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):

tests/devices/test_vork_vk6067aw_purifier.py

@@ -98,7 +98,9 @@ class TestVorkVK6267AWPurifier(
     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):