google-cloud-container 2.47.0__tar.gz → 2.47.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (55) hide show
  1. {google-cloud-container-2.47.0/google_cloud_container.egg-info → google-cloud-container-2.47.1}/PKG-INFO +3 -3
  2. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/README.rst +1 -1
  3. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container/gapic_version.py +1 -1
  4. {google-cloud-container-2.47.0/google/cloud/container_v1beta1 → google-cloud-container-2.47.1/google/cloud/container_v1}/gapic_version.py +1 -1
  5. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/transports/base.py +3 -1
  6. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/transports/grpc.py +2 -1
  7. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/transports/grpc_asyncio.py +2 -1
  8. {google-cloud-container-2.47.0/google/cloud/container_v1 → google-cloud-container-2.47.1/google/cloud/container_v1beta1}/gapic_version.py +1 -1
  9. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/transports/base.py +3 -1
  10. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/transports/grpc.py +2 -1
  11. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/transports/grpc_asyncio.py +2 -1
  12. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/types/cluster_service.py +2 -2
  13. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1/google_cloud_container.egg-info}/PKG-INFO +3 -3
  14. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google_cloud_container.egg-info/requires.txt +1 -1
  15. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/setup.py +1 -1
  16. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/gapic/container_v1/test_cluster_manager.py +34 -204
  17. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/gapic/container_v1beta1/test_cluster_manager.py +35 -210
  18. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/LICENSE +0 -0
  19. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/MANIFEST.in +0 -0
  20. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container/__init__.py +0 -0
  21. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container/py.typed +0 -0
  22. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/__init__.py +0 -0
  23. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/gapic_metadata.json +0 -0
  24. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/py.typed +0 -0
  25. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/__init__.py +0 -0
  26. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/__init__.py +0 -0
  27. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/async_client.py +0 -0
  28. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/client.py +0 -0
  29. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/pagers.py +0 -0
  30. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/transports/__init__.py +0 -0
  31. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/services/cluster_manager/transports/rest.py +0 -0
  32. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/types/__init__.py +0 -0
  33. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1/types/cluster_service.py +0 -0
  34. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/__init__.py +0 -0
  35. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/gapic_metadata.json +0 -0
  36. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/py.typed +0 -0
  37. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/__init__.py +0 -0
  38. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/__init__.py +0 -0
  39. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/async_client.py +0 -0
  40. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/client.py +0 -0
  41. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/pagers.py +0 -0
  42. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/services/cluster_manager/transports/__init__.py +0 -0
  43. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google/cloud/container_v1beta1/types/__init__.py +0 -0
  44. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google_cloud_container.egg-info/SOURCES.txt +0 -0
  45. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google_cloud_container.egg-info/dependency_links.txt +0 -0
  46. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google_cloud_container.egg-info/not-zip-safe +0 -0
  47. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/google_cloud_container.egg-info/top_level.txt +0 -0
  48. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/setup.cfg +0 -0
  49. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/__init__.py +0 -0
  50. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/system/__init__.py +0 -0
  51. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/system/smoke_test.py +0 -0
  52. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/__init__.py +0 -0
  53. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/gapic/__init__.py +0 -0
  54. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/gapic/container_v1/__init__.py +0 -0
  55. {google-cloud-container-2.47.0 → google-cloud-container-2.47.1}/tests/unit/gapic/container_v1beta1/__init__.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: google-cloud-container
3
- Version: 2.47.0
3
+ Version: 2.47.1
4
4
  Summary: Google Cloud Container API client library
5
5
  Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-container
6
6
  Author: Google LLC
@@ -25,7 +25,7 @@ License-File: LICENSE
25
25
  Requires-Dist: google-api-core[grpc]!=2.0.*,!=2.1.*,!=2.10.*,!=2.2.*,!=2.3.*,!=2.4.*,!=2.5.*,!=2.6.*,!=2.7.*,!=2.8.*,!=2.9.*,<3.0.0dev,>=1.34.1
26
26
  Requires-Dist: google-auth!=2.24.0,!=2.25.0,<3.0.0dev,>=2.14.1
27
27
  Requires-Dist: proto-plus<2.0.0dev,>=1.22.3
28
- Requires-Dist: protobuf!=3.20.0,!=3.20.1,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.19.5
28
+ Requires-Dist: protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2
29
29
 
30
30
  Python Client for Kubernetes Engine
31
31
  ===================================
@@ -44,7 +44,7 @@ Python Client for Kubernetes Engine
44
44
  .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-container.svg
45
45
  :target: https://pypi.org/project/google-cloud-container/
46
46
  .. _Kubernetes Engine: https://cloud.google.com/kubernetes-engine/
47
- .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest
47
+ .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest/summary_overview
48
48
  .. _Product Documentation: https://cloud.google.com/kubernetes-engine/
49
49
 
50
50
  Quick Start
@@ -15,7 +15,7 @@ Python Client for Kubernetes Engine
15
15
  .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-container.svg
16
16
  :target: https://pypi.org/project/google-cloud-container/
17
17
  .. _Kubernetes Engine: https://cloud.google.com/kubernetes-engine/
18
- .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest
18
+ .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest/summary_overview
19
19
  .. _Product Documentation: https://cloud.google.com/kubernetes-engine/
20
20
 
21
21
  Quick Start
@@ -13,4 +13,4 @@
13
13
  # See the License for the specific language governing permissions and
14
14
  # limitations under the License.
15
15
  #
16
- __version__ = "2.47.0" # {x-release-please-version}
16
+ __version__ = "2.47.1" # {x-release-please-version}
@@ -13,4 +13,4 @@
13
13
  # See the License for the specific language governing permissions and
14
14
  # limitations under the License.
15
15
  #
16
- __version__ = "2.47.0" # {x-release-please-version}
16
+ __version__ = "2.47.1" # {x-release-please-version}
@@ -82,6 +82,8 @@ class ClusterManagerTransport(abc.ABC):
82
82
 
83
83
  # Save the scopes.
84
84
  self._scopes = scopes
85
+ if not hasattr(self, "_ignore_credentials"):
86
+ self._ignore_credentials: bool = False
85
87
 
86
88
  # If no credentials are provided, then determine the appropriate
87
89
  # defaults.
@@ -94,7 +96,7 @@ class ClusterManagerTransport(abc.ABC):
94
96
  credentials, _ = google.auth.load_credentials_from_file(
95
97
  credentials_file, **scopes_kwargs, quota_project_id=quota_project_id
96
98
  )
97
- elif credentials is None:
99
+ elif credentials is None and not self._ignore_credentials:
98
100
  credentials, _ = google.auth.default(
99
101
  **scopes_kwargs, quota_project_id=quota_project_id
100
102
  )
@@ -122,7 +122,8 @@ class ClusterManagerGrpcTransport(ClusterManagerTransport):
122
122
 
123
123
  if isinstance(channel, grpc.Channel):
124
124
  # Ignore credentials if a channel was passed.
