fix: test

Author: Lash-L

roborock/version_1_apis/roborock_client_v1.py

@@ -150,7 +150,7 @@ class RoborockClientV1(RoborockClient, ABC):
     """Roborock client base class for version 1 devices."""
 
     _listeners: dict[str, ListenerModel] = {}
-    _map_response_decoder: Callable[[RoborockMessage], MapResponse] | None = None
+    _map_response_decoder: Callable[[RoborockMessage], MapResponse | None] | None = None
 
     def __init__(self, device_info: DeviceData, security_data: SecurityData | None) -> None:
         """Initializes the Roborock client."""
@@ -439,13 +439,14 @@ def on_message_received(self, messages: list[RoborockMessage]) -> None:
                 elif data.payload and protocol == RoborockMessageProtocol.MAP_RESPONSE:
                     if self._map_response_decoder is not None:
                         map_response = self._map_response_decoder(data)
-                        queue = self._waiting_queue.get(map_response.request_id)
-                        if queue:
-                            queue.set_result(map_response.data)
-                        else:
-                            self._logger.debug(
-                                "Received unsolicited map response for request_id %s", map_response.request_id
-                            )
+                        if map_response is not None:
+                            queue = self._waiting_queue.get(map_response.request_id)
+                            if queue:
+                                queue.set_result(map_response.data)
+                            else:
+                                self._logger.debug(
+                                    "Received unsolicited map response for request_id %s", map_response.request_id
+                                )
                 else:
                     queue = self._waiting_queue.get(data.seq)
                     if queue:

tests/protocols/test_v1_protocol.py

@@ -1,6 +1,7 @@
 """Tests for the v1 protocol message encoding and decoding."""
 
 import json
+import logging
 import pathlib
 from collections.abc import Generator
 from unittest.mock import patch
@@ -183,13 +184,14 @@ def test_create_map_response_decoder():
 
     decoder = create_map_response_decoder(SECURITY_DATA)
     result = decoder(message)
-
+    assert result is not None
     assert result.request_id == 44508
     assert result.data == test_data
 
 
-def test_create_map_response_decoder_invalid_endpoint():
+def test_create_map_response_decoder_invalid_endpoint(caplog: pytest.LogCaptureFixture):
     """Test map response decoder with invalid endpoint."""
+    caplog.set_level(logging.DEBUG)
     # Create header with wrong endpoint
     header = b"wrongend" + b"\x00" * 8 + b"\xdc\xad" + b"\x00" * 6
     payload = header + b"encrypted_data"
@@ -204,9 +206,8 @@ def test_create_map_response_decoder_invalid_endpoint():
     )
 
     decoder = create_map_response_decoder(SECURITY_DATA)
-
-    with pytest.raises(RoborockException, match="Invalid V1 map response endpoint"):
-        decoder(message)
+    assert decoder(message) is None
+    assert "Received map response requested not made by this device, ignoring." in caplog.text
 
 
 def test_create_map_response_decoder_invalid_payload():