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
64
65
66
67
68
69
70
71
72
73
74
75
|
from syrupy import SnapshotAssertion
from roborock import SHORT_MODEL_TO_ENUM
from roborock.data.code_mappings import RoborockProductNickname
from roborock.device_features import DeviceFeatures
def test_supported_features_qrevo_maxv():
"""Ensure that a QREVO MaxV has some more complicated features enabled."""
model = "roborock.vacuum.a87"
product_nickname = SHORT_MODEL_TO_ENUM.get(model.split(".")[-1])
device_features = DeviceFeatures.from_feature_flags(
new_feature_info=4499197267967999,
new_feature_info_str="508A977F7EFEFFFF",
feature_info=[111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125],
product_nickname=product_nickname,
)
assert device_features
print("\n".join(device_features.get_supported_features()))
num_true = sum(v for v in vars(device_features).values() if isinstance(v, bool))
print(num_true)
assert num_true != 0
assert device_features.is_dust_collection_setting_supported
assert device_features.is_led_status_switch_supported
assert not device_features.is_matter_supported
print(device_features)
def test_supported_features_s7():
"""Ensure that a S7 has some more basic features enabled."""
model = "roborock.vacuum.a15"
product_nickname = SHORT_MODEL_TO_ENUM.get(model.split(".")[-1])
device_features = DeviceFeatures.from_feature_flags(
new_feature_info=636084721975295,
new_feature_info_str="0000000000002000",
feature_info=[111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 122, 123, 124, 125],
product_nickname=product_nickname,
)
num_true = sum(v for v in vars(device_features).values() if isinstance(v, bool))
assert num_true != 0
assert device_features
assert device_features.is_custom_mode_supported
assert device_features.is_led_status_switch_supported
assert not device_features.is_hot_wash_towel_supported
num_true = sum(v for v in vars(device_features).values() if isinstance(v, bool))
assert num_true != 0
def test_device_feature_serialization(snapshot: SnapshotAssertion) -> None:
"""Test serialization and deserialization of DeviceFeatures."""
device_features = DeviceFeatures.from_feature_flags(
new_feature_info=636084721975295,
new_feature_info_str="0000000000002000",
feature_info=[111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 122, 123, 124, 125],
product_nickname=RoborockProductNickname.TANOSS,
)
serialized = device_features.as_dict()
deserialized = DeviceFeatures.from_dict(serialized)
assert deserialized == device_features
def test_new_feature_str_missing():
"""Ensure that DeviceFeatures missing fields can still be created."""
device_features = DeviceFeatures.from_feature_flags(
new_feature_info=0,
new_feature_info_str="",
feature_info=[],
product_nickname=None,
)
# Check arbitrary fields that are false by default
assert not device_features.is_dust_collection_setting_supported
assert not device_features.is_hot_wash_towel_supported
assert not device_features.is_show_clean_finish_reason_supported
|