125
- credentials = False
125
+ credentials = None
126
+ self._ignore_credentials = True
126
127
  # If a channel was explicitly provided, set it.
127
128
  self._grpc_channel = channel
128
129
  self._ssl_channel_credentials = None
@@ -169,7 +169,8 @@ class ClusterManagerGrpcAsyncIOTransport(ClusterManagerTransport):
169
169
 
170
170
  if isinstance(channel, aio.Channel):
171
171
  # Ignore credentials if a channel was passed.
172
- credentials = False
172
+ credentials = None
173
+ self._ignore_credentials = True
173
174
  # If a channel was explicitly provided, set it.
174
175
  self._grpc_channel = channel
175
176
  self._ssl_channel_credentials = None
@@ -13,4 +13,4 @@
13
13
  # See the License for the specific language governing permissions and
14
14
  # limitations under the License.
15
15
  #
16
- __version__ = "2.47.0" # {x-release-please-version}
16
+ __version__ = "2.47.1" # {x-release-please-version}
@@ -82,6 +82,8 @@ class ClusterManagerTransport(abc.ABC):
82
82
 
83
83
  # Save the scopes.
84
84
  self._scopes = scopes
85
+ if not hasattr(self, "_ignore_credentials"):
86
+ self._ignore_credentials: bool = False
85
87
 
86
88
  # If no credentials are provided, then determine the appropriate
87
89
  # defaults.
@@ -94,7 +96,7 @@ class ClusterManagerTransport(abc.ABC):
94
96
  credentials, _ = google.auth.load_credentials_from_file(
95
97
  credentials_file, **scopes_kwargs, quota_project_id=quota_project_id
96
98
  )
97
- elif credentials is None:
99
+ elif credentials is None and not self._ignore_credentials:
98
100
  credentials, _ = google.auth.default(
99
101
  **scopes_kwargs, quota_project_id=quota_project_id
100
102
  )
@@ -122,7 +122,8 @@ class ClusterManagerGrpcTransport(ClusterManagerTransport):
122
122
 
123
123
  if isinstance(channel, grpc.Channel):
124
124
  # Ignore credentials if a channel was passed.
125
- credentials = False
125
+ credentials = None
126
+ self._ignore_credentials = True
126
127
  # If a channel was explicitly provided, set it.
127
128
  self._grpc_channel = channel
128
129
  self._ssl_channel_credentials = None
@@ -169,7 +169,8 @@ class ClusterManagerGrpcAsyncIOTransport(ClusterManagerTransport):
169
169
 
170
170
  if isinstance(channel, aio.Channel):
171
171
  # Ignore credentials if a channel was passed.
172
- credentials = False
172
+ credentials = None
173
+ self._ignore_credentials = True
173
174
  # If a channel was explicitly provided, set it.
174
175
  self._grpc_channel = channel
175
176
  self._ssl_channel_credentials = None
@@ -9075,8 +9075,8 @@ class GatewayAPIConfig(proto.Message):
9075
9075
  CHANNEL_DISABLED (1):
9076
9076
  Gateway API support is disabled
9077
9077
  CHANNEL_EXPERIMENTAL (3):
9078
- Gateway API support is enabled, experimental
9079
- CRDs are installed
9078
+ Deprecated: use CHANNEL_STANDARD instead. Gateway API
9079
+ support is enabled, experimental CRDs are installed
9080
9080
  CHANNEL_STANDARD (4):
9081
9081
  Gateway API support is enabled, standard CRDs
9082
9082
  are installed
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: google-cloud-container
3
- Version: 2.47.0
3
+ Version: 2.47.1
4
4
  Summary: Google Cloud Container API client library
5
5
  Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-container
6
6
  Author: Google LLC
@@ -25,7 +25,7 @@ License-File: LICENSE
25
25
  Requires-Dist: google-api-core[grpc]!=2.0.*,!=2.1.*,!=2.10.*,!=2.2.*,!=2.3.*,!=2.4.*,!=2.5.*,!=2.6.*,!=2.7.*,!=2.8.*,!=2.9.*,<3.0.0dev,>=1.34.1
26
26
  Requires-Dist: google-auth!=2.24.0,!=2.25.0,<3.0.0dev,>=2.14.1
27
27
  Requires-Dist: proto-plus<2.0.0dev,>=1.22.3
28
- Requires-Dist: protobuf!=3.20.0,!=3.20.1,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.19.5
28
+ Requires-Dist: protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2
29
29
 
30
30
  Python Client for Kubernetes Engine
31
31
  ===================================
@@ -44,7 +44,7 @@ Python Client for Kubernetes Engine
44
44
  .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-container.svg
45
45
  :target: https://pypi.org/project/google-cloud-container/
46
46
  .. _Kubernetes Engine: https://cloud.google.com/kubernetes-engine/
47
- .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest
47
+ .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/container/latest/summary_overview
48
48
  .. _Product Documentation: https://cloud.google.com/kubernetes-engine/
49
49
 
50
50
  Quick Start
@@ -1,4 +1,4 @@
1
1
  google-api-core[grpc]!=2.0.*,!=2.1.*,!=2.10.*,!=2.2.*,!=2.3.*,!=2.4.*,!=2.5.*,!=2.6.*,!=2.7.*,!=2.8.*,!=2.9.*,<3.0.0dev,>=1.34.1
2
2
  google-auth!=2.24.0,!=2.25.0,<3.0.0dev,>=2.14.1
3
3
  proto-plus<2.0.0dev,>=1.22.3
4
- protobuf!=3.20.0,!=3.20.1,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.19.5
4
+ protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2
@@ -44,7 +44,7 @@ dependencies = [
44
44
  # See https://github.com/googleapis/google-cloud-python/issues/12364
45
45
  "google-auth >= 2.14.1, <3.0.0dev,!=2.24.0,!=2.25.0",
46
46
  "proto-plus >= 1.22.3, <2.0.0dev",
47
- "protobuf>=3.19.5,<5.0.0dev,!=3.20.0,!=3.20.1,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5",
47
+ "protobuf>=3.20.2,<6.0.0dev,!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5",
48
48
  ]
49
49
  url = "https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-container"
50
50
 
@@ -1289,12 +1289,7 @@ async def test_list_clusters_async_use_cached_wrapped_rpc(
1289
1289
  )
1290
1290
 
1291
1291
  # Replace cached wrapped function with mock
1292
- class AwaitableMock(mock.AsyncMock):
1293
- def __await__(self):
1294
- self.await_count += 1
1295
- return iter([])
1296
-
1297
- mock_object = AwaitableMock()
1292
+ mock_object = mock.AsyncMock()
1298
1293
  client._client._transport._wrapped_methods[
1299
1294
  client._client._transport.list_clusters
1300
1295
  ] = mock_object
@@ -1777,12 +1772,7 @@ async def test_get_cluster_async_use_cached_wrapped_rpc(
1777
1772
  )
1778
1773
 
1779
1774
  # Replace cached wrapped function with mock
