From e8cff6fef61e22a99b5848bc5534369006c00fbf Mon Sep 17 00:00:00 2001 From: Michel Hidalgo Date: Fri, 29 Mar 2024 09:25:25 -0300 Subject: [PATCH] Initialize motor power state in mocks Signed-off-by: Michel Hidalgo --- spot_wrapper/testing/mocks/power.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/spot_wrapper/testing/mocks/power.py b/spot_wrapper/testing/mocks/power.py index fb8f16f..3728ca4 100644 --- a/spot_wrapper/testing/mocks/power.py +++ b/spot_wrapper/testing/mocks/power.py @@ -1,5 +1,7 @@ # Copyright (c) 2024 Boston Dynamics AI Institute LLC. See LICENSE file for more info. +from typing import Any + import grpc from bosdyn.api.power_pb2 import ( FanPowerCommandFeedbackRequest, @@ -21,6 +23,10 @@ class MockPowerService(PowerServiceServicer, MockRobotStateService): """A mock Spot power service.""" + def __init__(self, **kwargs: Any) -> None: + super().__init__(**kwargs) + self.robot_state.power_state.motor_power_state = PowerState.MotorPowerState.MOTOR_POWER_STATE_OFF + def PowerCommand(self, request: PowerCommandRequest, context: grpc.ServicerContext) -> PowerCommandResponse: response = PowerCommandResponse() if request.request == PowerCommandRequest.Request.REQUEST_ON_MOTORS: