feat(zeo): add _discover_features() and supports() with FEATURE_BITS; add DP 243

fix test snapshot for FEATURE_BITS query; pass ruff lint

Author: NOisi-X

roborock/devices/traits/a01/__init__.py

@@ -25,9 +25,6 @@
 from datetime import time
 from typing import Any
 
-from roborock.exceptions import RoborockException
-from roborock.protocols.a01_protocol import decode_rpc_response
-
 from roborock.data import DyadProductInfo, DyadSndState, HomeDataProduct, RoborockCategory
 from roborock.data.dyad.dyad_code_mappings import (
     DyadBrushSpeed,
@@ -44,6 +41,7 @@
     ZeoDetergentType,
     ZeoDryingMode,
     ZeoError,
+    ZeoFeatureBits,
     ZeoMode,
     ZeoProgram,
     ZeoRinse,
@@ -56,6 +54,8 @@
 from roborock.devices.traits import Trait
 from roborock.devices.traits.common import TraitUpdateListener
 from roborock.devices.transport.mqtt_channel import MqttChannel
+from roborock.exceptions import RoborockException
+from roborock.protocols.a01_protocol import decode_rpc_response
 from roborock.roborock_message import (
     RoborockDyadDataProtocol,
     RoborockMessage,
@@ -179,10 +179,16 @@ def __init__(self, channel: MqttChannel) -> None:
         self._channel = channel
         self._dps_cache: dict[int, Any] = {}
         self._dps_unsub: Callable[[], None] | None = None
+        self._feature_bits: int = 0
 
     async def start(self) -> None:
-        """Subscribe to MQTT push (called once after device connects)."""
+        """Subscribe to MQTT push and discover device features.
+
+        Subscribes to the DPS MQTT topic, then queries FEATURE_BITS
+        (DP 237) to wake the device and cache supported capabilities.
+        """
         await self._ensure_subscribed()
+        await self._discover_features()
 
     def close(self) -> None:
         """Unsubscribe from MQTT push and release resources."""
@@ -196,6 +202,24 @@ async def _ensure_subscribed(self) -> None:
             return
         self._dps_unsub = await self._channel.subscribe(self._on_dps_message)
 
+    async def _discover_features(self) -> None:
+        """Query FEATURE_BITS to wake the device and cache capabilities.
+
+        Sending an RPC query after subscribing triggers the device to
+        start pushing its full state — equivalent to how V1's
+        ``discover_features()`` uses ``device_features.refresh()`` to
+        initiate the push cycle.
+        """
+        try:
+            result = await self.query_values([RoborockZeoProtocol.FEATURE_BITS])
+            self._feature_bits = result.get(RoborockZeoProtocol.FEATURE_BITS, 0)
+        except Exception:
+            self._feature_bits = 0
+
+    def supports(self, feature: ZeoFeatureBits) -> bool:
+        """Check whether the device supports a given feature bit."""
+        return bool(self._feature_bits & (1 << feature.value))
+
     def _on_dps_message(self, message: RoborockMessage) -> None:
         """Handle unsolicited MQTT push (protocol 102 — RPC_RESPONSE).
 

roborock/roborock_message.py

@@ -179,6 +179,9 @@ class RoborockZeoProtocol(RoborockEnum):
     SILENT_MODE_ON = 240  # rw  [independent]  use set_silent_mode() for bundled set
     SILENT_MODE_START_TIME = 241  # rw  [independent]  minute-of-day
     SILENT_MODE_END_TIME = 242  # rw  [independent]  minute-of-day
+    UNKNOWN_243 = (
+        243  # unknown, not found in plugin bundle; present in MQTT push from some devices, increments with each push
+    )
     DRY_CARE_MODE = 244  # rw  [startWith]
     SOFTENER_EXPANSION_TYPE = 245  # rw  [independent]
     SMILE_LIGHT_STATUS = 247  # rw  [independent]

tests/e2e/__snapshots__/test_device_manager.ambr

@@ -16,16 +16,30 @@
   00000000  30 5a 00 20 72 72 2f 6d 2f 69 2f 75 73 65 72 31  |0Z. rr/m/i/user1|
   00000010  32 33 2f 31 39 36 34 38 66 39 34 2f 7a 65 6f 5f  |23/19648f94/zeo_|
   00000020  64 75 69 64 00 41 30 31 00 00 23 82 00 00 23 83  |duid.A01..#...#.|
-  00000030  68 a6 a2 24 00 65 00 20 c5 de 2b f6 a9 ba 32 7e  |h..$.e. ..+...2~|
-  00000040  6b 73 82 bb d8 67 d4 db 7e cd 61 aa 8c 38 56 53  |ks...g..~.a..8VS|
-  00000050  ca 4e 15 0d b1 b7 80 a2 0f 16 58 36              |.N........X6|
+  00000030  68 a6 a2 25 00 65 00 20 c5 de 2b f6 a9 ba 32 7e  |h..%.e. ..+...2~|
+  00000040  6b 73 82 bb d8 67 d4 db 07 aa be 99 a5 42 ee 82  |ks...g.......B..|
+  00000050  9d fd 96 c6 3f ab ac 4e db 7b 92 69              |....?..N.{.i|
   [mqtt <]
   00000000  30 5e 00 20 72 72 2f 6d 2f 6f 2f 75 73 65 72 31  |0^. rr/m/o/user1|
   00000010  32 33 2f 31 39 36 34 38 66 39 34 2f 7a 65 6f 5f  |23/19648f94/zeo_|
   00000020  64 75 69 64 00 00 00 00 37 41 30 31 00 00 00 00  |duid....7A01....|
-  00000030  00 00 00 17 68 a6 a2 23 00 66 00 20 c6 d0 06 0c  |....h..#.f. ....|
+  00000030  00 00 00 17 68 a6 a2 23 00 66 00 20 fe 9c 2a 27  |....h..#.f. ..*'|
+  00000040  da c4 6b 9f 0e cf 2c 56 ba 5b e6 99 1f a1 29 78  |..k...,V.[....)x|
+  00000050  37 37 42 66 bb d5 70 0e d4 c0 a7 d1 7f ef 8b 33  |77Bf..p........3|
+  [mqtt >]
+  00000000  30 5a 00 20 72 72 2f 6d 2f 69 2f 75 73 65 72 31  |0Z. rr/m/i/user1|
+  00000010  32 33 2f 31 39 36 34 38 66 39 34 2f 7a 65 6f 5f  |23/19648f94/zeo_|
+  00000020  64 75 69 64 00 41 30 31 00 00 23 84 00 00 23 85  |duid.A01..#...#.|
+  00000030  68 a6 a2 26 00 65 00 20 00 5e b7 20 93 7b 53 a7  |h..&.e. .^. .{S.|
+  00000040  f9 47 d4 53 49 51 cd 59 13 cd 44 fa 4f 7e 12 e4  |.G.SIQ.Y..D.O~..|
+  00000050  11 46 37 83 81 32 9d 4d fd c9 4f a6              |.F7..2.M..O.|
+  [mqtt <]
+  00000000  30 5e 00 20 72 72 2f 6d 2f 6f 2f 75 73 65 72 31  |0^. rr/m/o/user1|
+  00000010  32 33 2f 31 39 36 34 38 66 39 34 2f 7a 65 6f 5f  |23/19648f94/zeo_|
+  00000020  64 75 69 64 00 00 00 00 37 41 30 31 00 00 00 00  |duid....7A01....|
+  00000030  00 00 00 17 68 a6 a2 24 00 66 00 20 c6 d0 06 0c  |....h..$.f. ....|
   00000040  04 eb 86 8c 96 8c 51 45 4f 8e 96 93 9e 3d de 35  |......QEO....=.5|
-  00000050  bb a3 92 cf 68 49 69 ba 83 25 cc 5d 77 e8 62 8a  |....hIi..%.]w.b.|
+  00000050  bb a3 92 cf 68 49 69 ba 83 25 cc 5d 43 da 0b 55  |....hIi..%.]C..U|
 # ---
 # name: test_l01_device
   [mqtt >]

tests/e2e/test_device_manager.py

@@ -527,8 +527,10 @@ async def test_a01_device(
     test_topic = TEST_TOPIC_FORMAT.format(duid="zeo_duid")
     mqtt_responses: list[bytes] = [
         *MQTT_DEFAULT_RESPONSES,
-        # ACK the Query state call sent below. id is deterministic based on deterministic_message_fixtures
-        mqtt_packet.gen_publish(test_topic, mid=2, payload=response_builder.build_a01_rpc({"203": 6})),
+        # ACK the FEATURE_BITS query sent by _discover_features()
+        mqtt_packet.gen_publish(test_topic, mid=2, payload=response_builder.build_a01_rpc({"237": 1})),
+        # ACK the Query state call sent below
+        mqtt_packet.gen_publish(test_topic, mid=3, payload=response_builder.build_a01_rpc({"203": 6})),
     ]
     for response in mqtt_responses:
         push_mqtt_response(response)