Fix add done callback test

pull/22713/head
Pau Freixes 5 years ago
parent 314e105f70
commit 7dbed97594
  1. 10
      src/python/grpcio_tests/tests_aio/unit/done_callback_test.py

@ -47,12 +47,12 @@ class TestDoneCallback(AioTestBase):
call = self._stub.UnaryCall(messages_pb2.SimpleRequest())
self.assertEqual(grpc.StatusCode.OK, await call.code())
validation = _inject_callbacks(call)
validation = inject_callbacks(call)
await validation
async def test_unary_unary(self):
call = self._stub.UnaryCall(messages_pb2.SimpleRequest())
validation = _inject_callbacks(call)
validation = inject_callbacks(call)
self.assertEqual(grpc.StatusCode.OK, await call.code())
@ -65,7 +65,7 @@ class TestDoneCallback(AioTestBase):
messages_pb2.ResponseParameters(size=_RESPONSE_PAYLOAD_SIZE))
call = self._stub.StreamingOutputCall(request)
validation = _inject_callbacks(call)
validation = inject_callbacks(call)
response_cnt = 0
async for response in call:
@ -88,7 +88,7 @@ class TestDoneCallback(AioTestBase):
yield request
call = self._stub.StreamingInputCall(gen())
validation = _inject_callbacks(call)
validation = inject_callbacks(call)
response = await call
self.assertIsInstance(response, messages_pb2.StreamingInputCallResponse)
@ -100,7 +100,7 @@ class TestDoneCallback(AioTestBase):
async def test_stream_stream(self):
call = self._stub.FullDuplexCall()
validation = _inject_callbacks(call)
validation = inject_callbacks(call)
request = messages_pb2.StreamingOutputCallRequest()
request.response_parameters.append(

Loading…
Cancel
Save