Add tests for FakeStreamUnaryCall

googleapis · Apr 22, 2020 · 204cc4a · 204cc4a
1 parent 96f8b17
commit 204cc4a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/asyncio/test_grpc_helpers_async.py b/tests/asyncio/test_grpc_helpers_async.py
@@ -362,3 +362,11 @@ def test_create_channel_without_grpc_gcp(grpc_secure_channel):
  grpc_helpers_async.create_channel(target, credentials=credentials, scopes=scopes)
  grpc_secure_channel.assert_called()
  credentials.with_scopes.assert_called_once_with(scopes)
+
+
+@pytest.mark.asyncio
+async def test_fake_stream_unary_call():
+ fake_call = grpc_helpers_async.FakeStreamUnaryCall()
+ await fake_call.wait_for_connection()
+ response = await fake_call
+ assert fake_call.response == response