1780
- class AwaitableMock(mock.AsyncMock):
1781
- def __await__(self):
1782
- self.await_count += 1
1783
- return iter([])
1784
-
1785
- mock_object = AwaitableMock()
1775
+ mock_object = mock.AsyncMock()
1786
1776
  client._client._transport._wrapped_methods[
1787
1777
  client._client._transport.get_cluster
1788
1778
  ] = mock_object
@@ -2272,12 +2262,7 @@ async def test_create_cluster_async_use_cached_wrapped_rpc(
2272
2262
  )
2273
2263
 
2274
2264
  # Replace cached wrapped function with mock
2275
- class AwaitableMock(mock.AsyncMock):
2276
- def __await__(self):
2277
- self.await_count += 1
2278
- return iter([])
2279
-
2280
- mock_object = AwaitableMock()
2265
+ mock_object = mock.AsyncMock()
2281
2266
  client._client._transport._wrapped_methods[
2282
2267
  client._client._transport.create_cluster
2283
2268
  ] = mock_object
@@ -2727,12 +2712,7 @@ async def test_update_cluster_async_use_cached_wrapped_rpc(
2727
2712
  )
2728
2713
 
2729
2714
  # Replace cached wrapped function with mock
2730
- class AwaitableMock(mock.AsyncMock):
2731
- def __await__(self):
2732
- self.await_count += 1
2733
- return iter([])
2734
-
2735
- mock_object = AwaitableMock()
2715
+ mock_object = mock.AsyncMock()
2736
2716
  client._client._transport._wrapped_methods[
2737
2717
  client._client._transport.update_cluster
2738
2718
  ] = mock_object
@@ -3218,12 +3198,7 @@ async def test_update_node_pool_async_use_cached_wrapped_rpc(
3218
3198
  )
3219
3199
 
3220
3200
  # Replace cached wrapped function with mock
3221
- class AwaitableMock(mock.AsyncMock):
3222
- def __await__(self):
3223
- self.await_count += 1
3224
- return iter([])
3225
-
3226
- mock_object = AwaitableMock()
3201
+ mock_object = mock.AsyncMock()
3227
3202
  client._client._transport._wrapped_methods[
3228
3203
  client._client._transport.update_node_pool
3229
3204
  ] = mock_object
@@ -3576,12 +3551,7 @@ async def test_set_node_pool_autoscaling_async_use_cached_wrapped_rpc(
3576
3551
  )
3577
3552
 
3578
3553
  # Replace cached wrapped function with mock
3579
- class AwaitableMock(mock.AsyncMock):
3580
- def __await__(self):
3581
- self.await_count += 1
3582
- return iter([])
3583
-
3584
- mock_object = AwaitableMock()
3554
+ mock_object = mock.AsyncMock()
3585
3555
  client._client._transport._wrapped_methods[
3586
3556
  client._client._transport.set_node_pool_autoscaling
3587
3557
  ] = mock_object
@@ -3940,12 +3910,7 @@ async def test_set_logging_service_async_use_cached_wrapped_rpc(
3940
3910
  )
3941
3911
 
3942
3912
  # Replace cached wrapped function with mock
3943
- class AwaitableMock(mock.AsyncMock):
3944
- def __await__(self):
3945
- self.await_count += 1
3946
- return iter([])
3947
-
3948
- mock_object = AwaitableMock()
3913
+ mock_object = mock.AsyncMock()
3949
3914
  client._client._transport._wrapped_methods[
3950
3915
  client._client._transport.set_logging_service
3951
3916
  ] = mock_object
@@ -4431,12 +4396,7 @@ async def test_set_monitoring_service_async_use_cached_wrapped_rpc(
4431
4396
  )
4432
4397
 
4433
4398
  # Replace cached wrapped function with mock
4434
- class AwaitableMock(mock.AsyncMock):
4435
- def __await__(self):
4436
- self.await_count += 1
4437
- return iter([])
4438
-
4439
- mock_object = AwaitableMock()
4399
+ mock_object = mock.AsyncMock()
4440
4400
  client._client._transport._wrapped_methods[
4441
4401
  client._client._transport.set_monitoring_service
4442
4402
  ] = mock_object
@@ -4917,12 +4877,7 @@ async def test_set_addons_config_async_use_cached_wrapped_rpc(
4917
4877
  )
4918
4878
 
4919
4879
  # Replace cached wrapped function with mock
4920
- class AwaitableMock(mock.AsyncMock):
4921
- def __await__(self):
4922
- self.await_count += 1
4923
- return iter([])
4924
-
4925
- mock_object = AwaitableMock()
4880
+ mock_object = mock.AsyncMock()
4926
4881
  client._client._transport._wrapped_methods[
4927
4882
  client._client._transport.set_addons_config
4928
4883
  ] = mock_object
@@ -5404,12 +5359,7 @@ async def test_set_locations_async_use_cached_wrapped_rpc(
5404
5359
  )
5405
5360
 
5406
5361
  # Replace cached wrapped function with mock
5407
- class AwaitableMock(mock.AsyncMock):
5408
- def __await__(self):
5409
- self.await_count += 1
5410
- return iter([])
5411
-
5412
- mock_object = AwaitableMock()
5362
+ mock_object = mock.AsyncMock()
5413
5363
  client._client._transport._wrapped_methods[
5414
5364
  client._client._transport.set_locations
5415
5365
  ] = mock_object
@@ -5871,12 +5821,7 @@ async def test_update_master_async_use_cached_wrapped_rpc(
5871
5821
  )
5872
5822
 
5873
5823
  # Replace cached wrapped function with mock
5874
- class AwaitableMock(mock.AsyncMock):
5875
- def __await__(self):
5876
- self.await_count += 1
5877
- return iter([])
5878
-
5879
- mock_object = AwaitableMock()
5824
+ mock_object = mock.AsyncMock()
5880
5825
  client._client._transport._wrapped_methods[
5881
5826
  client._client._transport.update_master
5882
5827
  ] = mock_object
@@ -6336,12 +6281,7 @@ async def test_set_master_auth_async_use_cached_wrapped_rpc(
6336
6281
  )
6337
6282
 
6338
6283
  # Replace cached wrapped function with mock
6339
- class AwaitableMock(mock.AsyncMock):
6340
- def __await__(self):
6341
- self.await_count += 1
6342
- return iter([])
6343
-
6344
- mock_object = AwaitableMock()
6284
+ mock_object = mock.AsyncMock()
6345
6285
  client._client._transport._wrapped_methods[
6346
6286
  client._client._transport.set_master_auth
6347
6287
  ] = mock_object
@@ -6679,12 +6619,7 @@ async def test_delete_cluster_async_use_cached_wrapped_rpc(
6679
6619
  )
6680
6620
 
6681
6621
  # Replace cached wrapped function with mock
6682
- class AwaitableMock(mock.AsyncMock):
6683
- def __await__(self):
6684
- self.await_count += 1
6685
- return iter([])
6686
-
6687
- mock_object = AwaitableMock()
6622
+ mock_object = mock.AsyncMock()
6688
6623
  client._client._transport._wrapped_methods[
6689
6624
  client._client._transport.delete_cluster
6690
6625
  ] = mock_object
