fix: add context manager return types (#1468)

Co-authored-by: Anthonios Partheniou <partheniou@google.com>

Author: henribru

packages/gapic-generator/gapic/ads-templates/%namespace/%name/%version/%sub/services/%service/client.py.j2

@@ -175,7 +175,7 @@ class {{ service.client_name }}(metaclass={{ service.client_name }}Meta):
         """
         return self._transport
 
-    def __enter__(self):
+    def __enter__(self) -> "{{ service.client_name }}":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/gapic/templates/%namespace/%name_%version/%sub/services/%service/client.py.j2

@@ -379,7 +379,7 @@ class {{ service.client_name }}(metaclass={{ service.client_name }}Meta):
 {% endif %}
     {% endfor %}
 
-    def __enter__(self):
+    def __enter__(self) -> "{{ service.client_name }}":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/asset/google/cloud/asset_v1/services/asset_service/client.py

@@ -1833,7 +1833,7 @@ def sample_analyze_iam_policy_longrunning():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "AssetServiceClient":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/credentials/google/iam/credentials_v1/services/iam_credentials/client.py

@@ -958,7 +958,7 @@ def sample_sign_jwt():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "IAMCredentialsClient":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/eventarc/google/cloud/eventarc_v1/services/eventarc/client.py

@@ -1034,7 +1034,7 @@ def sample_delete_trigger():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "EventarcClient":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/logging/google/cloud/logging_v2/services/config_service_v2/client.py

@@ -2831,7 +2831,7 @@ def sample_update_cmek_settings():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "ConfigServiceV2Client":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/logging/google/cloud/logging_v2/services/logging_service_v2/client.py

@@ -1113,7 +1113,7 @@ def request_generator():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "LoggingServiceV2Client":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/logging/google/cloud/logging_v2/services/metrics_service_v2/client.py

@@ -961,7 +961,7 @@ def sample_delete_log_metric():
             metadata=metadata,
         )
 
-    def __enter__(self):
+    def __enter__(self) -> "MetricsServiceV2Client":
         return self
 
     def __exit__(self, type, value, traceback):

packages/gapic-generator/tests/integration/goldens/redis/google/cloud/redis_v1/services/cloud_redis/client.py

@@ -1596,7 +1596,7 @@ def sample_delete_instance():
         # Done; return the response.
         return response
 
-    def __enter__(self):
+    def __enter__(self) -> "CloudRedisClient":
         return self
 
     def __exit__(self, type, value, traceback):