test: fixed tests after merge

midea-lan · Jul 12, 2024 · fc21ec7 · fc21ec7
1 parent deb682f
commit fc21ec7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/cli_test.py b/tests/cli_test.py
@@ -86,9 +86,12 @@ async def test_discover(self) -> None:
             ),
         ):
             mock_discover.return_value = {1: mock_device}
-            mock_cloud_instance.get_keys.return_value = {
+            mock_cloud_instance.get_cloud_keys.return_value = {
                 0: {"token": "token", "key": "key"},
             }
+            mock_cloud_instance.get_default_keys.return_value = {
+                99: {"token": "token", "key": "key"},
+            }
 
             await self.cli.discover()  # V3 device