diff --git a/docs/examples/_server.py b/docs/examples/_server.py index a44084680..a30b630ec 100644 --- a/docs/examples/_server.py +++ b/docs/examples/_server.py @@ -6,6 +6,11 @@ from grpclib.server import Server, Stream from grpclib.utils import graceful_exit +from viam.proto.robot import ( + UnimplementedRobotServiceBase, + GetMachineStatusRequest, + GetMachineStatusResponse +) from viam.app.data_client import DataClient from viam.proto.app import ( AddRoleRequest, @@ -595,6 +600,11 @@ async def GetRegistryItem(self, stream: Stream[GetRegistryItemRequest, GetRegist raise NotImplementedError() +class MockRobot(UnimplementedRobotServiceBase): + async def GetMachineStatus(self, stream: Stream[GetMachineStatusRequest, GetMachineStatusResponse]) -> None: + await stream.send_message(GetMachineStatusResponse(state: GetMachineStatusResponse.STATE_RUNNING)) + + async def main(*, host: str = "127.0.0.1", port: int = 9092) -> None: server = Server([MockData(), MockDataSync(), MockApp()]) with graceful_exit([server]):