diff --git a/tests/test_VitoairFs300E.py b/tests/test_VitoairFs300E.py index 58648a5f..8ec5ebdb 100644 --- a/tests/test_VitoairFs300E.py +++ b/tests/test_VitoairFs300E.py @@ -47,3 +47,13 @@ def test_ventilationState(self): self.assertEqual(self.device.getVentilationDemand(), "unknown") self.assertEqual(self.device.getVentilationLevel(), "levelFour") self.assertEqual(self.device.getVentilationReason(), "sensorOverride") + + def test_ventilationQuickmode(self): + self.assertEqual(self.device.getVentilationQuickmode("forcedLevelFour"), False) + self.assertEqual(self.device.getVentilationQuickmode("silent"), False) + + def test_ventilationQuickmodes(self): + self.assertEqual(self.device.getVentilationQuickmodes(), [ + "forcedLevelFour", + "silent", + ]) diff --git a/tests/test_vitocal-with-vitovent.py b/tests/test_vitocal-with-vitovent.py index 0223bc6d..a392c8ca 100644 --- a/tests/test_vitocal-with-vitovent.py +++ b/tests/test_vitocal-with-vitovent.py @@ -17,3 +17,15 @@ def test_isSolarThermalDevice(self): def test_isVentilationDevice(self): self.assertEqual(self.device.isVentilationDevice(), True) + + def test_ventilationQuickmode(self): + self.assertEqual(self.device.getVentilationQuickmode("comfort"), False) + self.assertEqual(self.device.getVentilationQuickmode("eco"), False) + self.assertEqual(self.device.getVentilationQuickmode("holiday"), False) + + def test_ventilationQuickmodes(self): + self.assertEqual(self.device.getVentilationQuickmodes(), [ + "comfort", + "eco", + "holiday", + ])