def test_list_prediction_api_key_registrations_field_headers(): client = PredictionApiKeyRegistryClient( 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 = prediction_apikey_registry_service.ListPredictionApiKeyRegistrationsRequest( parent="parent/value") # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_prediction_api_key_registrations), "__call__") as call: call.return_value = (prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse()) client.list_prediction_api_key_registrations(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", "parent=parent/value") in kw["metadata"]
def test_list_prediction_api_key_registrations(transport: str = "grpc"): client = PredictionApiKeyRegistryClient( 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 = (prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsRequest()) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_prediction_api_key_registrations), "__call__") as call: # Designate an appropriate return value for the call. call.return_value = prediction_apikey_registry_service.ListPredictionApiKeyRegistrationsResponse( next_page_token="next_page_token_value") response = client.list_prediction_api_key_registrations(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, pagers.ListPredictionApiKeyRegistrationsPager) assert response.next_page_token == "next_page_token_value"
def test_list_prediction_api_key_registrations_pager(): client = PredictionApiKeyRegistryClient( credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_prediction_api_key_registrations), "__call__") as call: # Set the response to a series of pages. call.side_effect = ( prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), ], next_page_token="abc", ), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[], next_page_token="def"), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration() ], next_page_token="ghi", ), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), ]), RuntimeError, ) results = [ i for i in client.list_prediction_api_key_registrations(request={}) ] assert len(results) == 6 assert all( isinstance( i, prediction_apikey_registry_service. PredictionApiKeyRegistration) for i in results)
def test_list_prediction_api_key_registrations_pages(): client = PredictionApiKeyRegistryClient( credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object( type(client._transport.list_prediction_api_key_registrations), "__call__") as call: # Set the response to a series of pages. call.side_effect = ( prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), ], next_page_token="abc", ), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[], next_page_token="def"), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration() ], next_page_token="ghi", ), prediction_apikey_registry_service. ListPredictionApiKeyRegistrationsResponse( prediction_api_key_registrations=[ prediction_apikey_registry_service. PredictionApiKeyRegistration(), prediction_apikey_registry_service. PredictionApiKeyRegistration(), ]), RuntimeError, ) pages = list( client.list_prediction_api_key_registrations(request={}).pages) for page, token in zip(pages, ["abc", "def", "ghi", ""]): assert page.raw_page.next_page_token == token