google-cloud-cloudcontrolspartner 0.1.3__tar.gz → 0.2.0__tar.gz

{google_cloud_cloudcontrolspartner-0.1.3/google_cloud_cloudcontrolspartner.egg-info → google_cloud_cloudcontrolspartner-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-cloud-cloudcontrolspartner
-Version: 0.1.3
+Version: 0.2.0
 Summary: Google Cloud Cloudcontrolspartner API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-cloudcontrolspartner
 Author: Google LLC

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.3"  # {x-release-please-version}
+__version__ = "0.2.0"  # {x-release-please-version}

{google_cloud_cloudcontrolspartner-0.1.3/google/cloud/cloudcontrolspartner_v1beta → google_cloud_cloudcontrolspartner-0.2.0/google/cloud/cloudcontrolspartner_v1}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.3"  # {x-release-please-version}
+__version__ = "0.2.0"  # {x-release-please-version}

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/services/cloud_controls_partner_core/async_client.py

@@ -14,7 +14,6 @@
 # limitations under the License.
 #
 from collections import OrderedDict
-import functools
 import re
 from typing import (
     Callable,
@@ -231,10 +230,7 @@ class CloudControlsPartnerCoreAsyncClient:
         """
         return self._client._universe_domain
 
-    get_transport_class = functools.partial(
-        type(CloudControlsPartnerCoreClient).get_transport_class,
-        type(CloudControlsPartnerCoreClient),
-    )
+    get_transport_class = CloudControlsPartnerCoreClient.get_transport_class
 
     def __init__(
         self,

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/services/cloud_controls_partner_core/client.py

@@ -798,7 +798,7 @@ class CloudControlsPartnerCoreClient(metaclass=CloudControlsPartnerCoreClientMet
                 Type[CloudControlsPartnerCoreTransport],
                 Callable[..., CloudControlsPartnerCoreTransport],
             ] = (
-                type(self).get_transport_class(transport)
+                CloudControlsPartnerCoreClient.get_transport_class(transport)
                 if isinstance(transport, str) or transport is None
                 else cast(Callable[..., CloudControlsPartnerCoreTransport], transport)
             )

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/services/cloud_controls_partner_monitoring/async_client.py

@@ -14,7 +14,6 @@
 # limitations under the License.
 #
 from collections import OrderedDict
-import functools
 import re
 from typing import (
     Callable,
@@ -204,10 +203,7 @@ class CloudControlsPartnerMonitoringAsyncClient:
         """
         return self._client._universe_domain
 
-    get_transport_class = functools.partial(
-        type(CloudControlsPartnerMonitoringClient).get_transport_class,
-        type(CloudControlsPartnerMonitoringClient),
-    )
+    get_transport_class = CloudControlsPartnerMonitoringClient.get_transport_class
 
     def __init__(
         self,

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/services/cloud_controls_partner_monitoring/client.py

@@ -688,7 +688,7 @@ class CloudControlsPartnerMonitoringClient(
                 Type[CloudControlsPartnerMonitoringTransport],
                 Callable[..., CloudControlsPartnerMonitoringTransport],
             ] = (
-                type(self).get_transport_class(transport)
+                CloudControlsPartnerMonitoringClient.get_transport_class(transport)
                 if isinstance(transport, str) or transport is None
                 else cast(
                     Callable[..., CloudControlsPartnerMonitoringTransport], transport

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/types/customers.py

@@ -45,13 +45,13 @@ class Customer(proto.Message):
             Identifier. Format:
             ``organizations/{organization}/locations/{location}/customers/{customer}``
         display_name (str):
-            The customer organization's display name.
-            E.g. "google.com".
+            Required. Display name for the customer
         customer_onboarding_state (google.cloud.cloudcontrolspartner_v1.types.CustomerOnboardingState):
-            Container for customer onboarding steps
+            Output only. Container for customer
+            onboarding steps
         is_onboarded (bool):
-            Indicates whether a customer is fully
-            onboarded
+            Output only. Indicates whether a customer is
+            fully onboarded
     """
 
     name: str = proto.Field(

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1/types/partner_permissions.py

@@ -57,12 +57,16 @@ class PartnerPermissions(proto.Message):
             ASSURED_WORKLOADS_EKM_CONNECTION_STATUS (4):
                 Permission for External Key Manager
                 connection status
+            ACCESS_TRANSPARENCY_LOGS_SUPPORT_CASE_VIEWER (5):
+                Permission for support case details for
+                Access Transparency log entries
         """
         PERMISSION_UNSPECIFIED = 0
         ACCESS_TRANSPARENCY_AND_EMERGENCY_ACCESS_LOGS = 1
         ASSURED_WORKLOADS_MONITORING = 2
         ACCESS_APPROVAL_REQUESTS = 3
         ASSURED_WORKLOADS_EKM_CONNECTION_STATUS = 4
+        ACCESS_TRANSPARENCY_LOGS_SUPPORT_CASE_VIEWER = 5
 
     name: str = proto.Field(
         proto.STRING,

{google_cloud_cloudcontrolspartner-0.1.3/google/cloud/cloudcontrolspartner_v1 → google_cloud_cloudcontrolspartner-0.2.0/google/cloud/cloudcontrolspartner_v1beta}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.3"  # {x-release-please-version}
+__version__ = "0.2.0"  # {x-release-please-version}

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/services/cloud_controls_partner_core/async_client.py

@@ -14,7 +14,6 @@
 # limitations under the License.
 #
 from collections import OrderedDict
-import functools
 import re
 from typing import (
     Callable,
@@ -231,10 +230,7 @@ class CloudControlsPartnerCoreAsyncClient:
         """
         return self._client._universe_domain
 
-    get_transport_class = functools.partial(
-        type(CloudControlsPartnerCoreClient).get_transport_class,
-        type(CloudControlsPartnerCoreClient),
-    )
+    get_transport_class = CloudControlsPartnerCoreClient.get_transport_class
 
     def __init__(
         self,

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/services/cloud_controls_partner_core/client.py

@@ -798,7 +798,7 @@ class CloudControlsPartnerCoreClient(metaclass=CloudControlsPartnerCoreClientMet
                 Type[CloudControlsPartnerCoreTransport],
                 Callable[..., CloudControlsPartnerCoreTransport],
             ] = (
-                type(self).get_transport_class(transport)
+                CloudControlsPartnerCoreClient.get_transport_class(transport)
                 if isinstance(transport, str) or transport is None
                 else cast(Callable[..., CloudControlsPartnerCoreTransport], transport)
             )

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/services/cloud_controls_partner_monitoring/async_client.py

@@ -14,7 +14,6 @@
 # limitations under the License.
 #
 from collections import OrderedDict
-import functools
 import re
 from typing import (
     Callable,
@@ -204,10 +203,7 @@ class CloudControlsPartnerMonitoringAsyncClient:
         """
         return self._client._universe_domain
 
-    get_transport_class = functools.partial(
-        type(CloudControlsPartnerMonitoringClient).get_transport_class,
-        type(CloudControlsPartnerMonitoringClient),
-    )
+    get_transport_class = CloudControlsPartnerMonitoringClient.get_transport_class
 
     def __init__(
         self,

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/services/cloud_controls_partner_monitoring/client.py

@@ -688,7 +688,7 @@ class CloudControlsPartnerMonitoringClient(
                 Type[CloudControlsPartnerMonitoringTransport],
                 Callable[..., CloudControlsPartnerMonitoringTransport],
             ] = (
-                type(self).get_transport_class(transport)
+                CloudControlsPartnerMonitoringClient.get_transport_class(transport)
                 if isinstance(transport, str) or transport is None
                 else cast(
                     Callable[..., CloudControlsPartnerMonitoringTransport], transport

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/types/customers.py

@@ -45,13 +45,13 @@ class Customer(proto.Message):
             Identifier. Format:
             ``organizations/{organization}/locations/{location}/customers/{customer}``
         display_name (str):
-            The customer organization's display name.
-            E.g. "google.com".
+            Required. Display name for the customer
         customer_onboarding_state (google.cloud.cloudcontrolspartner_v1beta.types.CustomerOnboardingState):
-            Container for customer onboarding steps
+            Output only. Container for customer
+            onboarding steps
         is_onboarded (bool):
-            Indicates whether a customer is fully
-            onboarded
+            Output only. Indicates whether a customer is
+            fully onboarded
     """
 
     name: str = proto.Field(

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/google/cloud/cloudcontrolspartner_v1beta/types/partner_permissions.py

@@ -57,12 +57,16 @@ class PartnerPermissions(proto.Message):
             ASSURED_WORKLOADS_EKM_CONNECTION_STATUS (4):
                 Permission for External Key Manager
                 connection status
+            ACCESS_TRANSPARENCY_LOGS_SUPPORT_CASE_VIEWER (5):
+                Permission for support case details for
+                Access Transparency log entries
         """
         PERMISSION_UNSPECIFIED = 0
         ACCESS_TRANSPARENCY_AND_EMERGENCY_ACCESS_LOGS = 1
         ASSURED_WORKLOADS_MONITORING = 2
         ACCESS_APPROVAL_REQUESTS = 3
         ASSURED_WORKLOADS_EKM_CONNECTION_STATUS = 4
+        ACCESS_TRANSPARENCY_LOGS_SUPPORT_CASE_VIEWER = 5
 
     name: str = proto.Field(
         proto.STRING,

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0/google_cloud_cloudcontrolspartner.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-cloud-cloudcontrolspartner
-Version: 0.1.3
+Version: 0.2.0
 Summary: Google Cloud Cloudcontrolspartner API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-cloudcontrolspartner
 Author: Google LLC

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/tests/unit/gapic/cloudcontrolspartner_v1/test_cloud_controls_partner_core.py

@@ -1388,22 +1388,23 @@ async def test_get_workload_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_workload
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_workload(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_workload(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -1776,22 +1777,23 @@ async def test_list_workloads_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_workloads
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_workloads(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_workloads(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -2343,22 +2345,23 @@ async def test_get_customer_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_customer
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_customer(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_customer(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -2716,22 +2719,23 @@ async def test_list_customers_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_customers
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_customers(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_customers(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -3288,22 +3292,23 @@ async def test_get_ekm_connections_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_ekm_connections
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_ekm_connections(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_ekm_connections(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -3685,22 +3690,23 @@ async def test_get_partner_permissions_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_partner_permissions
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_partner_permissions(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_partner_permissions(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -4088,22 +4094,23 @@ async def test_list_access_approval_requests_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_access_approval_requests
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_access_approval_requests(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_access_approval_requests(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -4687,22 +4694,23 @@ async def test_get_partner_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_partner
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_partner(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_partner(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/tests/unit/gapic/cloudcontrolspartner_v1/test_cloud_controls_partner_monitoring.py

@@ -1389,22 +1389,23 @@ async def test_list_violations_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_violations
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_violations(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_violations(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -1964,22 +1965,23 @@ async def test_get_violation_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_violation
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_violation(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_violation(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/tests/unit/gapic/cloudcontrolspartner_v1beta/test_cloud_controls_partner_core.py

@@ -1388,22 +1388,23 @@ async def test_get_workload_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_workload
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_workload(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_workload(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -1776,22 +1777,23 @@ async def test_list_workloads_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_workloads
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_workloads(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_workloads(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -2343,22 +2345,23 @@ async def test_get_customer_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_customer
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_customer(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_customer(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -2716,22 +2719,23 @@ async def test_list_customers_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_customers
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_customers(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_customers(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -3288,22 +3292,23 @@ async def test_get_ekm_connections_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_ekm_connections
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_ekm_connections(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_ekm_connections(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -3685,22 +3690,23 @@ async def test_get_partner_permissions_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_partner_permissions
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_partner_permissions(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_partner_permissions(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -4088,22 +4094,23 @@ async def test_list_access_approval_requests_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_access_approval_requests
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_access_approval_requests(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_access_approval_requests(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -4687,22 +4694,23 @@ async def test_get_partner_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_partner
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_partner(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_partner(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio

{google_cloud_cloudcontrolspartner-0.1.3 → google_cloud_cloudcontrolspartner-0.2.0}/tests/unit/gapic/cloudcontrolspartner_v1beta/test_cloud_controls_partner_monitoring.py

@@ -1389,22 +1389,23 @@ async def test_list_violations_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.list_violations
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.list_violations(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.list_violations(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio
@@ -1964,22 +1965,23 @@ async def test_get_violation_async_use_cached_wrapped_rpc(
         )
 
         # Replace cached wrapped function with mock
-        mock_object = mock.AsyncMock()
+        mock_rpc = mock.AsyncMock()
+        mock_rpc.return_value = mock.Mock()
         client._client._transport._wrapped_methods[
             client._client._transport.get_violation
-        ] = mock_object
+        ] = mock_rpc
 
         request = {}
         await client.get_violation(request)
 
         # Establish that the underlying gRPC stub method was called.
-        assert mock_object.call_count == 1
+        assert mock_rpc.call_count == 1
 
         await client.get_violation(request)
 
         # Establish that a new wrapper was not created for this call
         assert wrapper_fn.call_count == 0
-        assert mock_object.call_count == 2
+        assert mock_rpc.call_count == 2
 
 
 @pytest.mark.asyncio