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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
"""Tests for the DustCollectionModeTrait class."""
from unittest.mock import AsyncMock, call
import pytest
from roborock.data import RoborockDockDustCollectionModeCode, RoborockDockTypeCode
from roborock.devices.device import RoborockDevice
from roborock.devices.traits.v1.dust_collection_mode import DustCollectionModeTrait
from roborock.roborock_typing import RoborockCommand
DUST_COLLECTION_MODE_DATA = [{"mode": 2}]
@pytest.fixture(name="dust_collection_mode")
def dust_collection_mode_trait(
device: RoborockDevice,
discover_features_fixture: None,
) -> DustCollectionModeTrait | None:
"""Create a DustCollectionModeTrait instance with mocked dependencies."""
assert device.v1_properties
return device.v1_properties.dust_collection_mode
@pytest.mark.parametrize(
("dock_type_code"),
[
(RoborockDockTypeCode.s7_max_ultra_dock),
(RoborockDockTypeCode.s8_dock),
(RoborockDockTypeCode.p10_dock),
(RoborockDockTypeCode.qrevo_s_dock),
],
)
async def test_dust_collection_mode_available(
dust_collection_mode: DustCollectionModeTrait | None,
mock_rpc_channel: AsyncMock,
dock_type_code: RoborockDockTypeCode,
) -> None:
"""Test successfully refreshing the dust collection mode."""
assert dust_collection_mode is not None
mock_rpc_channel.send_command.side_effect = [
DUST_COLLECTION_MODE_DATA,
]
await dust_collection_mode.refresh()
mock_rpc_channel.send_command.assert_has_calls(
[
call(RoborockCommand.GET_DUST_COLLECTION_MODE),
]
)
assert dust_collection_mode.mode == RoborockDockDustCollectionModeCode.balanced
@pytest.mark.parametrize(("dock_type_code"), [(RoborockDockTypeCode.no_dock)])
async def test_unsupported_dust_collection_mode(
dust_collection_mode: DustCollectionModeTrait | None,
dock_type_code: RoborockDockTypeCode,
) -> None:
"""Test that the trait is not available for unsupported dock types."""
assert dust_collection_mode is None
|
