test: add async REST URL query params encoding tests and update goldens

Add async REST transport versions of URL query params encoding tests
to achieve 100% code coverage for rest_asyncio.py. Update integration
test golden files to reflect the $alt query parameter encoding fix.

Author: taimo3810

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

@@ -2171,4 +2171,369 @@ def test_test_iam_permissions_rest_url_query_params_encoding():
 
 {% endif %} {# iam_mixin #}
 
+{% if rest_async_io_enabled %}
+{# Async REST URL encoding tests for mixin methods #}
+
+{% if api.has_operations_mixin %}
+
+{% if "ListOperations" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_list_operations_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.list_operations.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.get.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/test/operations',
+            'method': 'get',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.get.called
+        call_url = mock_session.get.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "GetOperation" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_get_operation_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.get_operation.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.get.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/test/operations/op1',
+            'method': 'get',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.get.called
+        call_url = mock_session.get.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "DeleteOperation" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_delete_operation_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.delete_operation.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.delete.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/test/operations/op1',
+            'method': 'delete',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.delete.called
+        call_url = mock_session.delete.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "CancelOperation" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_cancel_operation_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.cancel_operation.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.post.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/test/operations/op1:cancel',
+            'method': 'post',
+            'body': {},
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+            body={},
+        )
+
+        assert mock_session.post.called
+        call_url = mock_session.post.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% endif %} {# operations_mixin #}
+
+{% if api.has_location_mixin %}
+
+{% if "ListLocations" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_list_locations_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.list_locations.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.get.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/projects/p1/locations',
+            'method': 'get',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.get.called
+        call_url = mock_session.get.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "GetLocation" in api.mixin_api_methods %}
+@pytest.mark.asyncio
+async def test_get_location_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.get_location.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.get.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/projects/p1/locations/l1',
+            'method': 'get',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.get.called
+        call_url = mock_session.get.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% endif %} {# location_mixin #}
+
+{% if api.has_iam_mixin or opts.add_iam_methods %}
+
+{% if "SetIamPolicy" in api.mixin_api_methods or opts.add_iam_methods %}
+@pytest.mark.asyncio
+async def test_set_iam_policy_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.set_iam_policy.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.post.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/resource:setIamPolicy',
+            'method': 'post',
+            'body': {},
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+            body={},
+        )
+
+        assert mock_session.post.called
+        call_url = mock_session.post.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "GetIamPolicy" in api.mixin_api_methods or opts.add_iam_methods %}
+@pytest.mark.asyncio
+async def test_get_iam_policy_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.get_iam_policy.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.get.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/resource:getIamPolicy',
+            'method': 'get',
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+        )
+
+        assert mock_session.get.called
+        call_url = mock_session.get.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% if "TestIamPermissions" in api.mixin_api_methods or opts.add_iam_methods %}
+@pytest.mark.asyncio
+async def test_test_iam_permissions_rest_asyncio_url_query_params_encoding():
+    if not HAS_ASYNC_REST_EXTRA:
+        pytest.skip("the library must be installed with the `async_rest` extra to test this feature.")
+    transport = transports.Async{{ service.name }}RestTransport(credentials=async_anonymous_credentials())
+    method_class = transport.test_iam_permissions.__class__
+    get_response_fn = method_class._get_response
+
+    mock_session = mock.AsyncMock()
+    mock_response = mock.Mock()
+    mock_response.status_code = 200
+    mock_session.post.return_value = mock_response
+
+    with mock.patch.object(rest_helpers, 'flatten_query_params') as mock_flatten:
+        mock_flatten.return_value = [('$alt', 'json;enum-encoding=int'), ('foo', 'bar')]
+
+        transcoded_request = {
+            'uri': '/v1/resource:testIamPermissions',
+            'method': 'post',
+            'body': {},
+        }
+
+        await get_response_fn(
+            host='https://example.com',
+            metadata=[],
+            query_params={},
+            session=mock_session,
+            timeout=None,
+            transcoded_request=transcoded_request,
+            body={},
+        )
+
+        assert mock_session.post.called
+        call_url = mock_session.post.call_args.args[0]
+        assert '$alt=json' in call_url
+        assert '%24alt' not in call_url
+        assert 'foo=bar' in call_url
+{% endif %}
+
+{% endif %} {# iam_mixin #}
+
+{% endif %} {# rest_async_io_enabled #}
+
 {% endif %} {# rest transport #}

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

@@ -1118,10 +1118,10 @@ def test_{{ service.client_name|snake_case }}_create_channel_credentials_file(cl
 
 {% for method in service.methods.values() if 'rest' in opts.transport %}
 {% if method.extended_lro %}
-{{ test_macros.rest_required_tests(method, service, numeric_enums=opts.rest_numeric_enums, full_extended_lro=True) }}
+{{ test_macros.rest_required_tests(method, service, numeric_enums=opts.rest_numeric_enums, full_extended_lro=True, rest_async_io_enabled=rest_async_io_enabled) }}
 
 {% endif %}
-{{ test_macros.rest_required_tests(method, service, numeric_enums=opts.rest_numeric_enums) }}
+{{ test_macros.rest_required_tests(method, service, numeric_enums=opts.rest_numeric_enums, rest_async_io_enabled=rest_async_io_enabled) }}
 
 {% endfor -%} {#- method in methods for rest #}