Push test coverage for AsyncClient to 100%

Author: lidizheng

gapic/templates/tests/unit/%name_%version/%sub/test_%service.py.j2

@@ -302,7 +302,7 @@ async def test_{{ method.name|snake_case }}_async(transport: str = 'grpc_asyncio
 def test_{{ method.name|snake_case }}_field_headers():
     client = {{ service.client_name }}(
         credentials=credentials.AnonymousCredentials(),
-  )
+    )
 
     # Any value that is part of the HTTP/1.1 URI should be sent as
     # a field header. Set these to a non-empty value.
@@ -333,6 +333,43 @@ def test_{{ method.name|snake_case }}_field_headers():
         {%- if not loop.last %}&{% endif -%}
         {%- endfor %}',
     ) in kw['metadata']
+
+
+@pytest.mark.asyncio
+async def test_{{ method.name|snake_case }}_field_headers_async():
+    client = {{ service.async_client_name }}(
+        credentials=credentials.AnonymousCredentials(),
+    )
+
+    # Any value that is part of the HTTP/1.1 URI should be sent as
+    # a field header. Set these to a non-empty value.
+    request = {{ method.input.ident }}(
+        {%- for field_header in method.field_headers %}
+        {{ field_header }}='{{ field_header }}/value',
+        {%- endfor %}
+    )
+
+    # Mock the actual call within the gRPC stub, and fake the request.
+    with mock.patch.object(
+            type(client._client._transport.{{ method.name|snake_case }}),
+            '__call__') as call:
+        call.return_value = grpc_helpers_async.FakeUnaryUnaryCall({{ method.output.ident }}())
+        await client.{{ method.name|snake_case }}(request)
+
+        # Establish that the underlying gRPC stub method was called.
+        assert len(call.mock_calls) == 1
+        _, args, _ = call.mock_calls[0]
+        assert args[0] == request
+
+    # Establish that the field header was sent.
+    _, _, kw = call.mock_calls[0]
+    assert (
+        'x-goog-request-params',
+        '{% for field_header in method.field_headers -%}
+        {{ field_header }}={{ field_header }}/value
+        {%- if not loop.last %}&{% endif -%}
+        {%- endfor %}',
+    ) in kw['metadata']
 {% endif %}
 
 {% if method.ident.package != method.input.ident.package %}
@@ -415,6 +452,80 @@ def test_{{ method.name|snake_case }}_flattened_error():
             {{ field.name }}={{ field.mock_value }},
             {%- endfor %}
         )
+
+
+@pytest.mark.asyncio
+async def test_{{ method.name|snake_case }}_flattened_async():
+    client = {{ service.async_client_name }}(
+        credentials=credentials.AnonymousCredentials(),
+    )
+
+    # Mock the actual call within the gRPC stub, and fake the request.
+    with mock.patch.object(
+            type(client._client._transport.{{ method.name|snake_case }}),
+            '__call__') as call:
+        # Designate an appropriate return value for the call.
+        {% if method.void -%}
+        call.return_value = None
+        {% elif method.lro -%}
+        call.return_value = operations_pb2.Operation(name='operations/op')
+        {% elif method.server_streaming -%}
+        call.return_value = iter([{{ method.output.ident }}()])
+        {% else -%}
+        call.return_value = {{ method.output.ident }}()
+        {% endif %}
+
+
+        {% if method.void -%}
+        call.return_value = grpc_helpers_async.FakeUnaryUnaryCall(None)
+        {% elif method.lro -%}
+        call.return_value = grpc_helpers_async.FakeUnaryUnaryCall(
+            operations_pb2.Operation(name='operations/spam')
+        )
+        {% elif not method.client_streaming and method.server_streaming -%}
+        call.return_value = mock.Mock(aio.UnaryStreamCall, autospec=True)
+        {% elif method.client_streaming and method.server_streaming -%}
+        call.return_value = mock.Mock(aio.StreamStreamCall, autospec=True)
+        {% else -%}
+        call.return_value ={{' '}}
+            {%- if not method.client_streaming and not method.server_streaming -%}
+                grpc_helpers_async.FakeUnaryUnaryCall
+            {%- else -%}
+                grpc_helpers_async.FakeStreamUnaryCall
+            {%- endif -%}({{ method.output.ident }}())
+        {% endif -%}
+        # Call the method with a truthy value for each flattened field,
+        # using the keyword arguments to the method.
+        response = await client.{{ method.name|snake_case }}(
+            {%- for field in method.flattened_fields.values() %}
+            {{ field.name }}={{ field.mock_value }},
+            {%- endfor %}
+        )
+
+        # Establish that the underlying call was made with the expected
+        # request object values.
+        assert len(call.mock_calls) == 1
+        _, args, _ = call.mock_calls[0]
+        {% for key, field in method.flattened_fields.items() -%}
+        assert args[0].{{ key }} == {{ field.mock_value }}
+        {% endfor %}
+
+
+@pytest.mark.asyncio
+async def test_{{ method.name|snake_case }}_flattened_error_async():
+    client = {{ service.async_client_name }}(
+        credentials=credentials.AnonymousCredentials(),
+    )
+
+    # Attempting to call a method with both a request object and flattened
+    # fields is an error.
+    with pytest.raises(ValueError):
+        await client.{{ method.name|snake_case }}(
+            {{ method.input.ident }}(),
+            {%- for field in method.flattened_fields.values() %}
+            {{ field.name }}={{ field.mock_value }},
+            {%- endfor %}
+        )
 {% endif %}