@@ -7102,12 +7037,7 @@ async def test_list_operations_async_use_cached_wrapped_rpc(
7102
7037
  )
7103
7038
 
7104
7039
  # Replace cached wrapped function with mock
7105
- class AwaitableMock(mock.AsyncMock):
7106
- def __await__(self):
7107
- self.await_count += 1
7108
- return iter([])
7109
-
7110
- mock_object = AwaitableMock()
7040
+ mock_object = mock.AsyncMock()
7111
7041
  client._client._transport._wrapped_methods[
7112
7042
  client._client._transport.list_operations
7113
7043
  ] = mock_object
@@ -7527,12 +7457,7 @@ async def test_get_operation_async_use_cached_wrapped_rpc(
7527
7457
  )
7528
7458
 
7529
7459
  # Replace cached wrapped function with mock
7530
- class AwaitableMock(mock.AsyncMock):
7531
- def __await__(self):
7532
- self.await_count += 1
7533
- return iter([])
7534
-
7535
- mock_object = AwaitableMock()
7460
+ mock_object = mock.AsyncMock()
7536
7461
  client._client._transport._wrapped_methods[
7537
7462
  client._client._transport.get_operation
7538
7463
  ] = mock_object
@@ -7947,12 +7872,7 @@ async def test_cancel_operation_async_use_cached_wrapped_rpc(
7947
7872
  )
7948
7873
 
7949
7874
  # Replace cached wrapped function with mock
7950
- class AwaitableMock(mock.AsyncMock):
7951
- def __await__(self):
7952
- self.await_count += 1
7953
- return iter([])
7954
-
7955
- mock_object = AwaitableMock()
7875
+ mock_object = mock.AsyncMock()
7956
7876
  client._client._transport._wrapped_methods[
7957
7877
  client._client._transport.cancel_operation
7958
7878
  ] = mock_object
@@ -8363,12 +8283,7 @@ async def test_get_server_config_async_use_cached_wrapped_rpc(
8363
8283
  )
8364
8284
 
8365
8285
  # Replace cached wrapped function with mock
8366
- class AwaitableMock(mock.AsyncMock):
8367
- def __await__(self):
8368
- self.await_count += 1
8369
- return iter([])
8370
-
8371
- mock_object = AwaitableMock()
8286
+ mock_object = mock.AsyncMock()
8372
8287
  client._client._transport._wrapped_methods[
8373
8288
  client._client._transport.get_server_config
8374
8289
  ] = mock_object
@@ -8775,12 +8690,7 @@ async def test_get_json_web_keys_async_use_cached_wrapped_rpc(
8775
8690
  )
8776
8691
 
8777
8692
  # Replace cached wrapped function with mock
8778
- class AwaitableMock(mock.AsyncMock):
8779
- def __await__(self):
8780
- self.await_count += 1
8781
- return iter([])
8782
-
8783
- mock_object = AwaitableMock()
8693
+ mock_object = mock.AsyncMock()
8784
8694
  client._client._transport._wrapped_methods[
8785
8695
  client._client._transport.get_json_web_keys
8786
8696
  ] = mock_object
@@ -9066,12 +8976,7 @@ async def test_list_node_pools_async_use_cached_wrapped_rpc(
9066
8976
  )
9067
8977
 
9068
8978
  # Replace cached wrapped function with mock
9069
- class AwaitableMock(mock.AsyncMock):
9070
- def __await__(self):
9071
- self.await_count += 1
9072
- return iter([])
9073
-
9074
- mock_object = AwaitableMock()
8979
+ mock_object = mock.AsyncMock()
9075
8980
  client._client._transport._wrapped_methods[
9076
8981
  client._client._transport.list_node_pools
9077
8982
  ] = mock_object
@@ -9497,12 +9402,7 @@ async def test_get_node_pool_async_use_cached_wrapped_rpc(
9497
9402
  )
9498
9403
 
9499
9404
  # Replace cached wrapped function with mock
9500
- class AwaitableMock(mock.AsyncMock):
9501
- def __await__(self):
9502
- self.await_count += 1
9503
- return iter([])
9504
-
9505
- mock_object = AwaitableMock()
9405
+ mock_object = mock.AsyncMock()
9506
9406
  client._client._transport._wrapped_methods[
9507
9407
  client._client._transport.get_node_pool
9508
9408
  ] = mock_object
@@ -9962,12 +9862,7 @@ async def test_create_node_pool_async_use_cached_wrapped_rpc(
9962
9862
  )
9963
9863
 
9964
9864
  # Replace cached wrapped function with mock
9965
- class AwaitableMock(mock.AsyncMock):
9966
- def __await__(self):
9967
- self.await_count += 1
9968
- return iter([])
9969
-
9970
- mock_object = AwaitableMock()
9865
+ mock_object = mock.AsyncMock()
9971
9866
  client._client._transport._wrapped_methods[
9972
9867
  client._client._transport.create_node_pool
9973
9868
  ] = mock_object
@@ -10431,12 +10326,7 @@ async def test_delete_node_pool_async_use_cached_wrapped_rpc(
10431
10326
  )
10432
10327
 
10433
10328
  # Replace cached wrapped function with mock
10434
- class AwaitableMock(mock.AsyncMock):
10435
- def __await__(self):
10436
- self.await_count += 1
10437
- return iter([])
10438
-
10439
- mock_object = AwaitableMock()
10329
+ mock_object = mock.AsyncMock()
10440
10330
  client._client._transport._wrapped_methods[
10441
10331
  client._client._transport.delete_node_pool
10442
10332
  ] = mock_object
@@ -10866,12 +10756,7 @@ async def test_complete_node_pool_upgrade_async_use_cached_wrapped_rpc(
10866
10756
  )
10867
10757
 
10868
10758
  # Replace cached wrapped function with mock
10869
- class AwaitableMock(mock.AsyncMock):
10870
- def __await__(self):
10871
- self.await_count += 1
10872
- return iter([])
10873
-
10874
- mock_object = AwaitableMock()
10759
+ mock_object = mock.AsyncMock()
10875
10760
  client._client._transport._wrapped_methods[
10876
10761
  client._client._transport.complete_node_pool_upgrade
10877
10762
  ] = mock_object
@@ -11204,12 +11089,7 @@ async def test_rollback_node_pool_upgrade_async_use_cached_wrapped_rpc(
11204
11089
  )
11205
11090
 
11206
11091
  # Replace cached wrapped function with mock
11207
- class AwaitableMock(mock.AsyncMock):
11208
- def __await__(self):
11209
- self.await_count += 1
11210
- return iter([])
11211
-
11212
- mock_object = AwaitableMock()
11092
+ mock_object = mock.AsyncMock()
11213
11093
  client._client._transport._wrapped_methods[
11214
11094
  client._client._transport.rollback_node_pool_upgrade
11215
11095
  ] = mock_object
