1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import math
import time
from collections.abc import Generator
from unittest.mock import patch
import pytest
from roborock.devices.traits.b01.q7 import Q7PropertiesApi
from tests.fixtures.channel_fixtures import FakeChannel
from . import B01MessageBuilder
@pytest.fixture(name="fake_channel")
def fake_channel_fixture() -> FakeChannel:
return FakeChannel()
@pytest.fixture(name="q7_api")
def q7_api_fixture(fake_channel: FakeChannel) -> Q7PropertiesApi:
return Q7PropertiesApi(fake_channel) # type: ignore[arg-type]
@pytest.fixture(name="expected_msg_id", autouse=True)
def next_message_id_fixture() -> Generator[int, None, None]:
"""Fixture to patch get_next_int to return the expected message ID.
We pick an arbitrary number, but just need it to ensure we can craft a fake
response with the message id matched to the outgoing RPC.
"""
expected_msg_id = math.floor(time.time())
# Patch get_next_int to return our expected msg_id so the channel waits for it
with patch("roborock.protocols.b01_q7_protocol.get_next_int", return_value=expected_msg_id):
yield expected_msg_id
@pytest.fixture(name="message_builder")
def message_builder_fixture(expected_msg_id: int) -> B01MessageBuilder:
builder = B01MessageBuilder()
builder.msg_id = expected_msg_id
return builder
|
