diff --git a/.gitignore b/.gitignore index b6829a9..d493c67 100644 --- a/.gitignore +++ b/.gitignore @@ -24,4 +24,8 @@ massa_test_framework.egg-info doc/_build # IDE -.idea \ No newline at end of file +.idea +.vscode + +kubeconfig* + diff --git a/massa_test_framework/node.py b/massa_test_framework/node.py index 697180f..101b244 100644 --- a/massa_test_framework/node.py +++ b/massa_test_framework/node.py @@ -20,6 +20,8 @@ PrivateServiceStub, GetStatusRequest, GetStatusResponse, + GetStakersRequest, + GetStakersResponse, QueryStateResponse, QueryStateRequest, ) @@ -483,7 +485,23 @@ def query_state(self, query_state_request: QueryStateRequest) -> QueryStateRespo ) return query_state_response - # + def get_stakers_grpc(self) -> GetStakersResponse: + """Queries the gRPC GetStakers method. + + Example: + res = node.get_stakers_grpc() + print(res.stakers) + + """ + + get_stakers_response: GetStakersResponse = asyncio.run( + self._public_grpc_call( + self.grpc_host, self.pub_grpc_port, "get_stakers", GetStakersRequest() + ) + ) + + return get_stakers_response + def wait_ready(self, timeout=20) -> None: """Wait for node to be ready