feat: update supported_schema_ids to include additional RoborockMessageProtocol and RoborockDataProtocol constants

Author: allenporter

roborock/data/containers.py

@@ -264,7 +264,8 @@ def supported_schema_codes(self) -> set[str]:
     def supported_schema_ids(self) -> set[int]:
         """Return a set of schema IDs (DPS integers) that are supported by the device.
 
-        These correspond to RoborockDataProtocol or RoborockB01Protocol enum number values.
+        These correspond to RoborockMessageProtocol and RoborockDataProtocol or
+        RoborockB01Protocol enum number values (depends on the device protocol versions).
         """
         if self.schema is None:
             return set()

tests/data/test_containers.py

@@ -15,6 +15,7 @@
     _camelize,
     _decamelize,
 )
+from roborock.roborock_message import RoborockDataProtocol, RoborockMessageProtocol
 from tests.mock_data import (
     HOME_DATA_RAW,
     K_VALUE,
@@ -213,6 +214,27 @@ def test_home_data():
         "task_complete",
         "water_box_mode",
     }
+    assert product.supported_schema_ids == {
+        int(v)
+        for v in (
+            RoborockMessageProtocol.RPC_REQUEST,
+            RoborockMessageProtocol.RPC_RESPONSE,
+            RoborockDataProtocol.ERROR_CODE,
+            RoborockDataProtocol.STATE,
+            RoborockDataProtocol.BATTERY,
+            RoborockDataProtocol.FAN_POWER,
+            RoborockDataProtocol.WATER_BOX_MODE,
+            RoborockDataProtocol.MAIN_BRUSH_WORK_TIME,
+            RoborockDataProtocol.SIDE_BRUSH_WORK_TIME,
+            RoborockDataProtocol.FILTER_WORK_TIME,
+            RoborockDataProtocol.ADDITIONAL_PROPS,
+            RoborockDataProtocol.TASK_COMPLETE,
+            RoborockDataProtocol.TASK_CANCEL_LOW_POWER,
+            RoborockDataProtocol.TASK_CANCEL_IN_MOTION,
+            RoborockDataProtocol.CHARGE_STATUS,
+            RoborockDataProtocol.DRYING_STATUS,
+        )
+    }
 
     device = hd.devices[0]
     assert device.duid == "abc123"