Merge pull request #22142 from ericgribkoff/add_timeout

set timeout on rpc call
5 years ago · 6521dbacea
parent 933d564e73 47ceb4e035
commit 6521dbacea
1 changed files with 4 additions and 1 deletions
--- a/tools/run_tests/run_xds_tests.py
+++ b/tools/run_tests/run_xds_tests.py
@ -166,8 +166,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: