def test_update_project_billing_info_flattened(): client = CloudBillingClient(credentials=credentials.AnonymousCredentials()) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.update_project_billing_info), "__call__" ) as call: # Designate an appropriate return value for the call. call.return_value = cloud_billing.ProjectBillingInfo() # Call the method with a truthy value for each flattened field, # using the keyword arguments to the method. response = client.update_project_billing_info( name="name_value", project_billing_info=cloud_billing.ProjectBillingInfo(name="name_value"), ) # Establish that the underlying call was made with the expected # request object values. assert len(call.mock_calls) == 1 _, args, _ = call.mock_calls[0] assert args[0].name == "name_value" assert args[0].project_billing_info == cloud_billing.ProjectBillingInfo( name="name_value" )
def test_update_project_billing_info(transport: str = "grpc"): client = CloudBillingClient( credentials=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 = cloud_billing.UpdateProjectBillingInfoRequest() # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.update_project_billing_info), "__call__" ) as call: # Designate an appropriate return value for the call. call.return_value = cloud_billing.ProjectBillingInfo( name="name_value", project_id="project_id_value", billing_account_name="billing_account_name_value", billing_enabled=True, ) response = client.update_project_billing_info(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 response is the type that we expect. assert isinstance(response, cloud_billing.ProjectBillingInfo) assert response.name == "name_value" assert response.project_id == "project_id_value" assert response.billing_account_name == "billing_account_name_value" assert response.billing_enabled == True
def test_update_project_billing_info_flattened_error(): client = CloudBillingClient(credentials=credentials.AnonymousCredentials()) # Attempting to call a method with both a request object and flattened # fields is an error. with pytest.raises(ValueError): client.update_project_billing_info( cloud_billing.UpdateProjectBillingInfoRequest(), name="name_value", project_billing_info=cloud_billing.ProjectBillingInfo(name="name_value"), )
def test_list_project_billing_info_pages(): client = CloudBillingClient(credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_project_billing_info), "__call__" ) as call: # Set the response to a series of pages. call.side_effect = ( cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[ cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), ], next_page_token="abc", ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[], next_page_token="def" ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[cloud_billing.ProjectBillingInfo()], next_page_token="ghi", ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[ cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), ] ), RuntimeError, ) pages = list(client.list_project_billing_info(request={}).pages) for page, token in zip(pages, ["abc", "def", "ghi", ""]): assert page.raw_page.next_page_token == token
def test_list_project_billing_info_pager(): client = CloudBillingClient(credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_project_billing_info), "__call__" ) as call: # Set the response to a series of pages. call.side_effect = ( cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[ cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), ], next_page_token="abc", ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[], next_page_token="def" ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[cloud_billing.ProjectBillingInfo()], next_page_token="ghi", ), cloud_billing.ListProjectBillingInfoResponse( project_billing_info=[ cloud_billing.ProjectBillingInfo(), cloud_billing.ProjectBillingInfo(), ] ), RuntimeError, ) results = [i for i in client.list_project_billing_info(request={})] assert len(results) == 6 assert all([isinstance(i, cloud_billing.ProjectBillingInfo) for i in results])
def test_get_project_billing_info_field_headers(): client = CloudBillingClient(credentials=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 = cloud_billing.GetProjectBillingInfoRequest(name="name/value") # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.get_project_billing_info), "__call__") as call: call.return_value = cloud_billing.ProjectBillingInfo() client.get_project_billing_info(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", "name=name/value") in kw["metadata"]