chore: fix test fixture after merging

allenporter · allenporter · commit ae797e34e842 · 2025-01-21T07:24:37.000-08:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -240,7 +240,7 @@ async def local_client_fixture(mock_create_local_connection: None) -> AsyncGener
         model=home_data.products[0].model,
         host=TEST_LOCAL_API_HOST,
     )
-    yield RoborockLocalClientV1(device_info, queue_timeout=QUEUE_TIMEOUT)
+    client = RoborockLocalClientV1(device_info, queue_timeout=QUEUE_TIMEOUT)
     try:
         yield client
     finally:
diff --git a/tests/test_a01_api.py b/tests/test_a01_api.py
@@ -52,7 +52,9 @@ async def a01_mqtt_client_fixture(
         device=home_data.devices[0],
         model=home_data.products[0].model,
     )
-    client = RoborockMqttClientA01(user_data, device_info, RoborockCategory.WASHING_MACHINE, queue_timeout=QUEUE_TIMEOUT)
+    client = RoborockMqttClientA01(
+        user_data, device_info, RoborockCategory.WASHING_MACHINE, queue_timeout=QUEUE_TIMEOUT
+    )
     try:
         yield client
     finally:

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ async def local_client_fixture(mock_create_local_connection: None) -> AsyncGener`
`240`	`240`	`model=home_data.products[0].model,`
`241`	`241`	`host=TEST_LOCAL_API_HOST,`
`242`	`242`	`)`
`243`		`- yield RoborockLocalClientV1(device_info, queue_timeout=QUEUE_TIMEOUT)`
	`243`	`+ client = RoborockLocalClientV1(device_info, queue_timeout=QUEUE_TIMEOUT)`
`244`	`244`	`try:`
`245`	`245`	`yield client`
`246`	`246`	`finally:`