From 908b7ca9c29908fd79c67fd621f60650d97ad9b9 Mon Sep 17 00:00:00 2001
From: Teemu R <tpr@iki.fi>
Date: Wed, 1 Dec 2021 08:08:59 +0100
Subject: [PATCH] Bump xiaomi_miio dependency (#60650)

---
 homeassistant/components/xiaomi_miio/__init__.py   |  6 +++---
 homeassistant/components/xiaomi_miio/manifest.json |  2 +-
 requirements_all.txt                               |  2 +-
 requirements_test_all.txt                          |  2 +-
 tests/components/xiaomi_miio/test_vacuum.py        | 12 +++++++++---
 5 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/homeassistant/components/xiaomi_miio/__init__.py b/homeassistant/components/xiaomi_miio/__init__.py
index 9d983d32252..62935f5e38b 100644
--- a/homeassistant/components/xiaomi_miio/__init__.py
+++ b/homeassistant/components/xiaomi_miio/__init__.py
@@ -26,8 +26,8 @@ from miio import (
     FanP10,
     FanP11,
     FanZA5,
+    RoborockVacuum,
     Timer,
-    Vacuum,
     VacuumStatus,
 )
 from miio.gateway.gateway import GatewayException
@@ -212,7 +212,7 @@ class VacuumCoordinatorDataAttributes:
     fan_speeds_reverse: str = "fan_speeds_reverse"
 
 
-def _async_update_data_vacuum(hass, device: Vacuum):
+def _async_update_data_vacuum(hass, device: RoborockVacuum):
     def update() -> VacuumCoordinatorData:
         timer = []
 
@@ -313,7 +313,7 @@ async def async_create_miio_device_and_coordinator(
         or model.startswith(ROBOROCK_GENERIC)
         or model.startswith(ROCKROBO_GENERIC)
     ):
-        device = Vacuum(host, token)
+        device = RoborockVacuum(host, token)
         update_method = _async_update_data_vacuum
         coordinator_class = DataUpdateCoordinator[VacuumCoordinatorData]
     # Pedestal fans
diff --git a/homeassistant/components/xiaomi_miio/manifest.json b/homeassistant/components/xiaomi_miio/manifest.json
index 37c6b8f8a09..0556b6f3ea5 100644
--- a/homeassistant/components/xiaomi_miio/manifest.json
+++ b/homeassistant/components/xiaomi_miio/manifest.json
@@ -3,7 +3,7 @@
   "name": "Xiaomi Miio",
   "config_flow": true,
   "documentation": "https://www.home-assistant.io/integrations/xiaomi_miio",
-  "requirements": ["construct==2.10.56", "micloud==0.4", "python-miio==0.5.8"],
+  "requirements": ["construct==2.10.56", "micloud==0.4", "python-miio==0.5.9"],
   "codeowners": ["@rytilahti", "@syssi", "@starkillerOG", "@bieniu"],
   "zeroconf": ["_miio._udp.local."],
   "iot_class": "local_polling"
diff --git a/requirements_all.txt b/requirements_all.txt
index d77c0a04207..86a88c3b915 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -1907,7 +1907,7 @@ python-kasa==0.4.0
 # python-lirc==1.2.3
 
 # homeassistant.components.xiaomi_miio
-python-miio==0.5.8
+python-miio==0.5.9
 
 # homeassistant.components.mpd
 python-mpd2==3.0.4
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 1f2037ba864..00b958e67a5 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -1145,7 +1145,7 @@ python-juicenet==1.0.2
 python-kasa==0.4.0
 
 # homeassistant.components.xiaomi_miio
-python-miio==0.5.8
+python-miio==0.5.9
 
 # homeassistant.components.nest
 python-nest==4.1.0
diff --git a/tests/components/xiaomi_miio/test_vacuum.py b/tests/components/xiaomi_miio/test_vacuum.py
index f2fef4bba4b..6c8b808eb94 100644
--- a/tests/components/xiaomi_miio/test_vacuum.py
+++ b/tests/components/xiaomi_miio/test_vacuum.py
@@ -115,7 +115,9 @@ def mirobo_is_got_error_fixture():
 
     mock_vacuum.timer.return_value = [mock_timer_1, mock_timer_2]
 
-    with patch("homeassistant.components.xiaomi_miio.Vacuum") as mock_vacuum_cls:
+    with patch(
+        "homeassistant.components.xiaomi_miio.RoborockVacuum"
+    ) as mock_vacuum_cls:
         mock_vacuum_cls.return_value = mock_vacuum
         yield mock_vacuum
 
@@ -149,7 +151,9 @@ def mirobo_old_speeds_fixture(request):
         2020, 4, 1, 13, 21, 10, tzinfo=dt_util.UTC
     )
 
-    with patch("homeassistant.components.xiaomi_miio.Vacuum") as mock_vacuum_cls:
+    with patch(
+        "homeassistant.components.xiaomi_miio.RoborockVacuum"
+    ) as mock_vacuum_cls:
         mock_vacuum_cls.return_value = mock_vacuum
         yield mock_vacuum
 
@@ -209,7 +213,9 @@ def mirobo_is_on_fixture():
 
     mock_vacuum.timer.return_value = [mock_timer_1, mock_timer_2]
 
-    with patch("homeassistant.components.xiaomi_miio.Vacuum") as mock_vacuum_cls:
+    with patch(
+        "homeassistant.components.xiaomi_miio.RoborockVacuum"
+    ) as mock_vacuum_cls:
         mock_vacuum_cls.return_value = mock_vacuum
         yield mock_vacuum