diff --git a/google/cloud/appengine_admin_v1/services/applications/async_client.py b/google/cloud/appengine_admin_v1/services/applications/async_client.py index 29ffd86..d7535b3 100644 --- a/google/cloud/appengine_admin_v1/services/applications/async_client.py +++ b/google/cloud/appengine_admin_v1/services/applications/async_client.py @@ -406,6 +406,12 @@ async def repair_application( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/applications/client.py b/google/cloud/appengine_admin_v1/services/applications/client.py index b43da0c..79ad4fc 100644 --- a/google/cloud/appengine_admin_v1/services/applications/client.py +++ b/google/cloud/appengine_admin_v1/services/applications/client.py @@ -330,10 +330,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def get_application( @@ -587,6 +584,19 @@ def repair_application( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/applications/transports/base.py b/google/cloud/appengine_admin_v1/services/applications/transports/base.py index 91338b9..5aaa063 100644 --- a/google/cloud/appengine_admin_v1/services/applications/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/applications/transports/base.py @@ -175,6 +175,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def operations_client(self) -> operations_v1.OperationsClient: """Return the client designed to process long-running operations.""" diff --git a/google/cloud/appengine_admin_v1/services/applications/transports/grpc.py b/google/cloud/appengine_admin_v1/services/applications/transports/grpc.py index 7e79a41..352667e 100644 --- a/google/cloud/appengine_admin_v1/services/applications/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/applications/transports/grpc.py @@ -373,5 +373,8 @@ def repair_application( ) return self._stubs["repair_application"] + def close(self): + self.grpc_channel.close() + __all__ = ("ApplicationsGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/applications/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/applications/transports/grpc_asyncio.py index 1a03ec4..7c448e9 100644 --- a/google/cloud/appengine_admin_v1/services/applications/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/applications/transports/grpc_asyncio.py @@ -386,5 +386,8 @@ def repair_application( ) return self._stubs["repair_application"] + def close(self): + return self.grpc_channel.close() + __all__ = ("ApplicationsGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/authorized_certificates/async_client.py b/google/cloud/appengine_admin_v1/services/authorized_certificates/async_client.py index 1110e38..b10fbf9 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_certificates/async_client.py +++ b/google/cloud/appengine_admin_v1/services/authorized_certificates/async_client.py @@ -429,6 +429,12 @@ async def delete_authorized_certificate( request, retry=retry, timeout=timeout, metadata=metadata, ) + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/authorized_certificates/client.py b/google/cloud/appengine_admin_v1/services/authorized_certificates/client.py index 02be3a5..940b3ca 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_certificates/client.py +++ b/google/cloud/appengine_admin_v1/services/authorized_certificates/client.py @@ -335,10 +335,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_authorized_certificates( @@ -618,6 +615,19 @@ def delete_authorized_certificate( request, retry=retry, timeout=timeout, metadata=metadata, ) + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/base.py b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/base.py index 1ae962d..c467d95 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/base.py @@ -187,6 +187,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def list_authorized_certificates( self, diff --git a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc.py b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc.py index 8f86bf8..58098d4 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc.py @@ -382,5 +382,8 @@ def delete_authorized_certificate( ) return self._stubs["delete_authorized_certificate"] + def close(self): + self.grpc_channel.close() + __all__ = ("AuthorizedCertificatesGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc_asyncio.py index 9b0d7ea..5999184 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/authorized_certificates/transports/grpc_asyncio.py @@ -388,5 +388,8 @@ def delete_authorized_certificate( ) return self._stubs["delete_authorized_certificate"] + def close(self): + return self.grpc_channel.close() + __all__ = ("AuthorizedCertificatesGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/authorized_domains/async_client.py b/google/cloud/appengine_admin_v1/services/authorized_domains/async_client.py index cba0655..5665df4 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_domains/async_client.py +++ b/google/cloud/appengine_admin_v1/services/authorized_domains/async_client.py @@ -222,6 +222,12 @@ async def list_authorized_domains( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/authorized_domains/client.py b/google/cloud/appengine_admin_v1/services/authorized_domains/client.py index 4d16281..bffcce0 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_domains/client.py +++ b/google/cloud/appengine_admin_v1/services/authorized_domains/client.py @@ -334,10 +334,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_authorized_domains( @@ -400,6 +397,19 @@ def list_authorized_domains( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/base.py b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/base.py index 22ebc61..fd3ab73 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/base.py @@ -165,6 +165,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def list_authorized_domains( self, diff --git a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc.py b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc.py index 5095922..34be2b1 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc.py @@ -257,5 +257,8 @@ def list_authorized_domains( ) return self._stubs["list_authorized_domains"] + def close(self): + self.grpc_channel.close() + __all__ = ("AuthorizedDomainsGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc_asyncio.py index f06afc5..97dd4da 100644 --- a/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/authorized_domains/transports/grpc_asyncio.py @@ -260,5 +260,8 @@ def list_authorized_domains( ) return self._stubs["list_authorized_domains"] + def close(self): + return self.grpc_channel.close() + __all__ = ("AuthorizedDomainsGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/domain_mappings/async_client.py b/google/cloud/appengine_admin_v1/services/domain_mappings/async_client.py index 7bd11b6..2d07ccc 100644 --- a/google/cloud/appengine_admin_v1/services/domain_mappings/async_client.py +++ b/google/cloud/appengine_admin_v1/services/domain_mappings/async_client.py @@ -469,6 +469,12 @@ async def delete_domain_mapping( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/domain_mappings/client.py b/google/cloud/appengine_admin_v1/services/domain_mappings/client.py index a3c7a4c..5182ceb 100644 --- a/google/cloud/appengine_admin_v1/services/domain_mappings/client.py +++ b/google/cloud/appengine_admin_v1/services/domain_mappings/client.py @@ -333,10 +333,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_domain_mappings( @@ -649,6 +646,19 @@ def delete_domain_mapping( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/base.py b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/base.py index b11cc15..9c8a596 100644 --- a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/base.py @@ -186,6 +186,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def operations_client(self) -> operations_v1.OperationsClient: """Return the client designed to process long-running operations.""" diff --git a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc.py b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc.py index 1f89b60..5eaac3e 100644 --- a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc.py @@ -384,5 +384,8 @@ def delete_domain_mapping( ) return self._stubs["delete_domain_mapping"] + def close(self): + self.grpc_channel.close() + __all__ = ("DomainMappingsGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc_asyncio.py index 46bb973..9ba3f67 100644 --- a/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/domain_mappings/transports/grpc_asyncio.py @@ -398,5 +398,8 @@ def delete_domain_mapping( ) return self._stubs["delete_domain_mapping"] + def close(self): + return self.grpc_channel.close() + __all__ = ("DomainMappingsGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/firewall/async_client.py b/google/cloud/appengine_admin_v1/services/firewall/async_client.py index fe8786f..69b4196 100644 --- a/google/cloud/appengine_admin_v1/services/firewall/async_client.py +++ b/google/cloud/appengine_admin_v1/services/firewall/async_client.py @@ -468,6 +468,12 @@ async def delete_ingress_rule( request, retry=retry, timeout=timeout, metadata=metadata, ) + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/firewall/client.py b/google/cloud/appengine_admin_v1/services/firewall/client.py index b15645d..6e5ef6c 100644 --- a/google/cloud/appengine_admin_v1/services/firewall/client.py +++ b/google/cloud/appengine_admin_v1/services/firewall/client.py @@ -338,10 +338,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_ingress_rules( @@ -657,6 +654,19 @@ def delete_ingress_rule( request, retry=retry, timeout=timeout, metadata=metadata, ) + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/firewall/transports/base.py b/google/cloud/appengine_admin_v1/services/firewall/transports/base.py index 04b20ea..a9bcf9c 100644 --- a/google/cloud/appengine_admin_v1/services/firewall/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/firewall/transports/base.py @@ -182,6 +182,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def list_ingress_rules( self, diff --git a/google/cloud/appengine_admin_v1/services/firewall/transports/grpc.py b/google/cloud/appengine_admin_v1/services/firewall/transports/grpc.py index 1809e0b..7dc0324 100644 --- a/google/cloud/appengine_admin_v1/services/firewall/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/firewall/transports/grpc.py @@ -404,5 +404,8 @@ def delete_ingress_rule( ) return self._stubs["delete_ingress_rule"] + def close(self): + self.grpc_channel.close() + __all__ = ("FirewallGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/firewall/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/firewall/transports/grpc_asyncio.py index 75a7250..ab20af5 100644 --- a/google/cloud/appengine_admin_v1/services/firewall/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/firewall/transports/grpc_asyncio.py @@ -412,5 +412,8 @@ def delete_ingress_rule( ) return self._stubs["delete_ingress_rule"] + def close(self): + return self.grpc_channel.close() + __all__ = ("FirewallGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/instances/async_client.py b/google/cloud/appengine_admin_v1/services/instances/async_client.py index 9033e01..ec2e849 100644 --- a/google/cloud/appengine_admin_v1/services/instances/async_client.py +++ b/google/cloud/appengine_admin_v1/services/instances/async_client.py @@ -424,6 +424,12 @@ async def debug_instance( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/instances/client.py b/google/cloud/appengine_admin_v1/services/instances/client.py index aa6f6f5..d937126 100644 --- a/google/cloud/appengine_admin_v1/services/instances/client.py +++ b/google/cloud/appengine_admin_v1/services/instances/client.py @@ -348,10 +348,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_instances( @@ -621,6 +618,19 @@ def debug_instance( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/instances/transports/base.py b/google/cloud/appengine_admin_v1/services/instances/transports/base.py index af17f3d..6903d56 100644 --- a/google/cloud/appengine_admin_v1/services/instances/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/instances/transports/base.py @@ -175,6 +175,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def operations_client(self) -> operations_v1.OperationsClient: """Return the client designed to process long-running operations.""" diff --git a/google/cloud/appengine_admin_v1/services/instances/transports/grpc.py b/google/cloud/appengine_admin_v1/services/instances/transports/grpc.py index 7252c39..2fa6b60 100644 --- a/google/cloud/appengine_admin_v1/services/instances/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/instances/transports/grpc.py @@ -374,5 +374,8 @@ def debug_instance( ) return self._stubs["debug_instance"] + def close(self): + self.grpc_channel.close() + __all__ = ("InstancesGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/instances/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/instances/transports/grpc_asyncio.py index 4825fdc..5f84080 100644 --- a/google/cloud/appengine_admin_v1/services/instances/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/instances/transports/grpc_asyncio.py @@ -385,5 +385,8 @@ def debug_instance( ) return self._stubs["debug_instance"] + def close(self): + return self.grpc_channel.close() + __all__ = ("InstancesGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/services/async_client.py b/google/cloud/appengine_admin_v1/services/services/async_client.py index 56ef3f9..5880026 100644 --- a/google/cloud/appengine_admin_v1/services/services/async_client.py +++ b/google/cloud/appengine_admin_v1/services/services/async_client.py @@ -411,6 +411,12 @@ async def delete_service( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/services/client.py b/google/cloud/appengine_admin_v1/services/services/client.py index bc61a0a..dcaecca 100644 --- a/google/cloud/appengine_admin_v1/services/services/client.py +++ b/google/cloud/appengine_admin_v1/services/services/client.py @@ -332,10 +332,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_services( @@ -596,6 +593,19 @@ def delete_service( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/services/transports/base.py b/google/cloud/appengine_admin_v1/services/services/transports/base.py index a276406..2a637c9 100644 --- a/google/cloud/appengine_admin_v1/services/services/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/services/transports/base.py @@ -175,6 +175,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def operations_client(self) -> operations_v1.OperationsClient: """Return the client designed to process long-running operations.""" diff --git a/google/cloud/appengine_admin_v1/services/services/transports/grpc.py b/google/cloud/appengine_admin_v1/services/services/transports/grpc.py index 6225fc5..e4df27f 100644 --- a/google/cloud/appengine_admin_v1/services/services/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/services/transports/grpc.py @@ -347,5 +347,8 @@ def delete_service( ) return self._stubs["delete_service"] + def close(self): + self.grpc_channel.close() + __all__ = ("ServicesGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/services/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/services/transports/grpc_asyncio.py index b479e83..af666f0 100644 --- a/google/cloud/appengine_admin_v1/services/services/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/services/transports/grpc_asyncio.py @@ -360,5 +360,8 @@ def delete_service( ) return self._stubs["delete_service"] + def close(self): + return self.grpc_channel.close() + __all__ = ("ServicesGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/services/versions/async_client.py b/google/cloud/appengine_admin_v1/services/versions/async_client.py index 3d1a81e..f0a057f 100644 --- a/google/cloud/appengine_admin_v1/services/versions/async_client.py +++ b/google/cloud/appengine_admin_v1/services/versions/async_client.py @@ -492,6 +492,12 @@ async def delete_version( # Done; return the response. return response + async def __aenter__(self): + return self + + async def __aexit__(self, exc_type, exc, tb): + await self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/versions/client.py b/google/cloud/appengine_admin_v1/services/versions/client.py index 2d3fe2d..a8a5cb4 100644 --- a/google/cloud/appengine_admin_v1/services/versions/client.py +++ b/google/cloud/appengine_admin_v1/services/versions/client.py @@ -335,10 +335,7 @@ def __init__( client_cert_source_for_mtls=client_cert_source_func, quota_project_id=client_options.quota_project_id, client_info=client_info, - always_use_jwt_access=( - Transport == type(self).get_transport_class("grpc") - or Transport == type(self).get_transport_class("grpc_asyncio") - ), + always_use_jwt_access=True, ) def list_versions( @@ -678,6 +675,19 @@ def delete_version( # Done; return the response. return response + def __enter__(self): + return self + + def __exit__(self, type, value, traceback): + """Releases underlying transport's resources. + + .. warning:: + ONLY use as a context manager if the transport is NOT shared + with other clients! Exiting the with block will CLOSE the transport + and may cause errors in other clients! + """ + self.transport.close() + try: DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( diff --git a/google/cloud/appengine_admin_v1/services/versions/transports/base.py b/google/cloud/appengine_admin_v1/services/versions/transports/base.py index 03f11cd..883d0eb 100644 --- a/google/cloud/appengine_admin_v1/services/versions/transports/base.py +++ b/google/cloud/appengine_admin_v1/services/versions/transports/base.py @@ -178,6 +178,15 @@ def _prep_wrapped_messages(self, client_info): ), } + def close(self): + """Closes resources associated with the transport. + + .. warning:: + Only call this method if the transport is NOT shared + with other clients - this may cause errors in other clients! + """ + raise NotImplementedError() + @property def operations_client(self) -> operations_v1.OperationsClient: """Return the client designed to process long-running operations.""" diff --git a/google/cloud/appengine_admin_v1/services/versions/transports/grpc.py b/google/cloud/appengine_admin_v1/services/versions/transports/grpc.py index 9abe8a4..8b61c61 100644 --- a/google/cloud/appengine_admin_v1/services/versions/transports/grpc.py +++ b/google/cloud/appengine_admin_v1/services/versions/transports/grpc.py @@ -408,5 +408,8 @@ def delete_version( ) return self._stubs["delete_version"] + def close(self): + self.grpc_channel.close() + __all__ = ("VersionsGrpcTransport",) diff --git a/google/cloud/appengine_admin_v1/services/versions/transports/grpc_asyncio.py b/google/cloud/appengine_admin_v1/services/versions/transports/grpc_asyncio.py index 52a24e5..1ada908 100644 --- a/google/cloud/appengine_admin_v1/services/versions/transports/grpc_asyncio.py +++ b/google/cloud/appengine_admin_v1/services/versions/transports/grpc_asyncio.py @@ -423,5 +423,8 @@ def delete_version( ) return self._stubs["delete_version"] + def close(self): + return self.grpc_channel.close() + __all__ = ("VersionsGrpcAsyncIOTransport",) diff --git a/google/cloud/appengine_admin_v1/types/app_yaml.py b/google/cloud/appengine_admin_v1/types/app_yaml.py index 9c7f879..f6acd01 100644 --- a/google/cloud/appengine_admin_v1/types/app_yaml.py +++ b/google/cloud/appengine_admin_v1/types/app_yaml.py @@ -93,6 +93,7 @@ class ApiConfigHandler(proto.Message): class ErrorHandler(proto.Message): r"""Custom static error page to be served when an error occurs. + Attributes: error_code (google.cloud.appengine_admin_v1.types.ErrorHandler.ErrorCode): Error condition this handler applies to. @@ -246,6 +247,7 @@ class ScriptHandler(proto.Message): class ApiEndpointHandler(proto.Message): r"""Uses Google Cloud Endpoints to handle requests. + Attributes: script_path (str): Path to the script from the application root diff --git a/google/cloud/appengine_admin_v1/types/appengine.py b/google/cloud/appengine_admin_v1/types/appengine.py index e50ef77..3c04ecd 100644 --- a/google/cloud/appengine_admin_v1/types/appengine.py +++ b/google/cloud/appengine_admin_v1/types/appengine.py @@ -103,6 +103,7 @@ class DomainOverrideStrategy(proto.Enum): class GetApplicationRequest(proto.Message): r"""Request message for ``Applications.GetApplication``. + Attributes: name (str): Name of the Application resource to get. Example: @@ -114,6 +115,7 @@ class GetApplicationRequest(proto.Message): class CreateApplicationRequest(proto.Message): r"""Request message for ``Applications.CreateApplication``. + Attributes: application (google.cloud.appengine_admin_v1.types.Application): Application configuration. @@ -126,6 +128,7 @@ class CreateApplicationRequest(proto.Message): class UpdateApplicationRequest(proto.Message): r"""Request message for ``Applications.UpdateApplication``. + Attributes: name (str): Name of the Application resource to update. Example: @@ -149,6 +152,7 @@ class UpdateApplicationRequest(proto.Message): class RepairApplicationRequest(proto.Message): r"""Request message for 'Applications.RepairApplication'. + Attributes: name (str): Name of the application to repair. Example: ``apps/myapp`` @@ -159,6 +163,7 @@ class RepairApplicationRequest(proto.Message): class ListServicesRequest(proto.Message): r"""Request message for ``Services.ListServices``. + Attributes: parent (str): Name of the parent Application resource. Example: @@ -177,6 +182,7 @@ class ListServicesRequest(proto.Message): class ListServicesResponse(proto.Message): r"""Response message for ``Services.ListServices``. + Attributes: services (Sequence[google.cloud.appengine_admin_v1.types.Service]): The services belonging to the requested @@ -196,6 +202,7 @@ def raw_page(self): class GetServiceRequest(proto.Message): r"""Request message for ``Services.GetService``. + Attributes: name (str): Name of the resource requested. Example: @@ -207,6 +214,7 @@ class GetServiceRequest(proto.Message): class UpdateServiceRequest(proto.Message): r"""Request message for ``Services.UpdateService``. + Attributes: name (str): Name of the resource to update. Example: @@ -245,6 +253,7 @@ class UpdateServiceRequest(proto.Message): class DeleteServiceRequest(proto.Message): r"""Request message for ``Services.DeleteService``. + Attributes: name (str): Name of the resource requested. Example: @@ -256,6 +265,7 @@ class DeleteServiceRequest(proto.Message): class ListVersionsRequest(proto.Message): r"""Request message for ``Versions.ListVersions``. + Attributes: parent (str): Name of the parent Service resource. Example: @@ -278,6 +288,7 @@ class ListVersionsRequest(proto.Message): class ListVersionsResponse(proto.Message): r"""Response message for ``Versions.ListVersions``. + Attributes: versions (Sequence[google.cloud.appengine_admin_v1.types.Version]): The versions belonging to the requested @@ -297,6 +308,7 @@ def raw_page(self): class GetVersionRequest(proto.Message): r"""Request message for ``Versions.GetVersion``. + Attributes: name (str): Name of the resource requested. Example: @@ -311,6 +323,7 @@ class GetVersionRequest(proto.Message): class CreateVersionRequest(proto.Message): r"""Request message for ``Versions.CreateVersion``. + Attributes: parent (str): Name of the parent resource to create this version under. @@ -325,6 +338,7 @@ class CreateVersionRequest(proto.Message): class UpdateVersionRequest(proto.Message): r"""Request message for ``Versions.UpdateVersion``. + Attributes: name (str): Name of the resource to update. Example: @@ -347,6 +361,7 @@ class UpdateVersionRequest(proto.Message): class DeleteVersionRequest(proto.Message): r"""Request message for ``Versions.DeleteVersion``. + Attributes: name (str): Name of the resource requested. Example: @@ -358,6 +373,7 @@ class DeleteVersionRequest(proto.Message): class ListInstancesRequest(proto.Message): r"""Request message for ``Instances.ListInstances``. + Attributes: parent (str): Name of the parent Version resource. Example: @@ -376,6 +392,7 @@ class ListInstancesRequest(proto.Message): class ListInstancesResponse(proto.Message): r"""Response message for ``Instances.ListInstances``. + Attributes: instances (Sequence[google.cloud.appengine_admin_v1.types.Instance]): The instances belonging to the requested @@ -395,6 +412,7 @@ def raw_page(self): class GetInstanceRequest(proto.Message): r"""Request message for ``Instances.GetInstance``. + Attributes: name (str): Name of the resource requested. Example: @@ -406,6 +424,7 @@ class GetInstanceRequest(proto.Message): class DeleteInstanceRequest(proto.Message): r"""Request message for ``Instances.DeleteInstance``. + Attributes: name (str): Name of the resource requested. Example: @@ -417,6 +436,7 @@ class DeleteInstanceRequest(proto.Message): class DebugInstanceRequest(proto.Message): r"""Request message for ``Instances.DebugInstance``. + Attributes: name (str): Name of the resource requested. Example: @@ -437,6 +457,7 @@ class DebugInstanceRequest(proto.Message): class ListIngressRulesRequest(proto.Message): r"""Request message for ``Firewall.ListIngressRules``. + Attributes: parent (str): Name of the Firewall collection to retrieve. Example: @@ -461,6 +482,7 @@ class ListIngressRulesRequest(proto.Message): class ListIngressRulesResponse(proto.Message): r"""Response message for ``Firewall.ListIngressRules``. + Attributes: ingress_rules (Sequence[google.cloud.appengine_admin_v1.types.FirewallRule]): The ingress FirewallRules for this @@ -482,6 +504,7 @@ def raw_page(self): class BatchUpdateIngressRulesRequest(proto.Message): r"""Request message for ``Firewall.BatchUpdateIngressRules``. + Attributes: name (str): Name of the Firewall collection to set. Example: @@ -499,6 +522,7 @@ class BatchUpdateIngressRulesRequest(proto.Message): class BatchUpdateIngressRulesResponse(proto.Message): r"""Response message for ``Firewall.UpdateAllIngressRules``. + Attributes: ingress_rules (Sequence[google.cloud.appengine_admin_v1.types.FirewallRule]): The full list of ingress FirewallRules for @@ -512,6 +536,7 @@ class BatchUpdateIngressRulesResponse(proto.Message): class CreateIngressRuleRequest(proto.Message): r"""Request message for ``Firewall.CreateIngressRule``. + Attributes: parent (str): Name of the parent Firewall collection in which to create a @@ -537,6 +562,7 @@ class CreateIngressRuleRequest(proto.Message): class GetIngressRuleRequest(proto.Message): r"""Request message for ``Firewall.GetIngressRule``. + Attributes: name (str): Name of the Firewall resource to retrieve. Example: @@ -548,6 +574,7 @@ class GetIngressRuleRequest(proto.Message): class UpdateIngressRuleRequest(proto.Message): r"""Request message for ``Firewall.UpdateIngressRule``. + Attributes: name (str): Name of the Firewall resource to update. Example: @@ -569,6 +596,7 @@ class UpdateIngressRuleRequest(proto.Message): class DeleteIngressRuleRequest(proto.Message): r"""Request message for ``Firewall.DeleteIngressRule``. + Attributes: name (str): Name of the Firewall resource to delete. Example: @@ -580,6 +608,7 @@ class DeleteIngressRuleRequest(proto.Message): class ListAuthorizedDomainsRequest(proto.Message): r"""Request message for ``AuthorizedDomains.ListAuthorizedDomains``. + Attributes: parent (str): Name of the parent Application resource. Example: @@ -598,6 +627,7 @@ class ListAuthorizedDomainsRequest(proto.Message): class ListAuthorizedDomainsResponse(proto.Message): r"""Response message for ``AuthorizedDomains.ListAuthorizedDomains``. + Attributes: domains (Sequence[google.cloud.appengine_admin_v1.types.AuthorizedDomain]): The authorized domains belonging to the user. @@ -738,6 +768,7 @@ class DeleteAuthorizedCertificateRequest(proto.Message): class ListDomainMappingsRequest(proto.Message): r"""Request message for ``DomainMappings.ListDomainMappings``. + Attributes: parent (str): Name of the parent Application resource. Example: @@ -756,6 +787,7 @@ class ListDomainMappingsRequest(proto.Message): class ListDomainMappingsResponse(proto.Message): r"""Response message for ``DomainMappings.ListDomainMappings``. + Attributes: domain_mappings (Sequence[google.cloud.appengine_admin_v1.types.DomainMapping]): The domain mappings for the application. @@ -776,6 +808,7 @@ def raw_page(self): class GetDomainMappingRequest(proto.Message): r"""Request message for ``DomainMappings.GetDomainMapping``. + Attributes: name (str): Name of the resource requested. Example: @@ -787,6 +820,7 @@ class GetDomainMappingRequest(proto.Message): class CreateDomainMappingRequest(proto.Message): r"""Request message for ``DomainMappings.CreateDomainMapping``. + Attributes: parent (str): Name of the parent Application resource. Example: @@ -810,6 +844,7 @@ class CreateDomainMappingRequest(proto.Message): class UpdateDomainMappingRequest(proto.Message): r"""Request message for ``DomainMappings.UpdateDomainMapping``. + Attributes: name (str): Name of the resource to update. Example: @@ -834,6 +869,7 @@ class UpdateDomainMappingRequest(proto.Message): class DeleteDomainMappingRequest(proto.Message): r"""Request message for ``DomainMappings.DeleteDomainMapping``. + Attributes: name (str): Name of the resource to delete. Example: diff --git a/google/cloud/appengine_admin_v1/types/application.py b/google/cloud/appengine_admin_v1/types/application.py index 89ee559..47022cf 100644 --- a/google/cloud/appengine_admin_v1/types/application.py +++ b/google/cloud/appengine_admin_v1/types/application.py @@ -110,6 +110,7 @@ class DatabaseType(proto.Enum): class IdentityAwareProxy(proto.Message): r"""Identity-Aware Proxy + Attributes: enabled (bool): Whether the serving infrastructure will authenticate and diff --git a/google/cloud/appengine_admin_v1/types/audit_data.py b/google/cloud/appengine_admin_v1/types/audit_data.py index c99e43a..eb59c60 100644 --- a/google/cloud/appengine_admin_v1/types/audit_data.py +++ b/google/cloud/appengine_admin_v1/types/audit_data.py @@ -26,6 +26,7 @@ class AuditData(proto.Message): r"""App Engine admin service audit log. + Attributes: update_service (google.cloud.appengine_admin_v1.types.UpdateServiceMethod): Detailed information about UpdateService @@ -45,6 +46,7 @@ class AuditData(proto.Message): class UpdateServiceMethod(proto.Message): r"""Detailed information about UpdateService call. + Attributes: request (google.cloud.appengine_admin_v1.types.UpdateServiceRequest): Update service request. @@ -57,6 +59,7 @@ class UpdateServiceMethod(proto.Message): class CreateVersionMethod(proto.Message): r"""Detailed information about CreateVersion call. + Attributes: request (google.cloud.appengine_admin_v1.types.CreateVersionRequest): Create version request. diff --git a/google/cloud/appengine_admin_v1/types/certificate.py b/google/cloud/appengine_admin_v1/types/certificate.py index 4d28039..cd20a74 100644 --- a/google/cloud/appengine_admin_v1/types/certificate.py +++ b/google/cloud/appengine_admin_v1/types/certificate.py @@ -131,6 +131,7 @@ class AuthorizedCertificate(proto.Message): class CertificateRawData(proto.Message): r"""An SSL certificate obtained from a certificate authority. + Attributes: public_certificate (str): PEM encoded x.509 public key certificate. This field is set @@ -167,6 +168,7 @@ class CertificateRawData(proto.Message): class ManagedCertificate(proto.Message): r"""A certificate managed by App Engine. + Attributes: last_renewal_time (google.protobuf.timestamp_pb2.Timestamp): Time at which the certificate was last renewed. The renewal diff --git a/google/cloud/appengine_admin_v1/types/deploy.py b/google/cloud/appengine_admin_v1/types/deploy.py index 6345d8f..0013251 100644 --- a/google/cloud/appengine_admin_v1/types/deploy.py +++ b/google/cloud/appengine_admin_v1/types/deploy.py @@ -132,6 +132,7 @@ class CloudBuildOptions(proto.Message): class ZipInfo(proto.Message): r"""The zip file information for a zip deployment. + Attributes: source_url (str): URL of the zip file to deploy from. Must be a diff --git a/google/cloud/appengine_admin_v1/types/domain_mapping.py b/google/cloud/appengine_admin_v1/types/domain_mapping.py index bfa48da..68f1188 100644 --- a/google/cloud/appengine_admin_v1/types/domain_mapping.py +++ b/google/cloud/appengine_admin_v1/types/domain_mapping.py @@ -24,6 +24,7 @@ class DomainMapping(proto.Message): r"""A domain serving an App Engine application. + Attributes: name (str): Full path to the ``DomainMapping`` resource in the API. @@ -55,6 +56,7 @@ class DomainMapping(proto.Message): class SslSettings(proto.Message): r"""SSL configuration for a ``DomainMapping`` resource. + Attributes: certificate_id (str): ID of the ``AuthorizedCertificate`` resource configuring SSL @@ -100,6 +102,7 @@ class SslManagementType(proto.Enum): class ResourceRecord(proto.Message): r"""A DNS resource record. + Attributes: name (str): Relative name of the object affected by this record. Only diff --git a/google/cloud/appengine_admin_v1/types/instance.py b/google/cloud/appengine_admin_v1/types/instance.py index 2311bb9..2c7c779 100644 --- a/google/cloud/appengine_admin_v1/types/instance.py +++ b/google/cloud/appengine_admin_v1/types/instance.py @@ -93,7 +93,8 @@ class Availability(proto.Enum): DYNAMIC = 2 class Liveness(proto.Message): - r"""Wrapper for LivenessState enum. """ + r"""Wrapper for LivenessState enum. + """ class LivenessState(proto.Enum): r"""Liveness health check status for Flex instances.""" diff --git a/google/cloud/appengine_admin_v1/types/version.py b/google/cloud/appengine_admin_v1/types/version.py index c475063..f1d353f 100644 --- a/google/cloud/appengine_admin_v1/types/version.py +++ b/google/cloud/appengine_admin_v1/types/version.py @@ -488,6 +488,7 @@ class ManualScaling(proto.Message): class CpuUtilization(proto.Message): r"""Target scaling by CPU usage. + Attributes: aggregation_window_length (google.protobuf.duration_pb2.Duration): Period of time over which CPU utilization is @@ -562,6 +563,7 @@ class NetworkUtilization(proto.Message): class StandardSchedulerSettings(proto.Message): r"""Scheduler settings for standard environment. + Attributes: target_cpu_utilization (float): Target CPU utilization ratio to maintain when @@ -657,6 +659,7 @@ class Volume(proto.Message): class Resources(proto.Message): r"""Machine resources for a version. + Attributes: cpu (float): Number of CPU cores needed. @@ -681,6 +684,7 @@ class Resources(proto.Message): class VpcAccessConnector(proto.Message): r"""VPC access connector specification. + Attributes: name (str): Full Serverless VPC Access Connector name @@ -693,6 +697,7 @@ class VpcAccessConnector(proto.Message): class Entrypoint(proto.Message): r"""The entrypoint for the application. + Attributes: shell (str): The format should be a shell command that can be fed to diff --git a/tests/unit/gapic/appengine_admin_v1/test_applications.py b/tests/unit/gapic/appengine_admin_v1/test_applications.py index fbea8f0..7c034b5 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_applications.py +++ b/tests/unit/gapic/appengine_admin_v1/test_applications.py @@ -32,6 +32,7 @@ from google.api_core import grpc_helpers_async from google.api_core import operation_async # type: ignore from google.api_core import operations_v1 +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.applications import ( @@ -1137,6 +1138,9 @@ def test_applications_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + # Additionally, the LRO client (a property) should # also raise NotImplementedError with pytest.raises(NotImplementedError): @@ -1643,3 +1647,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = ApplicationsAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = ApplicationsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = ApplicationsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_authorized_certificates.py b/tests/unit/gapic/appengine_admin_v1/test_authorized_certificates.py index a8d1ccd..6de2281 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_authorized_certificates.py +++ b/tests/unit/gapic/appengine_admin_v1/test_authorized_certificates.py @@ -29,6 +29,7 @@ from google.api_core import gapic_v1 from google.api_core import grpc_helpers from google.api_core import grpc_helpers_async +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.authorized_certificates import ( @@ -1618,6 +1619,9 @@ def test_authorized_certificates_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + @requires_google_auth_gte_1_25_0 def test_authorized_certificates_base_transport_with_credentials_file(): @@ -2108,3 +2112,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = AuthorizedCertificatesAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = AuthorizedCertificatesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = AuthorizedCertificatesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_authorized_domains.py b/tests/unit/gapic/appengine_admin_v1/test_authorized_domains.py index 773df28..4d570ea 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_authorized_domains.py +++ b/tests/unit/gapic/appengine_admin_v1/test_authorized_domains.py @@ -29,6 +29,7 @@ from google.api_core import gapic_v1 from google.api_core import grpc_helpers from google.api_core import grpc_helpers_async +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.authorized_domains import ( @@ -909,6 +910,9 @@ def test_authorized_domains_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + @requires_google_auth_gte_1_25_0 def test_authorized_domains_base_transport_with_credentials_file(): @@ -1395,3 +1399,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = AuthorizedDomainsAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = AuthorizedDomainsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = AuthorizedDomainsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_domain_mappings.py b/tests/unit/gapic/appengine_admin_v1/test_domain_mappings.py index 39edd26..9a13770 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_domain_mappings.py +++ b/tests/unit/gapic/appengine_admin_v1/test_domain_mappings.py @@ -32,6 +32,7 @@ from google.api_core import grpc_helpers_async from google.api_core import operation_async # type: ignore from google.api_core import operations_v1 +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.domain_mappings import ( @@ -1504,6 +1505,9 @@ def test_domain_mappings_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + # Additionally, the LRO client (a property) should # also raise NotImplementedError with pytest.raises(NotImplementedError): @@ -2021,3 +2025,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = DomainMappingsAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = DomainMappingsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = DomainMappingsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_firewall.py b/tests/unit/gapic/appengine_admin_v1/test_firewall.py index ccbc861..b60e491 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_firewall.py +++ b/tests/unit/gapic/appengine_admin_v1/test_firewall.py @@ -29,6 +29,7 @@ from google.api_core import gapic_v1 from google.api_core import grpc_helpers from google.api_core import grpc_helpers_async +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.firewall import FirewallAsyncClient @@ -1611,6 +1612,9 @@ def test_firewall_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + @requires_google_auth_gte_1_25_0 def test_firewall_base_transport_with_credentials_file(): @@ -2080,3 +2084,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = FirewallAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = FirewallClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = FirewallClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_instances.py b/tests/unit/gapic/appengine_admin_v1/test_instances.py index d82988f..d51b0a8 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_instances.py +++ b/tests/unit/gapic/appengine_admin_v1/test_instances.py @@ -32,6 +32,7 @@ from google.api_core import grpc_helpers_async from google.api_core import operation_async # type: ignore from google.api_core import operations_v1 +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.instances import InstancesAsyncClient @@ -1313,6 +1314,9 @@ def test_instances_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + # Additionally, the LRO client (a property) should # also raise NotImplementedError with pytest.raises(NotImplementedError): @@ -1839,3 +1843,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = InstancesAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = InstancesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = InstancesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_services.py b/tests/unit/gapic/appengine_admin_v1/test_services.py index 4f78a14..168aded 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_services.py +++ b/tests/unit/gapic/appengine_admin_v1/test_services.py @@ -32,6 +32,7 @@ from google.api_core import grpc_helpers_async from google.api_core import operation_async # type: ignore from google.api_core import operations_v1 +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.services import ServicesAsyncClient @@ -1222,6 +1223,9 @@ def test_services_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + # Additionally, the LRO client (a property) should # also raise NotImplementedError with pytest.raises(NotImplementedError): @@ -1722,3 +1726,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = ServicesAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = ServicesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = ServicesClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called() diff --git a/tests/unit/gapic/appengine_admin_v1/test_versions.py b/tests/unit/gapic/appengine_admin_v1/test_versions.py index f47e621..84a0499 100644 --- a/tests/unit/gapic/appengine_admin_v1/test_versions.py +++ b/tests/unit/gapic/appengine_admin_v1/test_versions.py @@ -32,6 +32,7 @@ from google.api_core import grpc_helpers_async from google.api_core import operation_async # type: ignore from google.api_core import operations_v1 +from google.api_core import path_template from google.auth import credentials as ga_credentials from google.auth.exceptions import MutualTLSChannelError from google.cloud.appengine_admin_v1.services.versions import VersionsAsyncClient @@ -1433,6 +1434,9 @@ def test_versions_base_transport(): with pytest.raises(NotImplementedError): getattr(transport, method)(request=object()) + with pytest.raises(NotImplementedError): + transport.close() + # Additionally, the LRO client (a property) should # also raise NotImplementedError with pytest.raises(NotImplementedError): @@ -1933,3 +1937,49 @@ def test_client_withDEFAULT_CLIENT_INFO(): credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) + + +@pytest.mark.asyncio +async def test_transport_close_async(): + client = VersionsAsyncClient( + credentials=ga_credentials.AnonymousCredentials(), transport="grpc_asyncio", + ) + with mock.patch.object( + type(getattr(client.transport, "grpc_channel")), "close" + ) as close: + async with client: + close.assert_not_called() + close.assert_called_once() + + +def test_transport_close(): + transports = { + "grpc": "_grpc_channel", + } + + for transport, close_name in transports.items(): + client = VersionsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + with mock.patch.object( + type(getattr(client.transport, close_name)), "close" + ) as close: + with client: + close.assert_not_called() + close.assert_called_once() + + +def test_client_ctx(): + transports = [ + "grpc", + ] + for transport in transports: + client = VersionsClient( + credentials=ga_credentials.AnonymousCredentials(), transport=transport + ) + # Test client calls underlying transport. + with mock.patch.object(type(client.transport), "close") as close: + close.assert_not_called() + with client: + pass + close.assert_called()