Add test of automower_ble activity mapping (#150983)

Co-authored-by: Copilot <[email protected]>

Author: emontnemery

tests/components/husqvarna_automower_ble/test_lawn_mower.py

@@ -3,10 +3,12 @@
 from datetime import timedelta
 from unittest.mock import Mock
 
+from automower_ble.protocol import MowerActivity, MowerState
 from bleak import BleakError
 from freezegun.api import FrozenDateTimeFactory
 import pytest
 
+from homeassistant.components.lawn_mower import LawnMowerActivity
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.const import STATE_UNAVAILABLE
 from homeassistant.core import HomeAssistant
@@ -124,3 +126,83 @@ async def test_bleak_error_data_update(
     await hass.async_block_till_done()
 
     assert hass.states.get("lawn_mower.husqvarna_automower").state == STATE_UNAVAILABLE
+
+
+OPERATIONAL_STATES = [
+    MowerState.IN_OPERATION,
+    MowerState.PENDING_START,
+    MowerState.RESTRICTED,
+]
+
+
+@pytest.mark.parametrize(
+    ("mower_states", "mower_activities", "expected_state"),
+    [
+        # MowerState ERROR, FATAL_ERROR, OFF, STOPPED, WAIT_FOR_SAFETYPIN -> Mapped to
+        # LawnMowerActivity.ERROR
+        (
+            [
+                MowerState.ERROR,
+                MowerState.FATAL_ERROR,
+                MowerState.OFF,
+                MowerState.STOPPED,
+                MowerState.WAIT_FOR_SAFETYPIN,
+            ],
+            list(MowerActivity),
+            LawnMowerActivity.ERROR,
+        ),
+        # MowerState PAUSED -> Mapped to LawnMowerActivity.PAUSED
+        ([MowerState.PAUSED], list(MowerActivity), LawnMowerActivity.PAUSED),
+        # Operational states are mapped according to the activity
+        (
+            OPERATIONAL_STATES,
+            [MowerActivity.CHARGING, MowerActivity.NONE, MowerActivity.PARKED],
+            LawnMowerActivity.DOCKED,
+        ),
+        (
+            OPERATIONAL_STATES,
+            [MowerActivity.GOING_OUT, MowerActivity.MOWING],
+            LawnMowerActivity.MOWING,
+        ),
+        (
+            OPERATIONAL_STATES,
+            [MowerActivity.GOING_HOME],
+            LawnMowerActivity.RETURNING,
+        ),
+        (
+            OPERATIONAL_STATES,
+            [MowerActivity.STOPPED_IN_GARDEN],
+            LawnMowerActivity.ERROR,
+        ),
+    ],
+)
+async def test_mower_activity_mapping(
+    hass: HomeAssistant,
+    mock_automower_client: Mock,
+    mock_config_entry: MockConfigEntry,
+    freezer: FrozenDateTimeFactory,
+    mower_states: list[MowerState],
+    mower_activities: list[MowerActivity],
+    expected_state: str,
+) -> None:
+    """Test mower state and activity mapping to LawnMowerActivity states."""
+
+    mock_config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert mock_config_entry.state is ConfigEntryState.LOADED
+
+    for mower_state in mower_states:
+        for mower_activity in mower_activities:
+            mock_automower_client.mower_state.return_value = mower_state
+            mock_automower_client.mower_activity.return_value = mower_activity
+
+            freezer.tick(timedelta(seconds=60))
+            async_fire_time_changed(hass)
+            await hass.async_block_till_done()
+
+            assert (
+                hass.states.get("lawn_mower.husqvarna_automower").state
+                == expected_state
+            )