@@ -11695,12 +11575,7 @@ async def test_set_node_pool_management_async_use_cached_wrapped_rpc(
11695
11575
  )
11696
11576
 
11697
11577
  # Replace cached wrapped function with mock
11698
- class AwaitableMock(mock.AsyncMock):
11699
- def __await__(self):
11700
- self.await_count += 1
11701
- return iter([])
11702
-
11703
- mock_object = AwaitableMock()
11578
+ mock_object = mock.AsyncMock()
11704
11579
  client._client._transport._wrapped_methods[
11705
11580
  client._client._transport.set_node_pool_management
11706
11581
  ] = mock_object
@@ -12045,12 +11920,7 @@ async def test_set_labels_async_use_cached_wrapped_rpc(transport: str = "grpc_as
12045
11920
  )
12046
11921
 
12047
11922
  # Replace cached wrapped function with mock
12048
- class AwaitableMock(mock.AsyncMock):
12049
- def __await__(self):
12050
- self.await_count += 1
12051
- return iter([])
12052
-
12053
- mock_object = AwaitableMock()
11923
+ mock_object = mock.AsyncMock()
12054
11924
  client._client._transport._wrapped_methods[
12055
11925
  client._client._transport.set_labels
12056
11926
  ] = mock_object
@@ -12388,12 +12258,7 @@ async def test_set_legacy_abac_async_use_cached_wrapped_rpc(
12388
12258
  )
12389
12259
 
12390
12260
  # Replace cached wrapped function with mock
12391
- class AwaitableMock(mock.AsyncMock):
12392
- def __await__(self):
12393
- self.await_count += 1
12394
- return iter([])
12395
-
12396
- mock_object = AwaitableMock()
12261
+ mock_object = mock.AsyncMock()
12397
12262
  client._client._transport._wrapped_methods[
12398
12263
  client._client._transport.set_legacy_abac
12399
12264
  ] = mock_object
@@ -12863,12 +12728,7 @@ async def test_start_ip_rotation_async_use_cached_wrapped_rpc(
12863
12728
  )
12864
12729
 
12865
12730
  # Replace cached wrapped function with mock
12866
- class AwaitableMock(mock.AsyncMock):
12867
- def __await__(self):
12868
- self.await_count += 1
12869
- return iter([])
12870
-
12871
- mock_object = AwaitableMock()
12731
+ mock_object = mock.AsyncMock()
12872
12732
  client._client._transport._wrapped_methods[
12873
12733
  client._client._transport.start_ip_rotation
12874
12734
  ] = mock_object
@@ -13340,12 +13200,7 @@ async def test_complete_ip_rotation_async_use_cached_wrapped_rpc(
13340
13200
  )
13341
13201
 
13342
13202
  # Replace cached wrapped function with mock
13343
- class AwaitableMock(mock.AsyncMock):
13344
- def __await__(self):
13345
- self.await_count += 1
13346
- return iter([])
13347
-
13348
- mock_object = AwaitableMock()
13203
+ mock_object = mock.AsyncMock()
13349
13204
  client._client._transport._wrapped_methods[
13350
13205
  client._client._transport.complete_ip_rotation
13351
13206
  ] = mock_object
@@ -13820,12 +13675,7 @@ async def test_set_node_pool_size_async_use_cached_wrapped_rpc(
13820
13675
  )
13821
13676
 
13822
13677
  # Replace cached wrapped function with mock
13823
- class AwaitableMock(mock.AsyncMock):
13824
- def __await__(self):
13825
- self.await_count += 1
13826
- return iter([])
13827
-
13828
- mock_object = AwaitableMock()
13678
+ mock_object = mock.AsyncMock()
13829
13679
  client._client._transport._wrapped_methods[
13830
13680
  client._client._transport.set_node_pool_size
13831
13681
  ] = mock_object
@@ -14181,12 +14031,7 @@ async def test_set_network_policy_async_use_cached_wrapped_rpc(
14181
14031
  )
14182
14032
 
14183
14033
  # Replace cached wrapped function with mock
14184
- class AwaitableMock(mock.AsyncMock):
14185
- def __await__(self):
14186
- self.await_count += 1
14187
- return iter([])
14188
-
14189
- mock_object = AwaitableMock()
14034
+ mock_object = mock.AsyncMock()
14190
14035
  client._client._transport._wrapped_methods[
14191
14036
  client._client._transport.set_network_policy
14192
14037
  ] = mock_object
@@ -14682,12 +14527,7 @@ async def test_set_maintenance_policy_async_use_cached_wrapped_rpc(
14682
14527
  )
14683
14528
 
14684
14529
  # Replace cached wrapped function with mock
14685
- class AwaitableMock(mock.AsyncMock):
14686
- def __await__(self):
14687
- self.await_count += 1
14688
- return iter([])
14689
-
14690
- mock_object = AwaitableMock()
14530
+ mock_object = mock.AsyncMock()
14691
14531
  client._client._transport._wrapped_methods[
14692
14532
  client._client._transport.set_maintenance_policy
14693
14533
  ] = mock_object
@@ -15175,12 +15015,7 @@ async def test_list_usable_subnetworks_async_use_cached_wrapped_rpc(
15175
15015
  )
15176
15016
 
15177
15017
  # Replace cached wrapped function with mock
15178
- class AwaitableMock(mock.AsyncMock):
15179
- def __await__(self):
15180
- self.await_count += 1
15181
- return iter([])
15182
-
15183
- mock_object = AwaitableMock()
15018
+ mock_object = mock.AsyncMock()
15184
15019
  client._client._transport._wrapped_methods[
15185
15020
  client._client._transport.list_usable_subnetworks
15186
15021
  ] = mock_object
@@ -15680,12 +15515,7 @@ async def test_check_autopilot_compatibility_async_use_cached_wrapped_rpc(
15680
15515
  )
15681
15516
 
15682
15517
  # Replace cached wrapped function with mock
15683
- class AwaitableMock(mock.AsyncMock):
15684
- def __await__(self):
15685
- self.await_count += 1
15686
- return iter([])
15687
-
15688
- mock_object = AwaitableMock()
15518
+ mock_object = mock.AsyncMock()
15689
15519
  client._client._transport._wrapped_methods[
15690
15520
  client._client._transport.check_autopilot_compatibility
15691
15521
  ] = mock_object
@@ -1266,12 +1266,7 @@ async def test_list_clusters_async_use_cached_wrapped_rpc(
1266
1266
  )
1267
1267
 
1268
1268
  # Replace cached wrapped function with mock
1269
- class AwaitableMock(mock.AsyncMock):
1270
- def __await__(self):
1271
- self.await_count += 1
1272
- return iter([])
1273
-
1274
- mock_object = AwaitableMock()
1269
+ mock_object = mock.AsyncMock()
1275
1270
  client._client._transport._wrapped_methods[
1276
1271
  client._client._transport.list_clusters
1277
1272
  ] = mock_object
@@ -1750,12 +1745,7 @@ async def test_get_cluster_async_use_cached_wrapped_rpc(
1750
1745
  )
1751
1746
 
