From 3fd20b0bab01c82120d569383bba73607c59d17b Mon Sep 17 00:00:00 2001 From: yoshi-automation Date: Sat, 12 Dec 2020 07:28:23 -0800 Subject: [PATCH] fix: remove client recv msg limit fix: add enums to `types/__init__.py` PiperOrigin-RevId: 347055288 Source-Author: Google APIs Source-Date: Fri Dec 11 12:44:37 2020 -0800 Source-Repo: googleapis/googleapis Source-Sha: dd372aa22ded7a8ba6f0e03a80e06358a3fa0907 Source-Link: https://github.com/googleapis/googleapis/commit/dd372aa22ded7a8ba6f0e03a80e06358a3fa0907 --- .../services/cloud_scheduler/transports/__init__.py | 1 - .../services/cloud_scheduler/transports/grpc.py | 10 +++++++++- .../cloud_scheduler/transports/grpc_asyncio.py | 8 ++++++++ google/cloud/scheduler_v1/types/__init__.py | 3 ++- .../services/cloud_scheduler/transports/__init__.py | 1 - .../services/cloud_scheduler/transports/grpc.py | 10 +++++++++- .../cloud_scheduler/transports/grpc_asyncio.py | 8 ++++++++ google/cloud/scheduler_v1beta1/types/__init__.py | 3 ++- synth.metadata | 6 +++--- tests/unit/gapic/scheduler_v1/test_cloud_scheduler.py | 8 ++++++++ .../gapic/scheduler_v1beta1/test_cloud_scheduler.py | 8 ++++++++ 11 files changed, 57 insertions(+), 9 deletions(-) diff --git a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/__init__.py b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/__init__.py index e9b56e23..99c6c24c 100644 --- a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/__init__.py +++ b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/__init__.py @@ -28,7 +28,6 @@ _transport_registry["grpc"] = CloudSchedulerGrpcTransport _transport_registry["grpc_asyncio"] = CloudSchedulerGrpcAsyncIOTransport - __all__ = ( "CloudSchedulerTransport", "CloudSchedulerGrpcTransport", diff --git a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc.py b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc.py index 06c15306..576f3665 100644 --- a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc.py +++ b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc.py @@ -150,6 +150,10 @@ def __init__( ssl_credentials=ssl_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._ssl_channel_credentials = ssl_credentials else: @@ -168,6 +172,10 @@ def __init__( ssl_credentials=ssl_channel_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._stubs = {} # type: Dict[str, Callable] @@ -194,7 +202,7 @@ def create_channel( ) -> grpc.Channel: """Create and return a gRPC channel object. Args: - address (Optionsl[str]): The host for the channel to use. + address (Optional[str]): The host for the channel to use. credentials (Optional[~.Credentials]): The authorization credentials to attach to requests. These credentials identify this application to the service. If diff --git a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc_asyncio.py b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc_asyncio.py index 99a0dc0f..66dbf1a8 100644 --- a/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc_asyncio.py +++ b/google/cloud/scheduler_v1/services/cloud_scheduler/transports/grpc_asyncio.py @@ -195,6 +195,10 @@ def __init__( ssl_credentials=ssl_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._ssl_channel_credentials = ssl_credentials else: @@ -213,6 +217,10 @@ def __init__( ssl_credentials=ssl_channel_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) # Run the base constructor. diff --git a/google/cloud/scheduler_v1/types/__init__.py b/google/cloud/scheduler_v1/types/__init__.py index 53e55ca7..33453074 100644 --- a/google/cloud/scheduler_v1/types/__init__.py +++ b/google/cloud/scheduler_v1/types/__init__.py @@ -22,6 +22,7 @@ AppEngineRouting, OAuthToken, OidcToken, + HttpMethod, ) from .job import ( Job, @@ -39,7 +40,6 @@ RunJobRequest, ) - __all__ = ( "HttpTarget", "AppEngineHttpTarget", @@ -47,6 +47,7 @@ "AppEngineRouting", "OAuthToken", "OidcToken", + "HttpMethod", "Job", "RetryConfig", "ListJobsRequest", diff --git a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/__init__.py b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/__init__.py index e9b56e23..99c6c24c 100644 --- a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/__init__.py +++ b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/__init__.py @@ -28,7 +28,6 @@ _transport_registry["grpc"] = CloudSchedulerGrpcTransport _transport_registry["grpc_asyncio"] = CloudSchedulerGrpcAsyncIOTransport - __all__ = ( "CloudSchedulerTransport", "CloudSchedulerGrpcTransport", diff --git a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc.py b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc.py index b9a6ce1c..99405ec8 100644 --- a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc.py +++ b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc.py @@ -150,6 +150,10 @@ def __init__( ssl_credentials=ssl_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._ssl_channel_credentials = ssl_credentials else: @@ -168,6 +172,10 @@ def __init__( ssl_credentials=ssl_channel_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._stubs = {} # type: Dict[str, Callable] @@ -194,7 +202,7 @@ def create_channel( ) -> grpc.Channel: """Create and return a gRPC channel object. Args: - address (Optionsl[str]): The host for the channel to use. + address (Optional[str]): The host for the channel to use. credentials (Optional[~.Credentials]): The authorization credentials to attach to requests. These credentials identify this application to the service. If diff --git a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc_asyncio.py b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc_asyncio.py index 6d052671..ee367e61 100644 --- a/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc_asyncio.py +++ b/google/cloud/scheduler_v1beta1/services/cloud_scheduler/transports/grpc_asyncio.py @@ -195,6 +195,10 @@ def __init__( ssl_credentials=ssl_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) self._ssl_channel_credentials = ssl_credentials else: @@ -213,6 +217,10 @@ def __init__( ssl_credentials=ssl_channel_credentials, scopes=scopes or self.AUTH_SCOPES, quota_project_id=quota_project_id, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) # Run the base constructor. diff --git a/google/cloud/scheduler_v1beta1/types/__init__.py b/google/cloud/scheduler_v1beta1/types/__init__.py index 53e55ca7..33453074 100644 --- a/google/cloud/scheduler_v1beta1/types/__init__.py +++ b/google/cloud/scheduler_v1beta1/types/__init__.py @@ -22,6 +22,7 @@ AppEngineRouting, OAuthToken, OidcToken, + HttpMethod, ) from .job import ( Job, @@ -39,7 +40,6 @@ RunJobRequest, ) - __all__ = ( "HttpTarget", "AppEngineHttpTarget", @@ -47,6 +47,7 @@ "AppEngineRouting", "OAuthToken", "OidcToken", + "HttpMethod", "Job", "RetryConfig", "ListJobsRequest", diff --git a/synth.metadata b/synth.metadata index 4f4876b2..451ffc93 100644 --- a/synth.metadata +++ b/synth.metadata @@ -4,15 +4,15 @@ "git": { "name": ".", "remote": "https://github.com/googleapis/python-scheduler.git", - "sha": "f9fc0f9613302de642680c87286de0a02f09d086" + "sha": "1c8765708716f2876ad86b49c950a2eec6efb63f" } }, { "git": { "name": "googleapis", "remote": "https://github.com/googleapis/googleapis.git", - "sha": "dabe30b45bc86fdaf4126f26cbcfd73babbfa3fe", - "internalRef": "345194337" + "sha": "dd372aa22ded7a8ba6f0e03a80e06358a3fa0907", + "internalRef": "347055288" } }, { diff --git a/tests/unit/gapic/scheduler_v1/test_cloud_scheduler.py b/tests/unit/gapic/scheduler_v1/test_cloud_scheduler.py index cfabf0bd..b16b09c4 100644 --- a/tests/unit/gapic/scheduler_v1/test_cloud_scheduler.py +++ b/tests/unit/gapic/scheduler_v1/test_cloud_scheduler.py @@ -2491,6 +2491,10 @@ def test_cloud_scheduler_transport_channel_mtls_with_client_cert_source( scopes=("https://www.googleapis.com/auth/cloud-platform",), ssl_credentials=mock_ssl_cred, quota_project_id=None, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) assert transport.grpc_channel == mock_grpc_channel assert transport._ssl_channel_credentials == mock_ssl_cred @@ -2532,6 +2536,10 @@ def test_cloud_scheduler_transport_channel_mtls_with_adc(transport_class): scopes=("https://www.googleapis.com/auth/cloud-platform",), ssl_credentials=mock_ssl_cred, quota_project_id=None, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) assert transport.grpc_channel == mock_grpc_channel diff --git a/tests/unit/gapic/scheduler_v1beta1/test_cloud_scheduler.py b/tests/unit/gapic/scheduler_v1beta1/test_cloud_scheduler.py index edd37602..8bafaf86 100644 --- a/tests/unit/gapic/scheduler_v1beta1/test_cloud_scheduler.py +++ b/tests/unit/gapic/scheduler_v1beta1/test_cloud_scheduler.py @@ -2493,6 +2493,10 @@ def test_cloud_scheduler_transport_channel_mtls_with_client_cert_source( scopes=("https://www.googleapis.com/auth/cloud-platform",), ssl_credentials=mock_ssl_cred, quota_project_id=None, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) assert transport.grpc_channel == mock_grpc_channel assert transport._ssl_channel_credentials == mock_ssl_cred @@ -2534,6 +2538,10 @@ def test_cloud_scheduler_transport_channel_mtls_with_adc(transport_class): scopes=("https://www.googleapis.com/auth/cloud-platform",), ssl_credentials=mock_ssl_cred, quota_project_id=None, + options=[ + ("grpc.max_send_message_length", -1), + ("grpc.max_receive_message_length", -1), + ], ) assert transport.grpc_channel == mock_grpc_channel