add test cases

openviess · Dec 23, 2024 · 6d33879 · 6d33879
1 parent 5ef6c47
commit 6d33879
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git 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
@@ -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",
+        ])