def test_get_recommendation(transport: str = 'grpc'): client = RewardRecommendationClient( 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 = reward_recommendation.RewardRecommendationRequest() # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.get_recommendation), '__call__') as call: # Designate an appropriate return value for the call. call.return_value = reward_recommendation.RewardRecommendationResponse( reward_ids=['reward_ids_value'], ) response = client.get_recommendation(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, reward_recommendation.RewardRecommendationResponse) assert response.reward_ids == ['reward_ids_value']
def test_reward_recommendation_client_from_service_account_file(): creds = credentials.AnonymousCredentials() with mock.patch.object(service_account.Credentials, 'from_service_account_file') as factory: factory.return_value = creds client = RewardRecommendationClient.from_service_account_file( "dummy/file/path.json") assert client._transport._credentials == creds client = RewardRecommendationClient.from_service_account_json( "dummy/file/path.json") assert client._transport._credentials == creds assert client._transport._host == 'ml-api.data.perxtech.net:443'
def test_reward_recommendation_host_with_port(): client = RewardRecommendationClient( credentials=credentials.AnonymousCredentials(), client_options=client_options.ClientOptions( api_endpoint='ml-api.data.perxtech.net:8000'), transport='grpc', ) assert client._transport._host == 'ml-api.data.perxtech.net:8000'
def test_transport_grpc_default(): # A client should use the gRPC transport by default. client = RewardRecommendationClient( credentials=credentials.AnonymousCredentials(), ) assert isinstance( client._transport, transports.RewardRecommendationGrpcTransport, )
def test_reward_recommendation_client_client_options_from_dict(): with mock.patch( 'perxtech.ml_v1beta1.services.reward_recommendation.RewardRecommendationClient.get_transport_class' ) as gtc: transport = gtc.return_value = mock.MagicMock() client = RewardRecommendationClient( client_options={'api_endpoint': 'squid.clam.whelk'}) transport.assert_called_once_with(credentials=None, host="squid.clam.whelk")
def test_credentials_transport_error(): # It is an error to provide credentials and a transport instance. transport = transports.RewardRecommendationGrpcTransport( credentials=credentials.AnonymousCredentials(), ) with pytest.raises(ValueError): client = RewardRecommendationClient( credentials=credentials.AnonymousCredentials(), transport=transport, )
def test_reward_recommendation_client_client_options(): # Check the default options have their expected values. assert RewardRecommendationClient.DEFAULT_OPTIONS.api_endpoint == 'ml-api.data.perxtech.net' # Check that options can be customized. options = client_options.ClientOptions(api_endpoint="squid.clam.whelk") with mock.patch( 'perxtech.ml_v1beta1.services.reward_recommendation.RewardRecommendationClient.get_transport_class' ) as gtc: transport = gtc.return_value = mock.MagicMock() client = RewardRecommendationClient(client_options=options) transport.assert_called_once_with(credentials=None, host="squid.clam.whelk")
def test_reward_recommendation_auth_adc(): # If no credentials are provided, we should use ADC credentials. with mock.patch.object(auth, 'default') as adc: adc.return_value = (credentials.AnonymousCredentials(), None) RewardRecommendationClient() adc.assert_called_once_with(scopes=())
def test_transport_instance(): # A client may be instantiated with a custom transport instance. transport = transports.RewardRecommendationGrpcTransport( credentials=credentials.AnonymousCredentials(), ) client = RewardRecommendationClient(transport=transport) assert client._transport is transport