1752
1747
  # Replace cached wrapped function with mock
1753
- class AwaitableMock(mock.AsyncMock):
1754
- def __await__(self):
1755
- self.await_count += 1
1756
- return iter([])
1757
-
1758
- mock_object = AwaitableMock()
1748
+ mock_object = mock.AsyncMock()
1759
1749
  client._client._transport._wrapped_methods[
1760
1750
  client._client._transport.get_cluster
1761
1751
  ] = mock_object
@@ -2239,12 +2229,7 @@ async def test_create_cluster_async_use_cached_wrapped_rpc(
2239
2229
  )
2240
2230
 
2241
2231
  # Replace cached wrapped function with mock
2242
- class AwaitableMock(mock.AsyncMock):
2243
- def __await__(self):
2244
- self.await_count += 1
2245
- return iter([])
2246
-
2247
- mock_object = AwaitableMock()
2232
+ mock_object = mock.AsyncMock()
2248
2233
  client._client._transport._wrapped_methods[
2249
2234
  client._client._transport.create_cluster
2250
2235
  ] = mock_object
@@ -2684,12 +2669,7 @@ async def test_update_cluster_async_use_cached_wrapped_rpc(
2684
2669
  )
2685
2670
 
2686
2671
  # Replace cached wrapped function with mock
2687
- class AwaitableMock(mock.AsyncMock):
2688
- def __await__(self):
2689
- self.await_count += 1
2690
- return iter([])
2691
-
2692
- mock_object = AwaitableMock()
2672
+ mock_object = mock.AsyncMock()
2693
2673
  client._client._transport._wrapped_methods[
2694
2674
  client._client._transport.update_cluster
2695
2675
  ] = mock_object
@@ -3165,12 +3145,7 @@ async def test_update_node_pool_async_use_cached_wrapped_rpc(
3165
3145
  )
3166
3146
 
3167
3147
  # Replace cached wrapped function with mock
3168
- class AwaitableMock(mock.AsyncMock):
3169
- def __await__(self):
3170
- self.await_count += 1
3171
- return iter([])
3172
-
3173
- mock_object = AwaitableMock()
3148
+ mock_object = mock.AsyncMock()
3174
3149
  client._client._transport._wrapped_methods[
3175
3150
  client._client._transport.update_node_pool
3176
3151
  ] = mock_object
@@ -3523,12 +3498,7 @@ async def test_set_node_pool_autoscaling_async_use_cached_wrapped_rpc(
3523
3498
  )
3524
3499
 
3525
3500
  # Replace cached wrapped function with mock
3526
- class AwaitableMock(mock.AsyncMock):
3527
- def __await__(self):
3528
- self.await_count += 1
3529
- return iter([])
3530
-
3531
- mock_object = AwaitableMock()
3501
+ mock_object = mock.AsyncMock()
3532
3502
  client._client._transport._wrapped_methods[
3533
3503
  client._client._transport.set_node_pool_autoscaling
3534
3504
  ] = mock_object
@@ -3887,12 +3857,7 @@ async def test_set_logging_service_async_use_cached_wrapped_rpc(
3887
3857
  )
3888
3858
 
3889
3859
  # Replace cached wrapped function with mock
3890
- class AwaitableMock(mock.AsyncMock):
3891
- def __await__(self):
3892
- self.await_count += 1
3893
- return iter([])
3894
-
3895
- mock_object = AwaitableMock()
3860
+ mock_object = mock.AsyncMock()
3896
3861
  client._client._transport._wrapped_methods[
3897
3862
  client._client._transport.set_logging_service
3898
3863
  ] = mock_object
@@ -4368,12 +4333,7 @@ async def test_set_monitoring_service_async_use_cached_wrapped_rpc(
4368
4333
  )
4369
4334
 
4370
4335
  # Replace cached wrapped function with mock
4371
- class AwaitableMock(mock.AsyncMock):
4372
- def __await__(self):
4373
- self.await_count += 1
4374
- return iter([])
4375
-
4376
- mock_object = AwaitableMock()
4336
+ mock_object = mock.AsyncMock()
4377
4337
  client._client._transport._wrapped_methods[
4378
4338
  client._client._transport.set_monitoring_service
4379
4339
  ] = mock_object
@@ -4844,12 +4804,7 @@ async def test_set_addons_config_async_use_cached_wrapped_rpc(
4844
4804
  )
4845
4805
 
4846
4806
  # Replace cached wrapped function with mock
4847
- class AwaitableMock(mock.AsyncMock):
4848
- def __await__(self):
4849
- self.await_count += 1
4850
- return iter([])
4851
-
4852
- mock_object = AwaitableMock()
4807
+ mock_object = mock.AsyncMock()
4853
4808
  client._client._transport._wrapped_methods[
4854
4809
  client._client._transport.set_addons_config
4855
4810
  ] = mock_object
@@ -5321,12 +5276,7 @@ async def test_set_locations_async_use_cached_wrapped_rpc(
5321
5276
  )
5322
5277
 
5323
5278
  # Replace cached wrapped function with mock
5324
- class AwaitableMock(mock.AsyncMock):
5325
- def __await__(self):
5326
- self.await_count += 1
5327
- return iter([])
5328
-
5329
- mock_object = AwaitableMock()
5279
+ mock_object = mock.AsyncMock()
5330
5280
  client._client._transport._wrapped_methods[
5331
5281
  client._client._transport.set_locations
5332
5282
  ] = mock_object
@@ -5778,12 +5728,7 @@ async def test_update_master_async_use_cached_wrapped_rpc(
5778
5728
  )
5779
5729
 
5780
5730
  # Replace cached wrapped function with mock
5781
- class AwaitableMock(mock.AsyncMock):
5782
- def __await__(self):
5783
- self.await_count += 1
5784
- return iter([])
5785
-
5786
- mock_object = AwaitableMock()
5731
+ mock_object = mock.AsyncMock()
5787
5732
  client._client._transport._wrapped_methods[
5788
5733
  client._client._transport.update_master
5789
5734
  ] = mock_object
@@ -6233,12 +6178,7 @@ async def test_set_master_auth_async_use_cached_wrapped_rpc(
6233
6178
  )
6234
6179
 
6235
6180
  # Replace cached wrapped function with mock
6236
- class AwaitableMock(mock.AsyncMock):
6237
- def __await__(self):
6238
- self.await_count += 1
6239
- return iter([])
6240
-
6241
- mock_object = AwaitableMock()
6181
+ mock_object = mock.AsyncMock()
6242
6182
  client._client._transport._wrapped_methods[
6243
6183
  client._client._transport.set_master_auth
6244
6184
  ] = mock_object
@@ -6576,12 +6516,7 @@ async def test_delete_cluster_async_use_cached_wrapped_rpc(
6576
6516
  )
6577
6517
 
6578
6518
  # Replace cached wrapped function with mock
