chore: testing

Author: Lash-L

tests/devices/test_local_channel.py

@@ -7,9 +7,10 @@
 
 import pytest
 
-from roborock.devices.local_channel import LocalChannel
+from roborock.devices.local_channel import LocalChannel, LocalChannelParams
 from roborock.exceptions import RoborockConnectionException
 from roborock.protocol import create_local_decoder, create_local_encoder
+from roborock.protocols.v1_protocol import LocalProtocolVersion
 from roborock.roborock_message import RoborockMessage, RoborockMessageProtocol
 
 TEST_HOST = "192.168.1.100"
@@ -56,9 +57,18 @@ def setup_mock_loop(mock_transport: Mock) -> Generator[Mock, None, None]:
 
 
 @pytest.fixture(name="local_channel")
-def setup_local_channel() -> LocalChannel:
-    """Fixture to set up the local channel for tests."""
-    return LocalChannel(host=TEST_HOST, local_key=TEST_LOCAL_KEY)
+async def setup_local_channel_with_hello_mock() -> LocalChannel:
+    """Fixture to set up the local channel with automatic hello mocking."""
+    channel = LocalChannel(host=TEST_HOST, local_key=TEST_LOCAL_KEY)
+
+    async def mock_do_hello(local_protocol_version):
+        """Mock _do_hello to return successful params without sending actual request."""
+        return LocalChannelParams(local_key=channel._local_key, connect_nonce=channel._connect_nonce, ack_nonce=54321)
+
+    # Replace the _do_hello method
+    setattr(channel, "_do_hello", mock_do_hello)
+
+    return channel
 
 
 @pytest.fixture(name="received_messages")
@@ -231,3 +241,33 @@ async def test_connection_lost_without_exception(
     assert local_channel._is_connected is False
     assert local_channel._transport is None
     assert "Connection lost to 192.168.1.100" in caplog.text
+
+
+async def test_hello_fallback_to_l01_protocol(mock_loop: Mock, mock_transport: Mock) -> None:
+    """Test that when first hello() message fails (V1) but second succeeds (L01), we use L01."""
+
+    # Create a channel without the automatic hello mocking
+    channel = LocalChannel(host=TEST_HOST, local_key=TEST_LOCAL_KEY)
+
+    # Mock _do_hello to fail for V1 but succeed for L01
+    async def mock_do_hello(local_protocol_version: LocalProtocolVersion) -> LocalChannelParams | None:
+        if local_protocol_version == LocalProtocolVersion.V1:
+            # First attempt (V1) fails - return None to simulate failure
+            return None
+        elif local_protocol_version == LocalProtocolVersion.L01:
+            # Second attempt (L01) succeeds
+            return LocalChannelParams(
+                local_key=channel._local_key, connect_nonce=channel._connect_nonce, ack_nonce=54321
+            )
+        return None
+
+    # Replace the _do_hello method
+    setattr(channel, "_do_hello", mock_do_hello)
+
+    # Connect and verify L01 protocol is used
+    await channel.connect()
+
+    # Verify that the channel is using L01 protocol
+    assert channel._local_protocol_version == LocalProtocolVersion.L01
+    assert channel._ack_nonce == 54321
+    assert channel._is_connected is True