test: fix capabilities removal

midea-lan · Jul 11, 2024 · 69224b5 · 69224b5
1 parent 45a3c77
commit 69224b5
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/device_test.py b/tests/device_test.py
@@ -66,17 +66,17 @@ def test_connect(self) -> None:
             patch.object(
                 self.device,
                 "authenticate",
-                side_effect=[AuthException(), None, None, None],
+                side_effect=[AuthException(), None, None],
             ),
             patch.object(
                 self.device,
                 "refresh_status",
-                side_effect=[RefreshFailed(), None, None],
+                side_effect=[RefreshFailed(), None],
             ),
             patch.object(
                 self.device,
                 "get_capabilities",
-                side_effect=[None, None],
+                side_effect=[None],
             ),
         ):
             connect_mock.side_effect = [
@@ -99,9 +99,6 @@ def test_connect(self) -> None:
             assert self.device.connect(True, True) is False
             assert self.device.available is False
 
-            assert self.device.connect(True, True) is False
-            assert self.device.available is False
-
             assert self.device.connect(True, True) is True
             assert self.device.available is True