11import json
2+ from collections .abc import AsyncGenerator
23from queue import Queue
34from typing import Any
45from unittest .mock import patch
5- from collections .abc import AsyncGenerator
66
77import paho .mqtt .client as mqtt
88import pytest
@@ -147,7 +147,9 @@ async def test_get_prop():
147147
148148
149149@pytest .fixture (name = "connected_mqtt_client" )
150- async def connected_mqtt_client_fixture (response_queue : Queue , mqtt_client : RoborockMqttClientV1 ) -> AsyncGenerator [RoborockMqttClientV1 , None ]:
150+ async def connected_mqtt_client_fixture (
151+ response_queue : Queue , mqtt_client : RoborockMqttClientV1
152+ ) -> AsyncGenerator [RoborockMqttClientV1 , None ]:
151153 response_queue .put (mqtt_packet .gen_connack (rc = 0 , flags = 2 ))
152154 response_queue .put (mqtt_packet .gen_suback (1 , 0 ))
153155 await mqtt_client .async_connect ()
@@ -156,9 +158,7 @@ async def connected_mqtt_client_fixture(response_queue: Queue, mqtt_client: Robo
156158 await mqtt_client .async_disconnect ()
157159
158160
159- async def test_async_connect (
160- received_requests : Queue , connected_mqtt_client : RoborockMqttClientV1
161- ) -> None :
161+ async def test_async_connect (received_requests : Queue , connected_mqtt_client : RoborockMqttClientV1 ) -> None :
162162 """Test connecting to the MQTT broker."""
163163
164164 assert connected_mqtt_client .is_connected ()
0 commit comments