6579
- class AwaitableMock(mock.AsyncMock):
6580
- def __await__(self):
6581
- self.await_count += 1
6582
- return iter([])
6583
-
6584
- mock_object = AwaitableMock()
6519
+ mock_object = mock.AsyncMock()
6585
6520
  client._client._transport._wrapped_methods[
6586
6521
  client._client._transport.delete_cluster
6587
6522
  ] = mock_object
@@ -6989,12 +6924,7 @@ async def test_list_operations_async_use_cached_wrapped_rpc(
6989
6924
  )
6990
6925
 
6991
6926
  # Replace cached wrapped function with mock
6992
- class AwaitableMock(mock.AsyncMock):
6993
- def __await__(self):
6994
- self.await_count += 1
6995
- return iter([])
6996
-
6997
- mock_object = AwaitableMock()
6927
+ mock_object = mock.AsyncMock()
6998
6928
  client._client._transport._wrapped_methods[
6999
6929
  client._client._transport.list_operations
7000
6930
  ] = mock_object
@@ -7404,12 +7334,7 @@ async def test_get_operation_async_use_cached_wrapped_rpc(
7404
7334
  )
7405
7335
 
7406
7336
  # Replace cached wrapped function with mock
7407
- class AwaitableMock(mock.AsyncMock):
7408
- def __await__(self):
7409
- self.await_count += 1
7410
- return iter([])
7411
-
7412
- mock_object = AwaitableMock()
7337
+ mock_object = mock.AsyncMock()
7413
7338
  client._client._transport._wrapped_methods[
7414
7339
  client._client._transport.get_operation
7415
7340
  ] = mock_object
@@ -7814,12 +7739,7 @@ async def test_cancel_operation_async_use_cached_wrapped_rpc(
7814
7739
  )
7815
7740
 
7816
7741
  # Replace cached wrapped function with mock
7817
- class AwaitableMock(mock.AsyncMock):
7818
- def __await__(self):
7819
- self.await_count += 1
7820
- return iter([])
7821
-
7822
- mock_object = AwaitableMock()
7742
+ mock_object = mock.AsyncMock()
7823
7743
  client._client._transport._wrapped_methods[
7824
7744
  client._client._transport.cancel_operation
7825
7745
  ] = mock_object
@@ -8220,12 +8140,7 @@ async def test_get_server_config_async_use_cached_wrapped_rpc(
8220
8140
  )
8221
8141
 
8222
8142
  # Replace cached wrapped function with mock
8223
- class AwaitableMock(mock.AsyncMock):
8224
- def __await__(self):
8225
- self.await_count += 1
8226
- return iter([])
8227
-
8228
- mock_object = AwaitableMock()
8143
+ mock_object = mock.AsyncMock()
8229
8144
  client._client._transport._wrapped_methods[
8230
8145
  client._client._transport.get_server_config
8231
8146
  ] = mock_object
@@ -8622,12 +8537,7 @@ async def test_get_json_web_keys_async_use_cached_wrapped_rpc(
8622
8537
  )
8623
8538
 
8624
8539
  # Replace cached wrapped function with mock
8625
- class AwaitableMock(mock.AsyncMock):
8626
- def __await__(self):
8627
- self.await_count += 1
8628
- return iter([])
8629
-
8630
- mock_object = AwaitableMock()
8540
+ mock_object = mock.AsyncMock()
8631
8541
  client._client._transport._wrapped_methods[
8632
8542
  client._client._transport.get_json_web_keys
8633
8543
  ] = mock_object
@@ -8913,12 +8823,7 @@ async def test_list_node_pools_async_use_cached_wrapped_rpc(
8913
8823
  )
8914
8824
 
8915
8825
  # Replace cached wrapped function with mock
8916
- class AwaitableMock(mock.AsyncMock):
8917
- def __await__(self):
8918
- self.await_count += 1
8919
- return iter([])
8920
-
8921
- mock_object = AwaitableMock()
8826
+ mock_object = mock.AsyncMock()
8922
8827
  client._client._transport._wrapped_methods[
8923
8828
  client._client._transport.list_node_pools
8924
8829
  ] = mock_object
@@ -9334,12 +9239,7 @@ async def test_get_node_pool_async_use_cached_wrapped_rpc(
9334
9239
  )
9335
9240
 
9336
9241
  # Replace cached wrapped function with mock
9337
- class AwaitableMock(mock.AsyncMock):
9338
- def __await__(self):
9339
- self.await_count += 1
9340
- return iter([])
9341
-
9342
- mock_object = AwaitableMock()
9242
+ mock_object = mock.AsyncMock()
9343
9243
  client._client._transport._wrapped_methods[
9344
9244
  client._client._transport.get_node_pool
9345
9245
  ] = mock_object
@@ -9789,12 +9689,7 @@ async def test_create_node_pool_async_use_cached_wrapped_rpc(
9789
9689
  )
9790
9690
 
9791
9691
  # Replace cached wrapped function with mock
9792
- class AwaitableMock(mock.AsyncMock):
9793
- def __await__(self):
9794
- self.await_count += 1
9795
- return iter([])
9796
-
9797
- mock_object = AwaitableMock()
9692
+ mock_object = mock.AsyncMock()
9798
9693
  client._client._transport._wrapped_methods[
9799
9694
  client._client._transport.create_node_pool
9800
9695
  ] = mock_object
@@ -10248,12 +10143,7 @@ async def test_delete_node_pool_async_use_cached_wrapped_rpc(
10248
10143
  )
10249
10144
 
10250
10145
  # Replace cached wrapped function with mock
10251
- class AwaitableMock(mock.AsyncMock):
10252
- def __await__(self):
10253
- self.await_count += 1
10254
- return iter([])
10255
-
10256
- mock_object = AwaitableMock()
10146
+ mock_object = mock.AsyncMock()
10257
10147
  client._client._transport._wrapped_methods[
10258
10148
  client._client._transport.delete_node_pool
10259
10149
  ] = mock_object
@@ -10673,12 +10563,7 @@ async def test_complete_node_pool_upgrade_async_use_cached_wrapped_rpc(
10673
10563
  )
10674
10564
 
10675
10565
  # Replace cached wrapped function with mock
10676
- class AwaitableMock(mock.AsyncMock):
10677
- def __await__(self):
10678
- self.await_count += 1
10679
- return iter([])
10680
-
10681
- mock_object = AwaitableMock()
10566
+ mock_object = mock.AsyncMock()
10682
10567
  client._client._transport._wrapped_methods[
10683
10568
  client._client._transport.complete_node_pool_upgrade
10684
10569
  ] = mock_object
@@ -11011,12 +10896,7 @@ async def test_rollback_node_pool_upgrade_async_use_cached_wrapped_rpc(
11011
10896
  )
11012
10897
 
11013
10898
  # Replace cached wrapped function with mock
11014
- class AwaitableMock(mock.AsyncMock):
11015
- def __await__(self):
11016
- self.await_count += 1
11017
- return iter([])
11018
-
11019
- mock_object = AwaitableMock()
10899
+ mock_object = mock.AsyncMock()
11020
10900
  client._client._transport._wrapped_methods[
11021
10901
  client._client._transport.rollback_node_pool_upgrade
11022
10902
  ] = mock_object
