diff --git a/massa_test_framework/node.py b/massa_test_framework/node.py index 593e9ca..7d49caf 100644 --- a/massa_test_framework/node.py +++ b/massa_test_framework/node.py @@ -486,7 +486,7 @@ def query_state(self, query_state_request: QueryStateRequest) -> QueryStateRespo return query_state_response def grpc_get_stakers(self) -> GetStakersResponse: - """Queries the execution state of the node. + """gRPC GetStakers. Example:: res = node.grpc_get_stakers() @@ -501,7 +501,24 @@ def grpc_get_stakers(self) -> GetStakersResponse: ) return get_stakers_response - # + + def grpc_get_status(self) -> GetStatusResponse: + """gRPC GetStatus. + + Example:: + res = node.grpc_get_status() + print(res.status) + + """ + + get_status_response: GetStatusResponse = asyncio.run( + self._public_grpc_call( + self.grpc_host, self.pub_grpc_port, "get_status", GetStatusRequest() + ) + ) + + return get_status_response + def wait_ready(self, timeout=20) -> None: """Wait for node to be ready