diff --git a/roborock/cli.py b/roborock/cli.py index 3719a1b6..9127c5a4 100644 --- a/roborock/cli.py +++ b/roborock/cli.py @@ -156,7 +156,7 @@ async def _discover(ctx): if not cache_data: raise Exception("You need to login first") client = RoborockApiClient(cache_data.email) - home_data = await client.get_home_data(cache_data.user_data) + home_data = await client.get_home_data_v3(cache_data.user_data) cache_data.home_data = home_data context.update(cache_data) click.echo(f"Discovered devices {', '.join([device.name for device in home_data.get_all_devices()])}") diff --git a/roborock/devices/device_manager.py b/roborock/devices/device_manager.py index b9276340..f7217d01 100644 --- a/roborock/devices/device_manager.py +++ b/roborock/devices/device_manager.py @@ -113,7 +113,7 @@ def create_home_data_api(email: str, user_data: UserData) -> HomeDataApi: client = RoborockApiClient(email) async def home_data_api() -> HomeData: - return await client.get_home_data(user_data) + return await client.get_home_data_v3(user_data) return home_data_api diff --git a/tests/devices/test_device_manager.py b/tests/devices/test_device_manager.py index d549bc16..d33ade93 100644 --- a/tests/devices/test_device_manager.py +++ b/tests/devices/test_device_manager.py @@ -92,7 +92,7 @@ async def home_data_api_exception() -> HomeData: async def test_create_home_data_api_exception() -> None: """Test that exceptions from the home data API are propagated through the wrapper.""" - with patch("roborock.devices.device_manager.RoborockApiClient.get_home_data") as mock_get_home_data: + with patch("roborock.devices.device_manager.RoborockApiClient.get_home_data_v3") as mock_get_home_data: mock_get_home_data.side_effect = RoborockException("Test exception") api = create_home_data_api(USER_DATA, mock_get_home_data)