@@ -11492,12 +11372,7 @@ async def test_set_node_pool_management_async_use_cached_wrapped_rpc(
11492
11372
  )
11493
11373
 
11494
11374
  # Replace cached wrapped function with mock
11495
- class AwaitableMock(mock.AsyncMock):
11496
- def __await__(self):
11497
- self.await_count += 1
11498
- return iter([])
11499
-
11500
- mock_object = AwaitableMock()
11375
+ mock_object = mock.AsyncMock()
11501
11376
  client._client._transport._wrapped_methods[
11502
11377
  client._client._transport.set_node_pool_management
11503
11378
  ] = mock_object
@@ -11968,12 +11843,7 @@ async def test_set_labels_async_use_cached_wrapped_rpc(transport: str = "grpc_as
11968
11843
  )
11969
11844
 
11970
11845
  # Replace cached wrapped function with mock
11971
- class AwaitableMock(mock.AsyncMock):
11972
- def __await__(self):
11973
- self.await_count += 1
11974
- return iter([])
11975
-
11976
- mock_object = AwaitableMock()
11846
+ mock_object = mock.AsyncMock()
11977
11847
  client._client._transport._wrapped_methods[
11978
11848
  client._client._transport.set_labels
11979
11849
  ] = mock_object
@@ -12433,12 +12303,7 @@ async def test_set_legacy_abac_async_use_cached_wrapped_rpc(
12433
12303
  )
12434
12304
 
12435
12305
  # Replace cached wrapped function with mock
12436
- class AwaitableMock(mock.AsyncMock):
12437
- def __await__(self):
12438
- self.await_count += 1
12439
- return iter([])
12440
-
12441
- mock_object = AwaitableMock()
12306
+ mock_object = mock.AsyncMock()
12442
12307
  client._client._transport._wrapped_methods[
12443
12308
  client._client._transport.set_legacy_abac
12444
12309
  ] = mock_object
@@ -12898,12 +12763,7 @@ async def test_start_ip_rotation_async_use_cached_wrapped_rpc(
12898
12763
  )
12899
12764
 
12900
12765
  # Replace cached wrapped function with mock
12901
- class AwaitableMock(mock.AsyncMock):
12902
- def __await__(self):
12903
- self.await_count += 1
12904
- return iter([])
12905
-
12906
- mock_object = AwaitableMock()
12766
+ mock_object = mock.AsyncMock()
12907
12767
  client._client._transport._wrapped_methods[
12908
12768
  client._client._transport.start_ip_rotation
12909
12769
  ] = mock_object
@@ -13365,12 +13225,7 @@ async def test_complete_ip_rotation_async_use_cached_wrapped_rpc(
13365
13225
  )
13366
13226
 
13367
13227
  # Replace cached wrapped function with mock
13368
- class AwaitableMock(mock.AsyncMock):
13369
- def __await__(self):
13370
- self.await_count += 1
13371
- return iter([])
13372
-
13373
- mock_object = AwaitableMock()
13228
+ mock_object = mock.AsyncMock()
13374
13229
  client._client._transport._wrapped_methods[
13375
13230
  client._client._transport.complete_ip_rotation
13376
13231
  ] = mock_object
@@ -13835,12 +13690,7 @@ async def test_set_node_pool_size_async_use_cached_wrapped_rpc(
13835
13690
  )
13836
13691
 
13837
13692
  # Replace cached wrapped function with mock
13838
- class AwaitableMock(mock.AsyncMock):
13839
- def __await__(self):
13840
- self.await_count += 1
13841
- return iter([])
13842
-
13843
- mock_object = AwaitableMock()
13693
+ mock_object = mock.AsyncMock()
13844
13694
  client._client._transport._wrapped_methods[
13845
13695
  client._client._transport.set_node_pool_size
13846
13696
  ] = mock_object
@@ -14196,12 +14046,7 @@ async def test_set_network_policy_async_use_cached_wrapped_rpc(
14196
14046
  )
14197
14047
 
14198
14048
  # Replace cached wrapped function with mock
14199
- class AwaitableMock(mock.AsyncMock):
14200
- def __await__(self):
14201
- self.await_count += 1
14202
- return iter([])
14203
-
14204
- mock_object = AwaitableMock()
14049
+ mock_object = mock.AsyncMock()
14205
14050
  client._client._transport._wrapped_methods[
14206
14051
  client._client._transport.set_network_policy
14207
14052
  ] = mock_object
@@ -14687,12 +14532,7 @@ async def test_set_maintenance_policy_async_use_cached_wrapped_rpc(
14687
14532
  )
14688
14533
 
14689
14534
  # Replace cached wrapped function with mock
14690
- class AwaitableMock(mock.AsyncMock):
14691
- def __await__(self):
14692
- self.await_count += 1
14693
- return iter([])
14694
-
14695
- mock_object = AwaitableMock()
14535
+ mock_object = mock.AsyncMock()
14696
14536
  client._client._transport._wrapped_methods[
14697
14537
  client._client._transport.set_maintenance_policy
14698
14538
  ] = mock_object
@@ -15170,12 +15010,7 @@ async def test_list_usable_subnetworks_async_use_cached_wrapped_rpc(
15170
15010
  )
15171
15011
 
15172
15012
  # Replace cached wrapped function with mock
15173
- class AwaitableMock(mock.AsyncMock):
15174
- def __await__(self):
15175
- self.await_count += 1
15176
- return iter([])
15177
-
15178
- mock_object = AwaitableMock()
15013
+ mock_object = mock.AsyncMock()
15179
15014
  client._client._transport._wrapped_methods[
15180
15015
  client._client._transport.list_usable_subnetworks
15181
15016
  ] = mock_object
@@ -15761,12 +15596,7 @@ async def test_check_autopilot_compatibility_async_use_cached_wrapped_rpc(
15761
15596
  )
15762
15597
 
15763
15598
  # Replace cached wrapped function with mock
15764
- class AwaitableMock(mock.AsyncMock):
15765
- def __await__(self):
15766
- self.await_count += 1
15767
- return iter([])
15768
-
15769
- mock_object = AwaitableMock()
15599
+ mock_object = mock.AsyncMock()
15770
15600
  client._client._transport._wrapped_methods[
15771
15601
  client._client._transport.check_autopilot_compatibility
15772
15602
  ] = mock_object
@@ -16056,12 +15886,7 @@ async def test_list_locations_async_use_cached_wrapped_rpc(
16056
15886
  )
16057
15887
 
16058
15888
  # Replace cached wrapped function with mock
16059
- class AwaitableMock(mock.AsyncMock):
16060
- def __await__(self):
16061
- self.await_count += 1
16062
- return iter([])
16063
-
16064
- mock_object = AwaitableMock()
15889
+ mock_object = mock.AsyncMock()
16065
15890
  client._client._transport._wrapped_methods[
16066
15891
  client._client._transport.list_locations
16067
15892
  ] = mock_object