chore: fix typing for encode_mqtt_payload

Author: allenporter

tests/protocols/test_a01_protocol.py

@@ -18,9 +18,7 @@
 def test_encode_mqtt_payload_basic():
     """Test basic MQTT payload encoding."""
     # Test data with proper protocol keys
-    data: dict[RoborockDyadDataProtocol | RoborockZeoProtocol, Any] = {
-        RoborockDyadDataProtocol.START: {"test": "data", "number": 42}
-    }
+    data: dict[RoborockDyadDataProtocol, Any] = {RoborockDyadDataProtocol.START: {"test": "data", "number": 42}}
 
     result = encode_mqtt_payload(data)
 
@@ -39,7 +37,7 @@ def test_encode_mqtt_payload_basic():
 
 def test_encode_mqtt_payload_empty_data():
     """Test encoding with empty data."""
-    data: dict[RoborockDyadDataProtocol | RoborockZeoProtocol, Any] = {}
+    data: dict[RoborockZeoProtocol, Any] = {}
 
     result = encode_mqtt_payload(data)
 
@@ -54,14 +52,14 @@ def test_encode_mqtt_payload_empty_data():
 
 def test_encode_mqtt_payload_complex_data():
     """Test encoding with complex nested data."""
-    data: dict[RoborockDyadDataProtocol | RoborockZeoProtocol, Any] = {
+    data: dict[RoborockDyadDataProtocol, Any] = {
         RoborockDyadDataProtocol.STATUS: {
             "nested": {"deep": {"value": 123}},
             "list": [1, 2, 3, "test"],
             "boolean": True,
             "null": None,
         },
-        RoborockZeoProtocol.MODE: "simple_value",
+        RoborockDyadDataProtocol.PRODUCT_INFO: "simple_value",
     }
 
     result = encode_mqtt_payload(data)
@@ -80,7 +78,7 @@ def test_encode_mqtt_payload_complex_data():
             "boolean": True,
             "null": None,
         },
-        204: "simple_value",
+        10005: "simple_value",
     }