def test_get_billing_setup_field_headers():
client = BillingSetupServiceClient(
credentials=ga_credentials.AnonymousCredentials(), )
# Any value that is part of the HTTP/1.1 URI should be sent as
# a field header. Set these to a non-empty value.
request = billing_setup_service.GetBillingSetupRequest()
request.resource_name = 'resource_name/value'
# Mock the actual call within the gRPC stub, and fake the request.
with mock.patch.object(type(client.transport.get_billing_setup),
'__call__') as call:
call.return_value = billing_setup.BillingSetup()
client.get_billing_setup(request)
# Establish that the underlying gRPC stub method was called.
assert len(call.mock_calls) == 1
_, args, _ = call.mock_calls[0]
assert args[0] == request
# Establish that the field header was sent.
_, _, kw = call.mock_calls[0]
assert (
'x-goog-request-params',
'resource_name=resource_name/value',
) in kw['metadata']
def test_get_billing_setup(
transport: str = 'grpc',
request_type=billing_setup_service.GetBillingSetupRequest):
client = BillingSetupServiceClient(
credentials=ga_credentials.AnonymousCredentials(),
transport=transport,
)
# Everything is optional in proto3 as far as the runtime is concerned,
# and we are mocking out the actual API, so just send an empty request.
request = request_type()
# Mock the actual call within the gRPC stub, and fake the request.
with mock.patch.object(type(client.transport.get_billing_setup),
'__call__') as call:
# Designate an appropriate return value for the call.
call.return_value = billing_setup.BillingSetup(
resource_name='resource_name_value',
status=billing_setup_status.BillingSetupStatusEnum.
BillingSetupStatus.UNKNOWN,
start_date_time=wrappers_pb2.StringValue(value='value_value'),
end_date_time=wrappers_pb2.StringValue(value='value_value'),
)
response = client.get_billing_setup(request)
# Establish that the underlying gRPC stub method was called.
assert len(call.mock_calls) == 1
_, args, _ = call.mock_calls[0]
assert args[0] == billing_setup_service.GetBillingSetupRequest()
# Establish that the response is the type that we expect.
assert isinstance(response, billing_setup.BillingSetup)
assert response.resource_name == 'resource_name_value'
assert response.status == billing_setup_status.BillingSetupStatusEnum.BillingSetupStatus.UNKNOWN
def test_get_billing_setup_flattened_error():
client = BillingSetupServiceClient(
credentials=ga_credentials.AnonymousCredentials(), )
# Attempting to call a method with both a request object and flattened
# fields is an error.
with pytest.raises(ValueError):
client.get_billing_setup(
billing_setup_service.GetBillingSetupRequest(),
resource_name='resource_name_value',
)
def get_billing_setup(
self,
request: billing_setup_service.GetBillingSetupRequest = None,
*,
resource_name: str = None,
retry: retries.Retry = gapic_v1.method.DEFAULT,
timeout: float = None,
metadata: Sequence[Tuple[str, str]] = (),
) -> billing_setup.BillingSetup:
r"""Returns a billing setup.
Args:
request (:class:`google.ads.googleads.v5.services.types.GetBillingSetupRequest`):
The request object. Request message for
[BillingSetupService.GetBillingSetup][google.ads.googleads.v5.services.BillingSetupService.GetBillingSetup].
resource_name (:class:`str`):
Required. The resource name of the
billing setup to fetch.
This corresponds to the ``resource_name`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
retry (google.api_core.retry.Retry): Designation of what errors, if any,
should be retried.
timeout (float): The timeout for this request.
metadata (Sequence[Tuple[str, str]]): Strings which should be
sent along with the request as metadata.
Returns:
google.ads.googleads.v5.resources.types.BillingSetup:
A billing setup, which associates a
payments account and an advertiser. A
billing setup is specific to one
advertiser.
"""
# Create or coerce a protobuf request object.
# Sanity check: If we got a request object, we should *not* have
# gotten any keyword arguments that map to the request.
if request is not None and any([resource_name]):
raise ValueError("If the `request` argument is set, then none of "
"the individual field arguments should be set.")
# Minor optimization to avoid making a copy if the user passes
# in a billing_setup_service.GetBillingSetupRequest.
# There's no risk of modifying the input as we've already verified
# there are no flattened fields.
if not isinstance(request,
billing_setup_service.GetBillingSetupRequest):
request = billing_setup_service.GetBillingSetupRequest(request)
# If we have keyword arguments corresponding to fields on the
# request, apply these.
if resource_name is not None:
request.resource_name = resource_name
# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = self._transport._wrapped_methods[
self._transport.get_billing_setup]
# Certain fields should be provided within the metadata header;
# add these here.
metadata = tuple(metadata) + (gapic_v1.routing_header.to_grpc_metadata(
(("resource_name", request.resource_name), )), )
# Send the request.
response = rpc(
request,
retry=retry,
timeout=timeout,
metadata=metadata,
)
# Done; return the response.
return response