Merge pull request #2812 from soltanmm/timeout-increase

Use common timeouts in Python face-layer test-cases.

3 files changed, 80 insertions, 63 deletions

diff --git a/src/python/grpcio_test/grpc_test/framework/face/testing/blocking_invocation_inline_service_test_case.py b/src/python/grpcio_test/grpc_test/framework/face/testing/blocking_invocation_inline_service_test_case.py
index 7e1158f96b..251e1eb68e 100644
--- a/src/python/grpcio_test/grpc_test/framework/face/testing/blocking_invocation_inline_service_test_case.py
+++ b/src/python/grpcio_test/grpc_test/framework/face/testing/blocking_invocation_inline_service_test_case.py
@@ -34,15 +34,13 @@ import abc
 import unittest  # pylint: disable=unused-import
 
 from grpc.framework.face import exceptions
+from grpc_test.framework.common import test_constants
 from grpc_test.framework.face.testing import control
 from grpc_test.framework.face.testing import coverage
 from grpc_test.framework.face.testing import digest
 from grpc_test.framework.face.testing import stock_service
 from grpc_test.framework.face.testing import test_case
 
-_TIMEOUT = 3
-_LONG_TIMEOUT = 45
-
 
 class BlockingInvocationInlineServiceTestCase(
     test_case.FaceTestCase, coverage.BlockingCoverage):
@@ -79,7 +77,7 @@ class BlockingInvocationInlineServiceTestCase(
         request = test_messages.request()
 
         response = self.stub.blocking_value_in_value_out(
-            name, request, _LONG_TIMEOUT)
+            name, request, test_constants.LONG_TIMEOUT)
 
         test_messages.verify(request, response, self)
 
@@ -90,7 +88,7 @@ class BlockingInvocationInlineServiceTestCase(
         request = test_messages.request()
 
         response_iterator = self.stub.inline_value_in_stream_out(
-            name, request, _LONG_TIMEOUT)
+            name, request, test_constants.LONG_TIMEOUT)
         responses = list(response_iterator)
 
         test_messages.verify(request, responses, self)
@@ -102,7 +100,7 @@ class BlockingInvocationInlineServiceTestCase(
         requests = test_messages.requests()
 
         response = self.stub.blocking_stream_in_value_out(
-            name, iter(requests), _LONG_TIMEOUT)
+            name, iter(requests), test_constants.LONG_TIMEOUT)
 
         test_messages.verify(requests, response, self)
 
@@ -113,7 +111,7 @@ class BlockingInvocationInlineServiceTestCase(
         requests = test_messages.requests()
 
         response_iterator = self.stub.inline_stream_in_stream_out(
-            name, iter(requests), _LONG_TIMEOUT)
+            name, iter(requests), test_constants.LONG_TIMEOUT)
         responses = list(response_iterator)
 
         test_messages.verify(requests, responses, self)
@@ -126,12 +124,12 @@ class BlockingInvocationInlineServiceTestCase(
         second_request = test_messages.request()
 
         first_response = self.stub.blocking_value_in_value_out(
-            name, first_request, _TIMEOUT)
+            name, first_request, test_constants.SHORT_TIMEOUT)
 
         test_messages.verify(first_request, first_response, self)
 
         second_response = self.stub.blocking_value_in_value_out(
-            name, second_request, _TIMEOUT)
+            name, second_request, test_constants.SHORT_TIMEOUT)
 
         test_messages.verify(second_request, second_response, self)
 
