Skip to content

Commit 99c2b53

Browse files
author
NOisi-X
committed
fix: enable tick option for freeze_time in A01 device test
1 parent 8dd5c3b commit 99c2b53

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

roborock/devices/rpc/a01_channel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ async def send_decoded_command(
3030
mqtt_channel: MqttChannel,
3131
params: dict[RoborockDyadDataProtocol, Any],
3232
value_encoder: Callable[[Any], Any] | None = None,
33+
qos: int = 0,
3334
) -> dict[RoborockDyadDataProtocol, Any]: ...
3435

3536

@@ -38,6 +39,7 @@ async def send_decoded_command(
3839
mqtt_channel: MqttChannel,
3940
params: dict[RoborockZeoProtocol, Any],
4041
value_encoder: Callable[[Any], Any] | None = None,
42+
qos: int = 0,
4143
) -> dict[RoborockZeoProtocol, Any]: ...
4244

4345

tests/e2e/test_device_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
import pytest
1717
import syrupy
1818
from Crypto.Cipher import AES
19-
from freezegun import freeze_time
2019
from Crypto.Util.Padding import pad
20+
from freezegun import freeze_time
2121

2222
from roborock.data.b01_q7 import WorkStatusMapping
2323
from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP
@@ -513,7 +513,7 @@ async def test_q7_device(
513513
)
514514
],
515515
)
516-
@freeze_time("2025-01-20T12:00:00")
516+
@freeze_time("2025-01-20T12:00:00", tick=True)
517517
async def test_a01_device(
518518
mock_rest: Any,
519519
push_mqtt_response: Callable[[bytes], None],

0 commit comments

Comments
 (0)