diff --git a/tools/run_tests/run_xds_tests.py b/tools/run_tests/run_xds_tests.py index 226dbfbb000..be4533d1664 100755 --- a/tools/run_tests/run_xds_tests.py +++ b/tools/run_tests/run_xds_tests.py @@ -148,8 +148,11 @@ def get_client_stats(num_rpcs, timeout_sec): request = messages_pb2.LoadBalancerStatsRequest() request.num_rpcs = num_rpcs request.timeout_sec = timeout_sec + rpc_timeout = timeout_sec * 2 # Allow time for connection establishment try: - response = stub.GetClientStats(request, wait_for_ready=True) + response = stub.GetClientStats(request, + wait_for_ready=True, + timeout=rpc_timeout) logger.debug('Invoked GetClientStats RPC: %s', response) return response except grpc.RpcError as rpc_error: