@@ -238,7 +238,7 @@ async def test_{{ method_name }}_async_use_cached_wrapped_rpc(transport: str = "
238238
239239 {% if method .client_streaming %}
240240 request = [{}]
241- await client.{{ method.name |snake_case }}(request)
241+ await client.{{ method.safe_name |snake_case }}(request)
242242 {% else %}
243243 request = {}
244244 await client.{{ method_name }}(request)
@@ -255,7 +255,7 @@ async def test_{{ method_name }}_async_use_cached_wrapped_rpc(transport: str = "
255255 {% endif %}
256256
257257 {% if method .client_streaming %}
258- await client.{{ method.name |snake_case }}(request)
258+ await client.{{ method.safe_name |snake_case }}(request)
259259 {% else %}
260260 await client.{{ method_name }}(request)
261261 {% endif %}
@@ -321,9 +321,9 @@ async def test_{{ method_name }}_async(transport: str = 'grpc_asyncio', request_
321321 ))
322322 {% endif %}
323323 {% if method .client_streaming and method .server_streaming %}
324- response = await client.{{ method.name |snake_case }}(iter(requests))
324+ response = await client.{{ method.safe_name |snake_case }}(iter(requests))
325325 {% elif method .client_streaming and not method .server_streaming %}
326- response = await (await client.{{ method.name |snake_case }}(iter(requests)))
326+ response = await (await client.{{ method.safe_name |snake_case }}(iter(requests)))
327327 {% else %}
328328 response = await client.{{ method_name }}(request)
329329 {% endif %}
@@ -1001,7 +1001,7 @@ def test_{{ method_name }}_raw_page_lro():
10011001{% endmacro %}
10021002
10031003{% macro rest_required_tests (method , service , numeric_enums =False , full_extended_lro =False ) %}
1004- {% with method_name = method .safe_name |snake_case + "_unary" if method .extended_lro and not full_extended_lro else method .name |snake_case , method_output = method .extended_lro .operation_type if method .extended_lro and not full_extended_lro else method .output %}{% if method .http_options %}
1004+ {% with method_name = method .safe_name |snake_case + "_unary" if method .extended_lro and not full_extended_lro else method .safe_name |snake_case , method_output = method .extended_lro .operation_type if method .extended_lro and not full_extended_lro else method .output %}{% if method .http_options %}
10051005{# TODO(kbandes): remove this if condition when lro and client streaming are supported. #}
10061006{% if not method .client_streaming %}
10071007def test_{{ method_name }}_rest_use_cached_wrapped_rpc():
@@ -1460,7 +1460,7 @@ def test_{{ method_name }}_rest_no_http_options():
14601460#}
14611461{% macro method_call_test_generic (test_name , method , service , api , transport , request_dict , is_async =False , routing_param =None ) %}
14621462{% set transport_name = get_transport_name (transport , is_async ) %}
1463- {% with method_name = ( method .name + ( "_unary" if method .operation_service else "" )) | snake_case %}
1463+ {% with method_name = method .name | snake_case + "_unary" if method .operation_service else method . safe_name | snake_case %}
14641464{% set async_method_prefix = "async " if is_async else "" %}
14651465{% if is_async %}
14661466@pytest.mark.asyncio
@@ -1713,7 +1713,7 @@ def test_unsupported_parameter_rest_asyncio():
17131713{% set async_prefix = get_async_prefix (is_async ) %}
17141714{% set async_decorator = get_async_decorator (is_async ) %}
17151715{% set transport_name = get_transport_name (transport , is_async ) %}
1716- {% set method_name = method .name |snake_case %}
1716+ {% set method_name = method .safe_name |snake_case %}
17171717{{async_decorator}}
17181718{{async_prefix}}def test_{{ method_name }}_{{transport_name}}_error():
17191719 {% if transport_name == 'rest_asyncio' %}
@@ -1763,7 +1763,7 @@ def test_initialize_client_w_{{transport_name}}():
17631763{% set async_prefix = get_async_prefix (is_async ) %}
17641764{% set async_decorator = get_async_decorator (is_async ) %}
17651765{% set transport_name = get_transport_name (transport , is_async ) %}
1766- {% set method_name = method .name |snake_case %}
1766+ {% set method_name = method .safe_name |snake_case %}
17671767{% set mocked_session = "AsyncAuthorizedSession" if is_async else "Session" %}
17681768{{ async_decorator }}
17691769{{ async_prefix }}def test_{{ method_name }}_{{transport_name}}_bad_request(request_type={{ method.input.ident }}):
@@ -1862,7 +1862,7 @@ def test_initialize_client_w_{{transport_name}}():
18621862{% set async_prefix = get_async_prefix (is_async ) %}
18631863{% set async_decorator = get_async_decorator (is_async ) %}
18641864{% set transport_name = get_transport_name (transport , is_async ) %}
1865- {% set method_name = method .name |snake_case %}
1865+ {% set method_name = method .safe_name |snake_case %}
18661866{# NOTE: set method_output to method.extended_lro.operation_type for the following method types:
18671867 # (method.extended_lro and not full_extended_lro)
18681868#}
@@ -2183,7 +2183,7 @@ def test_initialize_client_w_{{transport_name}}():
21832183{% set async_prefix = get_async_prefix (is_async ) %}
21842184{% set async_decorator = get_async_decorator (is_async ) %}
21852185{% set transport_name = get_transport_name (transport , is_async ) %}
2186- {% set method_name = method .name |snake_case %}
2186+ {% set method_name = method .safe_name |snake_case %}
21872187{% set async_method_prefix = "Async" if is_async else "" %}
21882188{{async_decorator}}
21892189@pytest.mark.parametrize("null_interceptor", [True, False])
0 commit comments