@@ -144,7 +142,7 @@ class BlockingInvocationInlineServiceTestCase(
         with self.control.pause(), self.assertRaises(
             exceptions.ExpirationError):
           multi_callable = self.stub.unary_unary_multi_callable(name)
-          multi_callable(request, _TIMEOUT)
+          multi_callable(request, test_constants.SHORT_TIMEOUT)
 
   def testExpiredUnaryRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -155,7 +153,7 @@ class BlockingInvocationInlineServiceTestCase(
         with self.control.pause(), self.assertRaises(
             exceptions.ExpirationError):
           response_iterator = self.stub.inline_value_in_stream_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           list(response_iterator)
 
   def testExpiredStreamRequestUnaryResponse(self):
@@ -167,7 +165,7 @@ class BlockingInvocationInlineServiceTestCase(
         with self.control.pause(), self.assertRaises(
             exceptions.ExpirationError):
           multi_callable = self.stub.stream_unary_multi_callable(name)
-          multi_callable(iter(requests), _TIMEOUT)
+          multi_callable(iter(requests), test_constants.SHORT_TIMEOUT)
 
   def testExpiredStreamRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -178,7 +176,7 @@ class BlockingInvocationInlineServiceTestCase(
         with self.control.pause(), self.assertRaises(
             exceptions.ExpirationError):
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           list(response_iterator)
 
   def testFailedUnaryRequestUnaryResponse(self):
@@ -188,7 +186,8 @@ class BlockingInvocationInlineServiceTestCase(
         request = test_messages.request()
 
         with self.control.fail(), self.assertRaises(exceptions.ServicerError):
-          self.stub.blocking_value_in_value_out(name, request, _TIMEOUT)
+          self.stub.blocking_value_in_value_out(name, request,
+                                                test_constants.SHORT_TIMEOUT)
 
   def testFailedUnaryRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -198,7 +197,7 @@ class BlockingInvocationInlineServiceTestCase(
 
         with self.control.fail(), self.assertRaises(exceptions.ServicerError):
           response_iterator = self.stub.inline_value_in_stream_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           list(response_iterator)
 
   def testFailedStreamRequestUnaryResponse(self):
@@ -208,7 +207,8 @@ class BlockingInvocationInlineServiceTestCase(
         requests = test_messages.requests()
 
         with self.control.fail(), self.assertRaises(exceptions.ServicerError):
-          self.stub.blocking_stream_in_value_out(name, iter(requests), _TIMEOUT)
+          self.stub.blocking_stream_in_value_out(name, iter(requests),
+                                                 test_constants.SHORT_TIMEOUT)
 
   def testFailedStreamRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -218,5 +218,5 @@ class BlockingInvocationInlineServiceTestCase(
 
         with self.control.fail(), self.assertRaises(exceptions.ServicerError):
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           list(response_iterator)
diff --git a/src/python/grpcio_test/grpc_test/framework/face/testing/event_invocation_synchronous_event_service_test_case.py b/src/python/grpcio_test/grpc_test/framework/face/testing/event_invocation_synchronous_event_service_test_case.py
index 18eed53d6e..9df77678eb 100644
--- a/src/python/grpcio_test/grpc_test/framework/face/testing/event_invocation_synchronous_event_service_test_case.py
+++ b/src/python/grpcio_test/grpc_test/framework/face/testing/event_invocation_synchronous_event_service_test_case.py
@@ -33,6 +33,7 @@ import abc
 import unittest
 
 from grpc.framework.face import interfaces
+from grpc_test.framework.common import test_constants
 from grpc_test.framework.face.testing import callback as testing_callback
 from grpc_test.framework.face.testing import control
 from grpc_test.framework.face.testing import coverage
@@ -40,8 +41,6 @@ from grpc_test.framework.face.testing import digest
 from grpc_test.framework.face.testing import stock_service
 from grpc_test.framework.face.testing import test_case
 
-_TIMEOUT = 3
-
 
 class EventInvocationSynchronousEventServiceTestCase(
     test_case.FaceTestCase, coverage.FullCoverage):
@@ -79,7 +78,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         self.stub.event_value_in_value_out(
-            name, request, callback.complete, callback.abort, _TIMEOUT)
+            name, request, callback.complete, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         callback.block_until_terminated()
         response = callback.response()
 
@@ -93,7 +93,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         self.stub.event_value_in_stream_out(
-            name, request, callback, callback.abort, _TIMEOUT)
+            name, request, callback, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         callback.block_until_terminated()
         responses = callback.responses()
 
@@ -107,7 +108,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         unused_call, request_consumer = self.stub.event_stream_in_value_out(
-            name, callback.complete, callback.abort, _TIMEOUT)
+            name, callback.complete, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         for request in requests:
           request_consumer.consume(request)
         request_consumer.terminate()
@@ -124,7 +126,7 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         unused_call, request_consumer = self.stub.event_stream_in_stream_out(
-            name, callback, callback.abort, _TIMEOUT)
+            name, callback, callback.abort, test_constants.SHORT_TIMEOUT)
         for request in requests:
           request_consumer.consume(request)
         request_consumer.terminate()
@@ -147,11 +149,11 @@ class EventInvocationSynchronousEventServiceTestCase(
           first_callback.complete(first_response)
           self.stub.event_value_in_value_out(
               name, second_request, second_callback.complete,
-              second_callback.abort, _TIMEOUT)
+              second_callback.abort, test_constants.SHORT_TIMEOUT)
 
         self.stub.event_value_in_value_out(
             name, first_request, make_second_invocation, first_callback.abort,
-            _TIMEOUT)
+           test_constants.SHORT_TIMEOUT)
         second_callback.block_until_terminated()
 
         first_response = first_callback.response()
@@ -168,7 +170,8 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.pause():
           self.stub.event_value_in_value_out(
-              name, request, callback.complete, callback.abort, _TIMEOUT)
+              name, request, callback.complete, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           callback.block_until_terminated()
 
         self.assertEqual(interfaces.Abortion.EXPIRED, callback.abortion())
@@ -182,7 +185,8 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.pause():
           self.stub.event_value_in_stream_out(
-              name, request, callback, callback.abort, _TIMEOUT)
+              name, request, callback, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           callback.block_until_terminated()
 
         self.assertEqual(interfaces.Abortion.EXPIRED, callback.abortion())
@@ -194,7 +198,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         self.stub.event_stream_in_value_out(
-            name, callback.complete, callback.abort, _TIMEOUT)
+            name, callback.complete, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         callback.block_until_terminated()
 
         self.assertEqual(interfaces.Abortion.EXPIRED, callback.abortion())
@@ -207,7 +212,7 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         unused_call, request_consumer = self.stub.event_stream_in_stream_out(
-            name, callback, callback.abort, _TIMEOUT)
+            name, callback, callback.abort, test_constants.SHORT_TIMEOUT)
         for request in requests:
           request_consumer.consume(request)
         callback.block_until_terminated()
@@ -223,10 +228,12 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.fail():
           self.stub.event_value_in_value_out(
-              name, request, callback.complete, callback.abort, _TIMEOUT)
+              name, request, callback.complete, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           callback.block_until_terminated()
 
-        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE, callback.abortion())
+        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE,
+                         callback.abortion())
 
   def testFailedUnaryRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -237,10 +244,12 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.fail():
           self.stub.event_value_in_stream_out(
-              name, request, callback, callback.abort, _TIMEOUT)
+              name, request, callback, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           callback.block_until_terminated()
 
-        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE, callback.abortion())
+        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE,
+                         callback.abortion())
 
   def testFailedStreamRequestUnaryResponse(self):
     for name, test_messages_sequence in (
@@ -251,13 +260,15 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.fail():
           unused_call, request_consumer = self.stub.event_stream_in_value_out(
-              name, callback.complete, callback.abort, _TIMEOUT)
+              name, callback.complete, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           for request in requests:
             request_consumer.consume(request)
           request_consumer.terminate()
           callback.block_until_terminated()
 
-        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE, callback.abortion())
+        self.assertEqual(interfaces.Abortion.SERVICER_FAILURE,
+                         callback.abortion())
 
   def testFailedStreamRequestStreamResponse(self):
     for name, test_messages_sequence in (
@@ -268,7 +279,7 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.fail():
           unused_call, request_consumer = self.stub.event_stream_in_stream_out(
-              name, callback, callback.abort, _TIMEOUT)
+              name, callback, callback.abort, test_constants.SHORT_TIMEOUT)
           for request in requests:
             request_consumer.consume(request)
           request_consumer.terminate()
@@ -287,10 +298,10 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         self.stub.event_value_in_value_out(
             name, first_request, first_callback.complete, first_callback.abort,
-            _TIMEOUT)
+           test_constants.SHORT_TIMEOUT)
         self.stub.event_value_in_value_out(
             name, second_request, second_callback.complete,
-            second_callback.abort, _TIMEOUT)
+            second_callback.abort, test_constants.SHORT_TIMEOUT)
         first_callback.block_until_terminated()
         second_callback.block_until_terminated()
 
