add mqtt_client fixture

Author: Lash-L

tests/conftest.py

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

tests/test_api.py

@@ -22,12 +22,9 @@ def test_can_create_mqtt_roborock():
     RoborockMqttClient(UserData(USER_DATA), device_map)
 
 
-def test_sync_connect():
-    home_data = HomeData(HOME_DATA_RAW)
-    device_map = {home_data.devices[0].duid: home_data.devices[0]}
-    client = RoborockMqttClient(UserData(USER_DATA), device_map)
+def test_sync_connect(mqtt_client):
     with patch("paho.mqtt.client.Client.connect", return_value=mqtt.MQTT_ERR_SUCCESS):
-        client.sync_connect()
+        mqtt_client.sync_connect()
 
 
 @pytest.mark.asyncio