Skip to content

Commit b901fb8

Browse files
authored
q10: accept customized clean mode code 4 (#838)
* fix(q10): accept customized clean mode code 4 * fix(q10): adjust CUSTOMIZED mode comment formatting --------- Co-authored-by: Ludovic BOUÉ <938089+lboue@users.noreply.github.com>
1 parent 3772670 commit b901fb8

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

roborock/data/b01_q10/b01_q10_code_mappings.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ class YXCleanType(RoborockModeEnum):
162162
VAC_AND_MOP = "vac_and_mop", 1 # bothwork
163163
VACUUM = "vacuum", 2 # onlysweep
164164
MOP = "mop", 3 # onlymop
165+
CUSTOMIZED = "customized", 4 # custom mode
165166

166167

167168
class YXDeviceState(RoborockModeEnum):

tests/data/test_code_mappings.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,9 +97,15 @@ def test_homedata_product_unknown_category():
9797
("vac_and_mop", YXCleanType.VAC_AND_MOP),
9898
("vacuum", YXCleanType.VACUUM),
9999
("mop", YXCleanType.MOP),
100+
("customized", YXCleanType.CUSTOMIZED),
100101
],
101102
)
102103
def test_yx_clean_type_from_value_readable_values(readable_value: str, expected_clean_type: YXCleanType) -> None:
103104
"""Test YXCleanType accepts canonical readable values."""
104105
assert YXCleanType.from_value(readable_value) is expected_clean_type
105106
assert expected_clean_type.value == readable_value
107+
108+
109+
def test_yx_clean_type_from_code_customized() -> None:
110+
"""Test YXCleanType accepts custom mode code used by Q10 status updates."""
111+
assert YXCleanType.from_code(4) is YXCleanType.CUSTOMIZED

0 commit comments

Comments
 (0)