@@ -312,7 +323,8 @@ class EventInvocationSynchronousEventServiceTestCase(
 
         with self.control.pause():
           call = self.stub.event_value_in_value_out(
-              name, request, callback.complete, callback.abort, _TIMEOUT)
+              name, request, callback.complete, callback.abort,
+              test_constants.SHORT_TIMEOUT)
           call.cancel()
           callback.block_until_terminated()
 
@@ -326,7 +338,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         call = self.stub.event_value_in_stream_out(
-            name, request, callback, callback.abort, _TIMEOUT)
+            name, request, callback, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         call.cancel()
         callback.block_until_terminated()
 
@@ -340,7 +353,8 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         call, request_consumer = self.stub.event_stream_in_value_out(
-            name, callback.complete, callback.abort, _TIMEOUT)
+            name, callback.complete, callback.abort,
+            test_constants.SHORT_TIMEOUT)
         for request in requests:
           request_consumer.consume(request)
         call.cancel()
@@ -355,7 +369,7 @@ class EventInvocationSynchronousEventServiceTestCase(
         callback = testing_callback.Callback()
 
         call, unused_request_consumer = self.stub.event_stream_in_stream_out(
-            name, callback, callback.abort, _TIMEOUT)
+            name, callback, callback.abort, test_constants.SHORT_TIMEOUT)
         call.cancel()
         callback.block_until_terminated()
 
