Skip to content

Commit f1bfa39

Browse files
committed
fix: test
1 parent 0fa3456 commit f1bfa39

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313

1414
from roborock import HomeData, UserData
1515
from roborock.data import DeviceData
16+
from roborock.protocols.v1_protocol import LocalProtocolVersion
1617
from roborock.roborock_message import RoborockMessage
1718
from roborock.version_1_apis.roborock_local_client_v1 import RoborockLocalClientV1
1819
from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1
@@ -361,6 +362,7 @@ def __init__(self):
361362
self.subscribe = AsyncMock(side_effect=self._subscribe)
362363
self.connect = AsyncMock(side_effect=self._connect)
363364
self.close = MagicMock(side_effect=self._close)
365+
self.protocol_version = LocalProtocolVersion.V1
364366

365367
async def _connect(self) -> None:
366368
self._is_connected = True

0 commit comments

Comments
 (0)