diff options
| author |  Nathaniel Manista <nathaniel@google.com> | 2015-08-28 16:07:23 +0000 |
|---|---|---|
| committer | Nathaniel Manista <nathaniel@google.com> | 2015-08-28 21:08:25 +0000 |
| commit | 4354f3e6809f12512d93b33e667133bc15f44029 (patch) | |
| tree | 2645242f420423d261d57c267bc53738ac8b0e36 /src/python/grpcio_test/grpc_test/_links | |
| parent | 400e0cdd321fbec7311d58d9e407c2426490eb69 (diff) | |
Make ServiceLink shut-down a two step process
Diffstat (limited to 'src/python/grpcio_test/grpc_test/_links')
| -rw-r--r-- | src/python/grpcio_test/grpc_test/_links/_transmission_test.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/python/grpcio_test/grpc_test/_links/_transmission_test.py b/src/python/grpcio_test/grpc_test/_links/_transmission_test.py index db011bca66..616d8a670d 100644 --- a/src/python/grpcio_test/grpc_test/_links/_transmission_test.py +++ b/src/python/grpcio_test/grpc_test/_links/_transmission_test.py @@ -62,7 +62,8 @@ class TransmissionTest(test_cases.TransmissionTest, unittest.TestCase): def destroy_transmitting_links(self, invocation_side_link, service_side_link): invocation_side_link.stop() - service_side_link.stop_gracefully() + service_side_link.begin_stop() + service_side_link.end_stop() def create_invocation_initial_metadata(self): return ( @@ -140,7 +141,8 @@ class RoundTripTest(unittest.TestCase): invocation_mate.block_until_tickets_satisfy(test_cases.terminated) invocation_link.stop() - service_link.stop_gracefully() + service_link.begin_stop() + service_link.end_stop() self.assertIs( service_mate.tickets()[-1].termination, @@ -206,7 +208,8 @@ class RoundTripTest(unittest.TestCase): invocation_mate.block_until_tickets_satisfy(test_cases.terminated) invocation_link.stop() - service_link.stop_gracefully() + service_link.begin_stop() + service_link.end_stop() observed_requests = tuple( ticket.payload for ticket in service_mate.tickets() |