fix: add async context manager return types

gapic/templates/%namespace/%name_%version/%sub/services/%service/async_client.py.j2

@@ -672,7 +672,7 @@ class {{ service.async_client_name }}:
         return response
     {% endif %}
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "{{ service.async_client_name }}":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/asset/google/cloud/asset_v1/services/asset_service/async_client.py

@@ -1674,7 +1674,7 @@ async def sample_analyze_iam_policy_longrunning():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "AssetServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/credentials/google/iam/credentials_v1/services/iam_credentials/async_client.py

@@ -792,7 +792,7 @@ async def sample_sign_jwt():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "IAMCredentialsAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/eventarc/google/cloud/eventarc_v1/services/eventarc/async_client.py

@@ -825,7 +825,7 @@ async def sample_delete_trigger():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "EventarcAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/logging/google/cloud/logging_v2/services/config_service_v2/async_client.py

@@ -2648,7 +2648,7 @@ async def sample_update_cmek_settings():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ConfigServiceV2AsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/logging/google/cloud/logging_v2/services/logging_service_v2/async_client.py

@@ -971,7 +971,7 @@ def request_generator():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "LoggingServiceV2AsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/logging/google/cloud/logging_v2/services/metrics_service_v2/async_client.py

@@ -799,7 +799,7 @@ async def sample_delete_log_metric():
             metadata=metadata,
         )
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MetricsServiceV2AsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

tests/integration/goldens/redis/google/cloud/redis_v1/services/cloud_redis/async_client.py

@@ -1399,7 +1399,7 @@ async def sample_delete_instance():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "CloudRedisAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):