chore: linting

Author: humbertogontijo

tests/conftest.py

@@ -2,14 +2,15 @@
 
 from roborock import HomeData, UserData
 from roborock.cloud_api import RoborockMqttClient
+from roborock.containers import RoborockDeviceInfo
 from tests.mock_data import HOME_DATA_RAW, USER_DATA
 
 
 @pytest.fixture(name="mqtt_client")
 def mqtt_client():
     user_data = UserData.from_dict(USER_DATA)
     home_data = HomeData.from_dict(HOME_DATA_RAW)
-    device_map = {home_data.devices[0].duid: home_data.devices[0].local_key}
-    client = RoborockMqttClient(user_data, device_map)
+    device_info = RoborockDeviceInfo(device=home_data.devices[0])
+    client = RoborockMqttClient(user_data, device_info)
     yield client
     # Clean up any resources after the test