Test the creation of AsyncClient

Author: lidizheng

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

@@ -51,33 +51,38 @@ def test__get_default_mtls_endpoint():
     assert {{ service.client_name }}._get_default_mtls_endpoint(non_googleapi) == non_googleapi
 
 
-def test_{{ service.client_name|snake_case }}_from_service_account_file():
+@pytest.mark.parametrize("client_class", [{{ service.client_name }}, {{ service.async_client_name }}])
+def test_{{ service.client_name|snake_case }}_from_service_account_file(client_class):
     creds = credentials.AnonymousCredentials()
     with mock.patch.object(service_account.Credentials, 'from_service_account_file') as factory:
         factory.return_value = creds
-        client = {{ service.client_name }}.from_service_account_file("dummy/file/path.json")
+        client = client_class.from_service_account_file("dummy/file/path.json")
         assert client._transport._credentials == creds
 
-        client = {{ service.client_name }}.from_service_account_json("dummy/file/path.json")
+        client = client_class.from_service_account_json("dummy/file/path.json")
         assert client._transport._credentials == creds
 
         {% if service.host %}assert client._transport._host == '{{ service.host }}{% if ":" not in service.host %}:443{% endif %}'{% endif %}
 
 
-def test_{{ service.client_name|snake_case }}_client_options():
+@pytest.mark.parametrize("client_class,transport_class", [
+    ({{ service.client_name }}, transports.{{ service.grpc_transport_name }}),
+    ({{ service.async_client_name }}, transports.{{ service.grpc_asyncio_transport_name }})
+])
+def test_{{ service.client_name|snake_case }}_client_options(client_class, transport_class):
     # Check that if channel is provided we won't create a new one.
-    with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.{{ service.client_name }}.get_transport_class') as gtc:
-        transport = transports.{{ service.name }}GrpcTransport(
+    with mock.patch.object(client_class, 'get_transport_class') as gtc:
+        transport = transport_class(
             credentials=credentials.AnonymousCredentials()
         )
-        client = {{ service.client_name }}(transport=transport)
+        client = client_class(transport=transport)
         gtc.assert_not_called()
 
     # Check mTLS is not triggered with empty client options.
     options = client_options.ClientOptions()
-    with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.{{ service.client_name }}.get_transport_class') as gtc:
+    with mock.patch.object(client_class, 'get_transport_class') as gtc:
         transport = gtc.return_value = mock.MagicMock()
-        client = {{ service.client_name }}(client_options=options)
+        client = client_class(client_options=options)
         transport.assert_called_once_with(
             credentials=None,
             host=client.DEFAULT_ENDPOINT,
@@ -86,10 +91,10 @@ def test_{{ service.client_name|snake_case }}_client_options():
     # Check mTLS is not triggered if api_endpoint is provided but
     # client_cert_source is None.
     options = client_options.ClientOptions(api_endpoint="squid.clam.whelk")
-    with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.transports.{{ service.name }}GrpcTransport.__init__') as grpc_transport:
-        grpc_transport.return_value = None
-        client = {{ service.client_name }}(client_options=options)
-        grpc_transport.assert_called_once_with(
+    with mock.patch.object(transport_class, '__init__') as patched:
+        patched.return_value = None
+        client = client_class(client_options=options)
+        patched.assert_called_once_with(
             api_mtls_endpoint=None,
             client_cert_source=None,
             credentials=None,
@@ -100,10 +105,10 @@ def test_{{ service.client_name|snake_case }}_client_options():
     options = client_options.ClientOptions(
         client_cert_source=client_cert_source_callback
     )
-    with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.transports.{{ service.name }}GrpcTransport.__init__') as grpc_transport:
-        grpc_transport.return_value = None
-        client = {{ service.client_name }}(client_options=options)
-        grpc_transport.assert_called_once_with(
+    with mock.patch.object(transport_class, '__init__') as patched:
+        patched.return_value = None
+        client = client_class(client_options=options)
+        patched.assert_called_once_with(
             api_mtls_endpoint=client.DEFAULT_MTLS_ENDPOINT,
             client_cert_source=client_cert_source_callback,
             credentials=None,
@@ -115,16 +120,17 @@ def test_{{ service.client_name|snake_case }}_client_options():
         api_endpoint="squid.clam.whelk",
         client_cert_source=client_cert_source_callback
     )
-    with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.transports.{{ service.name }}GrpcTransport.__init__') as grpc_transport:
-        grpc_transport.return_value = None
-        client = {{ service.client_name }}(client_options=options)
-        grpc_transport.assert_called_once_with(
+    with mock.patch.object(transport_class, '__init__') as patched:
+        patched.return_value = None
+        client = client_class(client_options=options)
+        patched.assert_called_once_with(
             api_mtls_endpoint="squid.clam.whelk",
             client_cert_source=client_cert_source_callback,
             credentials=None,
             host="squid.clam.whelk",
         )
 
+
 def test_{{ service.client_name|snake_case }}_client_options_from_dict():
     with mock.patch('{{ (api.naming.module_namespace + (api.naming.versioned_module_name,) + service.meta.address.subpackage)|join(".") }}.services.{{ service.name|snake_case }}.transports.{{ service.name }}GrpcTransport.__init__') as grpc_transport:
         grpc_transport.return_value = None