diff --git a/roborock/device_features.py b/roborock/device_features.py index 9b03b972..17295a72 100644 --- a/roborock/device_features.py +++ b/roborock/device_features.py @@ -650,6 +650,7 @@ def get_supported_features(self) -> list[str]: RoborockDockTypeCode.p10_dock, RoborockDockTypeCode.p10_pro_dock, RoborockDockTypeCode.s8_maxv_ultra_dock, + RoborockDockTypeCode.qrevo_master_dock, RoborockDockTypeCode.qrevo_s_dock, RoborockDockTypeCode.saros_r10_dock, RoborockDockTypeCode.qrevo_curv_dock, diff --git a/tests/devices/traits/v1/test_device_features.py b/tests/devices/traits/v1/test_device_features.py index 7beb529d..fc057c10 100644 --- a/tests/devices/traits/v1/test_device_features.py +++ b/tests/devices/traits/v1/test_device_features.py @@ -3,8 +3,9 @@ import pytest from syrupy import SnapshotAssertion -from roborock.data import HomeDataDevice +from roborock.data import HomeDataDevice, RoborockDockTypeCode from roborock.data.v1.v1_containers import ConsumableField, StatusField +from roborock.device_features import is_wash_n_fill_dock from roborock.devices.device import RoborockDevice from roborock.devices.traits.v1.consumeable import ConsumableTrait from roborock.devices.traits.v1.status import StatusTrait @@ -15,6 +16,11 @@ } +def test_qrevo_master_dock_is_wash_n_fill_dock() -> None: + """Test that the Qrevo Master dock supports wash and fill features.""" + assert is_wash_n_fill_dock(RoborockDockTypeCode.qrevo_master_dock) + + @pytest.mark.parametrize( ("device_info"), V1_DEVICES.values(),