add async rest tests

Author: parthea

tests/system/conftest.py

@@ -45,6 +45,7 @@
     try:
         from google.showcase_v1beta1.services.echo.transports import (
             AsyncEchoRestTransport,
+            AsyncEchoRestInterceptor,
         )
 
         HAS_ASYNC_REST_ECHO_TRANSPORT = True
@@ -272,6 +273,28 @@ def post_expand_with_metadata(self, request, metadata):
         return request, metadata
 
 
+if HAS_ASYNC_REST_ECHO_TRANSPORT:
+    class EchoMetadataClientRestAsyncInterceptor(AsyncEchoRestInterceptor):
+        request_metadata: Sequence[Tuple[str, str]] = []
+        response_metadata: Sequence[Tuple[str, str]] = []
+
+        async def pre_echo(self, request, metadata):
+            self.request_metadata = metadata
+            return request, metadata
+
+        async def post_echo_with_metadata(self, request, metadata):
+            self.response_metadata = metadata
+            return request, metadata
+
+        async def pre_expand(self, request, metadata):
+            self.request_metadata = metadata
+            return request, metadata
+
+        async def post_expand_with_metadata(self, request, metadata):
+            self.response_metadata = metadata
+            return request, metadata
+
+
 class EchoMetadataClientGrpcInterceptor(
     grpc.UnaryUnaryClientInterceptor,
     grpc.UnaryStreamClientInterceptor,
@@ -401,6 +424,7 @@ def intercepted_echo_grpc_async():
     )
     return EchoAsyncClient(transport=transport), interceptor
 
+
 @pytest.fixture
 def intercepted_echo_rest():
     transport_name = "rest"
@@ -415,3 +439,22 @@ def intercepted_echo_rest():
         interceptor=interceptor,
     )
     return EchoClient(transport=transport), interceptor
+
+
+@pytest.fixture
+def intercepted_echo_rest_async():
+    if not HAS_ASYNC_REST_ECHO_TRANSPORT:
+        pytest.skip("Skipping test with async rest.")
+
+    transport_name = "rest_asyncio"
+    transport_cls = EchoAsyncClient.get_transport_class(transport_name)
+    interceptor = EchoMetadataClientRestAsyncInterceptor()
+
+    # The custom host explicitly bypasses https.
+    transport = transport_cls(
+        credentials=async_anonymous_credentials(),
+        host="localhost:7469",
+        url_scheme="http",
+        interceptor=interceptor,
+    )
+    return EchoAsyncClient(transport=transport), interceptor

tests/system/test_grpc_interceptor_streams.py

@@ -23,14 +23,14 @@
 
 def test_unary_stream(intercepted_echo_grpc):
     client, interceptor = intercepted_echo_grpc
-    content = 'The hail in Wales falls mainly on the snails.'
+    content = "The hail in Wales falls mainly on the snails."
     responses = client.expand(
         {
             "content": content,
         }
     )
 
-    for ground_truth, response in zip(content.split(' '), responses):
+    for ground_truth, response in zip(content.split(" "), responses):
         assert response.content == ground_truth
     assert ground_truth == "snails."
 

tests/system/test_response_metadata.py

@@ -62,15 +62,45 @@ def test_metadata_response_rest_streams(intercepted_echo_rest):
 if os.environ.get("GAPIC_PYTHON_ASYNC", "true") == "true":
 
     @pytest.mark.asyncio
-    async def test_metadata_response_grpc_unary_async(intercepted_echo_grpc_async):
+    async def test_metadata_response_rest_streams_async(intercepted_echo_rest_async):
+        request_content = "The hail in Wales falls mainly on the snails."
+        request_metadata = ("something2", "something_value2")
+        response_metadata = ("X-Showcase-Request-Something2", "something_value2")
+        client, interceptor = intercepted_echo_rest_async
+        await client.expand(
+            {
+                "content": request_content,
+            },
+            metadata=(request_metadata,),
+        )
+
+        assert request_metadata in interceptor.request_metadata
+        assert response_metadata in interceptor.response_metadata
+
+    @pytest.mark.parametrize(
+        "transport,response_metadata",
+        [
+            ("grpc_asyncio", ("something3", "something_value3")),
+            ("rest_asyncio", ("X-Showcase-Request-Something3", "something_value3")),
+        ],
+    )
+    @pytest.mark.asyncio
+    async def test_metadata_response_unary_async(
+        intercepted_echo_grpc_async,
+        intercepted_echo_rest_async,
+        transport,
+        response_metadata,
+    ):
         request_content = "The hail in Wales falls mainly on the snails."
         request_metadata = ("something3", "something_value3")
-        response_metadata = ("something3", "something_value3")
 
-        client, interceptor = intercepted_echo_grpc_async
+        if transport == "grpc_asyncio":
+            client, interceptor = intercepted_echo_grpc_async
+        else:
+            client, interceptor = intercepted_echo_rest_async
         response = await client.echo(
             request=showcase.EchoRequest(content=request_content),
-            metadata=(("something3", "something_value3"),),
+            metadata=(request_metadata,),
         )
         assert response.content == request_content
         assert request_metadata in interceptor.request_metadata