diff --git a/src/python/grpcio_test/grpc_test/framework/face/testing/future_invocation_asynchronous_event_service_test_case.py b/src/python/grpcio_test/grpc_test/framework/face/testing/future_invocation_asynchronous_event_service_test_case.py
index 3b42914342..70d86a0422 100644
--- a/src/python/grpcio_test/grpc_test/framework/face/testing/future_invocation_asynchronous_event_service_test_case.py
+++ b/src/python/grpcio_test/grpc_test/framework/face/testing/future_invocation_asynchronous_event_service_test_case.py
@@ -37,13 +37,13 @@ import unittest
 from grpc.framework.face import exceptions
 from grpc.framework.foundation import future
 from grpc.framework.foundation import logging_pool
+from grpc_test.framework.common import test_constants
 from grpc_test.framework.face.testing import control
 from grpc_test.framework.face.testing import coverage
 from grpc_test.framework.face.testing import digest
 from grpc_test.framework.face.testing import stock_service
 from grpc_test.framework.face.testing import test_case
 
-_TIMEOUT = 3
 _MAXIMUM_POOL_SIZE = 10
 
 
@@ -110,7 +110,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         request = test_messages.request()
 
         response_future = self.stub.future_value_in_value_out(
-            name, request, _TIMEOUT)
+            name, request, test_constants.SHORT_TIMEOUT)
         response = response_future.result()
 
         test_messages.verify(request, response, self)
@@ -122,7 +122,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         request = test_messages.request()
 
         response_iterator = self.stub.inline_value_in_stream_out(
-            name, request, _TIMEOUT)
+            name, request, test_constants.SHORT_TIMEOUT)
         responses = list(response_iterator)
 
         test_messages.verify(request, responses, self)
@@ -138,7 +138,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         # returned to calling code before the iterator yields any requests.
         with request_iterator.pause():
           response_future = self.stub.future_stream_in_value_out(
-              name, request_iterator, _TIMEOUT)
+              name, request_iterator, test_constants.SHORT_TIMEOUT)
         response = response_future.result()
 
         test_messages.verify(requests, response, self)
@@ -154,7 +154,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         # returned to calling code before the iterator yields any requests.
         with request_iterator.pause():
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, request_iterator, _TIMEOUT)
+              name, request_iterator, test_constants.SHORT_TIMEOUT)
         responses = list(response_iterator)
 
         test_messages.verify(requests, responses, self)
