|
3 | 3 | import logging |
4 | 4 | import re |
5 | 5 | from asyncio import Protocol |
6 | | -from collections.abc import Callable, Generator |
| 6 | +from collections.abc import AsyncGenerator, Callable, Generator |
7 | 7 | from queue import Queue |
8 | 8 | from typing import Any |
9 | 9 | from unittest.mock import Mock, patch |
@@ -138,7 +138,7 @@ def handle_select(rlist: list, wlist: list, *args: Any) -> list: |
138 | 138 |
|
139 | 139 |
|
140 | 140 | @pytest.fixture(name="mqtt_client") |
141 | | -async def mqtt_client(mock_create_connection: None, mock_select: None) -> Generator[RoborockMqttClientV1, None, None]: |
| 141 | +async def mqtt_client(mock_create_connection: None, mock_select: None) -> AsyncGenerator[RoborockMqttClientV1, None]: |
142 | 142 | user_data = UserData.from_dict(USER_DATA) |
143 | 143 | home_data = HomeData.from_dict(HOME_DATA_RAW) |
144 | 144 | device_info = DeviceData( |
@@ -232,7 +232,7 @@ def handle_write(data: bytes) -> None: |
232 | 232 |
|
233 | 233 |
|
234 | 234 | @pytest.fixture(name="local_client") |
235 | | -async def local_client_fixture(mock_create_local_connection: None) -> Generator[RoborockLocalClientV1, None, None]: |
| 235 | +async def local_client_fixture(mock_create_local_connection: None) -> AsyncGenerator[RoborockLocalClientV1, None]: |
236 | 236 | home_data = HomeData.from_dict(HOME_DATA_RAW) |
237 | 237 | device_info = DeviceData( |
238 | 238 | device=home_data.devices[0], |
|
0 commit comments