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:
parent
f3da9de744
commit
54188b4128
2 changed files with 3 additions and 1 deletions
|
@ -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
|
||||
):
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Add table
Reference in a new issue