@@ -167,13 +167,13 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         second_request = test_messages.request()
 
         first_response_future = self.stub.future_value_in_value_out(
-            name, first_request, _TIMEOUT)
+            name, first_request, test_constants.SHORT_TIMEOUT)
         first_response = first_response_future.result()
 
         test_messages.verify(first_request, first_response, self)
 
         second_response_future = self.stub.future_value_in_value_out(
-            name, second_request, _TIMEOUT)
+            name, second_request, test_constants.SHORT_TIMEOUT)
         second_response = second_response_future.result()
 
         test_messages.verify(second_request, second_response, self)
@@ -186,7 +186,8 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           multi_callable = self.stub.unary_unary_multi_callable(name)
-          response_future = multi_callable.future(request, _TIMEOUT)
+          response_future = multi_callable.future(request,
+                                                  test_constants.SHORT_TIMEOUT)
           self.assertIsInstance(
               response_future.exception(), exceptions.ExpirationError)
           with self.assertRaises(exceptions.ExpirationError):
@@ -200,7 +201,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_iterator = self.stub.inline_value_in_stream_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           with self.assertRaises(exceptions.ExpirationError):
             list(response_iterator)
 
@@ -212,7 +213,8 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           multi_callable = self.stub.stream_unary_multi_callable(name)
-          response_future = multi_callable.future(iter(requests), _TIMEOUT)
+          response_future = multi_callable.future(iter(requests),
+                                                  test_constants.SHORT_TIMEOUT)
           self.assertIsInstance(
               response_future.exception(), exceptions.ExpirationError)
           with self.assertRaises(exceptions.ExpirationError):
@@ -226,7 +228,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           with self.assertRaises(exceptions.ExpirationError):
             list(response_iterator)
 
@@ -238,7 +240,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.fail():
           response_future = self.stub.future_value_in_value_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
 
           # Because the servicer fails outside of the thread from which the
           # servicer-side runtime called into it its failure is
@@ -261,7 +263,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         # expiration of the RPC.
         with self.control.fail(), self.assertRaises(exceptions.ExpirationError):
           response_iterator = self.stub.inline_value_in_stream_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           list(response_iterator)
 
   def testFailedStreamRequestUnaryResponse(self):
@@ -272,7 +274,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.fail():
           response_future = self.stub.future_stream_in_value_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
 
           # Because the servicer fails outside of the thread from which the
           # servicer-side runtime called into it its failure is
@@ -295,7 +297,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         # expiration of the RPC.
         with self.control.fail(), self.assertRaises(exceptions.ExpirationError):
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           list(response_iterator)
 
   def testParallelInvocations(self):
@@ -305,10 +307,11 @@ class FutureInvocationAsynchronousEventServiceTestCase(
         first_request = test_messages.request()
         second_request = test_messages.request()
 
+        # TODO(bug 2039): use LONG_TIMEOUT instead
         first_response_future = self.stub.future_value_in_value_out(
-            name, first_request, _TIMEOUT)
+            name, first_request, test_constants.SHORT_TIMEOUT)
         second_response_future = self.stub.future_value_in_value_out(
-            name, second_request, _TIMEOUT)
+            name, second_request, test_constants.SHORT_TIMEOUT)
         first_response = first_response_future.result()
         second_response = second_response_future.result()
 
@@ -327,7 +330,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_future = self.stub.future_value_in_value_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           cancel_method_return_value = response_future.cancel()
 
         self.assertFalse(cancel_method_return_value)
@@ -341,7 +344,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_iterator = self.stub.inline_value_in_stream_out(
-              name, request, _TIMEOUT)
+              name, request, test_constants.SHORT_TIMEOUT)
           response_iterator.cancel()
 
         with self.assertRaises(future.CancelledError):
@@ -355,7 +358,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_future = self.stub.future_stream_in_value_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           cancel_method_return_value = response_future.cancel()
 
         self.assertFalse(cancel_method_return_value)
@@ -369,7 +372,7 @@ class FutureInvocationAsynchronousEventServiceTestCase(
 
         with self.control.pause():
           response_iterator = self.stub.inline_stream_in_stream_out(
-              name, iter(requests), _TIMEOUT)
+              name, iter(requests), test_constants.SHORT_TIMEOUT)
           response_iterator.cancel()
 
         with self.assertRaises(future.CancelledError):