Add returning activity to Husqvarna lawn mower (#124511)

* add returning activity to husqvarna lawn mower

* Update test, fix bug with comparison operator
This commit is contained in:
Michael Arthur 2024-08-30 22:59:13 +12:00 committed by GitHub
parent f3da9de744
commit 54188b4128
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions

View file

@ -26,7 +26,6 @@ DOCKED_ACTIVITIES = (MowerActivities.PARKED_IN_CS, MowerActivities.CHARGING)
MOWING_ACTIVITIES = (
MowerActivities.MOWING,
MowerActivities.LEAVING,
MowerActivities.GOING_HOME,
)
PAUSED_STATES = [
MowerStates.PAUSED,
@ -107,6 +106,8 @@ class AutomowerLawnMowerEntity(AutomowerAvailableEntity, LawnMowerEntity):
return LawnMowerActivity.PAUSED
if mower_attributes.mower.activity in MOWING_ACTIVITIES:
return LawnMowerActivity.MOWING
if mower_attributes.mower.activity == MowerActivities.GOING_HOME:
return LawnMowerActivity.RETURNING
if (mower_attributes.mower.state == "RESTRICTED") or (
mower_attributes.mower.activity in DOCKED_ACTIVITIES
):

View file

@ -44,6 +44,7 @@ async def test_lawn_mower_states(
("UNKNOWN", "PAUSED", LawnMowerActivity.PAUSED),
("MOWING", "NOT_APPLICABLE", LawnMowerActivity.MOWING),
("NOT_APPLICABLE", "ERROR", LawnMowerActivity.ERROR),
("GOING_HOME", "IN_OPERATION", LawnMowerActivity.RETURNING),
):
values[TEST_MOWER_ID].mower.activity = activity
values[TEST_MOWER_ID].mower.state = state