def test_credentials_transport_error(): # It is an error to provide credentials and a transport instance. transport = transports.LookupServiceGrpcTransport( credentials=ga_credentials.AnonymousCredentials(), ) with pytest.raises(ValueError): client = LookupServiceClient( credentials=ga_credentials.AnonymousCredentials(), transport=transport, ) # It is an error to provide a credentials file and a transport instance. transport = transports.LookupServiceGrpcTransport( credentials=ga_credentials.AnonymousCredentials(), ) with pytest.raises(ValueError): client = LookupServiceClient( client_options={"credentials_file": "credentials.json"}, transport=transport, ) # It is an error to provide scopes and a transport instance. transport = transports.LookupServiceGrpcTransport( credentials=ga_credentials.AnonymousCredentials(), ) with pytest.raises(ValueError): client = LookupServiceClient( client_options={"scopes": ["1", "2"]}, transport=transport, )
def test_lookup_service_grpc_transport_channel_mtls_with_adc( grpc_create_channel, api_mtls_endpoint): # Check that if channel and client_cert_source are None, but api_mtls_endpoint # is provided, then a mTLS channel will be created with SSL ADC. mock_grpc_channel = mock.Mock() grpc_create_channel.return_value = mock_grpc_channel # Mock google.auth.transport.grpc.SslCredentials class. mock_ssl_cred = mock.Mock() with mock.patch.multiple( "google.auth.transport.grpc.SslCredentials", __init__=mock.Mock(return_value=None), ssl_credentials=mock.PropertyMock(return_value=mock_ssl_cred), ): mock_cred = mock.Mock() transport = transports.LookupServiceGrpcTransport( host="squid.clam.whelk", credentials=mock_cred, api_mtls_endpoint=api_mtls_endpoint, client_cert_source=None, ) grpc_create_channel.assert_called_once_with( "mtls.squid.clam.whelk:443", credentials=mock_cred, ssl_credentials=mock_ssl_cred, scopes=("https://www.googleapis.com/auth/cloud-platform", ), ) assert transport.grpc_channel == mock_grpc_channel
def test_lookup_service_grpc_transport_channel_mtls_with_client_cert_source( grpc_create_channel, grpc_ssl_channel_cred): # Check that if channel is None, but api_mtls_endpoint and client_cert_source # are provided, then a mTLS channel will be created. mock_cred = mock.Mock() mock_ssl_cred = mock.Mock() grpc_ssl_channel_cred.return_value = mock_ssl_cred mock_grpc_channel = mock.Mock() grpc_create_channel.return_value = mock_grpc_channel transport = transports.LookupServiceGrpcTransport( host="squid.clam.whelk", credentials=mock_cred, api_mtls_endpoint="mtls.squid.clam.whelk", client_cert_source=client_cert_source_callback, ) grpc_ssl_channel_cred.assert_called_once_with( certificate_chain=b"cert bytes", private_key=b"key bytes") grpc_create_channel.assert_called_once_with( "mtls.squid.clam.whelk:443", credentials=mock_cred, ssl_credentials=mock_ssl_cred, scopes=("https://www.googleapis.com/auth/cloud-platform", ), ) assert transport.grpc_channel == mock_grpc_channel
def test_credentials_transport_error(): # It is an error to provide credentials and a transport instance. transport = transports.LookupServiceGrpcTransport( credentials=credentials.AnonymousCredentials()) with pytest.raises(ValueError): client = LookupServiceClient( credentials=credentials.AnonymousCredentials(), transport=transport)
def test_lookup_service_transport_auth_adc(): # If credentials and host are not provided, the transport class should use # ADC credentials. with mock.patch.object(auth, "default") as adc: adc.return_value = (credentials.AnonymousCredentials(), None) transports.LookupServiceGrpcTransport(host="squid.clam.whelk") adc.assert_called_once_with( scopes=("https://www.googleapis.com/auth/cloud-platform", ))
def test_transport_get_channel(): # A client may be instantiated with a custom transport instance. transport = transports.LookupServiceGrpcTransport( credentials=ga_credentials.AnonymousCredentials(), ) channel = transport.grpc_channel assert channel transport = transports.LookupServiceGrpcAsyncIOTransport( credentials=ga_credentials.AnonymousCredentials(), ) channel = transport.grpc_channel assert channel
def test_lookup_service_grpc_transport_channel(): channel = grpc.secure_channel("http://localhost/", grpc.local_channel_credentials()) # Check that channel is used if provided. transport = transports.LookupServiceGrpcTransport( host="squid.clam.whelk", channel=channel, ) assert transport.grpc_channel == channel assert transport._host == "squid.clam.whelk:443" assert transport._ssl_channel_credentials == None
def test_lookup_service_grpc_transport_channel(): channel = grpc.insecure_channel("http://localhost/") # Check that if channel is provided, mtls endpoint and client_cert_source # won't be used. callback = mock.MagicMock() transport = transports.LookupServiceGrpcTransport( host="squid.clam.whelk", channel=channel, api_mtls_endpoint="mtls.squid.clam.whelk", client_cert_source=callback, ) assert transport.grpc_channel == channel assert transport._host == "squid.clam.whelk:443" assert not callback.called
def test_lookup_service_client_client_options(): # Check that if channel is provided we won't create a new one. with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.LookupServiceClient.get_transport_class" ) as gtc: transport = transports.LookupServiceGrpcTransport( credentials=credentials.AnonymousCredentials()) client = LookupServiceClient(transport=transport) gtc.assert_not_called() # Check mTLS is not triggered with empty client options. options = client_options.ClientOptions() with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.LookupServiceClient.get_transport_class" ) as gtc: transport = gtc.return_value = mock.MagicMock() client = LookupServiceClient(client_options=options) transport.assert_called_once_with(credentials=None, host=client.DEFAULT_ENDPOINT) # Check mTLS is not triggered if api_endpoint is provided but # client_cert_source is None. options = client_options.ClientOptions(api_endpoint="squid.clam.whelk") with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient(client_options=options) grpc_transport.assert_called_once_with( api_mtls_endpoint=None, client_cert_source=None, credentials=None, host="squid.clam.whelk", ) # Check mTLS is triggered if client_cert_source is provided. options = client_options.ClientOptions( client_cert_source=client_cert_source_callback) with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient(client_options=options) grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_MTLS_ENDPOINT, client_cert_source=client_cert_source_callback, credentials=None, host=client.DEFAULT_ENDPOINT, ) # Check mTLS is triggered if api_endpoint and client_cert_source are provided. options = client_options.ClientOptions( api_endpoint="squid.clam.whelk", client_cert_source=client_cert_source_callback) with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient(client_options=options) grpc_transport.assert_called_once_with( api_mtls_endpoint="squid.clam.whelk", client_cert_source=client_cert_source_callback, credentials=None, host="squid.clam.whelk", )
def test_transport_instance(): # A client may be instantiated with a custom transport instance. transport = transports.LookupServiceGrpcTransport( credentials=credentials.AnonymousCredentials()) client = LookupServiceClient(transport=transport) assert client._transport is transport
def test_lookup_service_client_client_options(): # Check that if channel is provided we won't create a new one. with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.LookupServiceClient.get_transport_class" ) as gtc: transport = transports.LookupServiceGrpcTransport( credentials=credentials.AnonymousCredentials()) client = LookupServiceClient(transport=transport) gtc.assert_not_called() # Check that if channel is provided via str we will create a new one. with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.LookupServiceClient.get_transport_class" ) as gtc: client = LookupServiceClient(transport="grpc") gtc.assert_called() # Check the case api_endpoint is provided. options = client_options.ClientOptions(api_endpoint="squid.clam.whelk") with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient(client_options=options) grpc_transport.assert_called_once_with( api_mtls_endpoint="squid.clam.whelk", client_cert_source=None, credentials=None, host="squid.clam.whelk", ) # Check the case api_endpoint is not provided and GOOGLE_API_USE_MTLS is # "never". os.environ["GOOGLE_API_USE_MTLS"] = "never" with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient() grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_ENDPOINT, client_cert_source=None, credentials=None, host=client.DEFAULT_ENDPOINT, ) # Check the case api_endpoint is not provided and GOOGLE_API_USE_MTLS is # "always". os.environ["GOOGLE_API_USE_MTLS"] = "always" with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient() grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_MTLS_ENDPOINT, client_cert_source=None, credentials=None, host=client.DEFAULT_MTLS_ENDPOINT, ) # Check the case api_endpoint is not provided, GOOGLE_API_USE_MTLS is # "auto", and client_cert_source is provided. os.environ["GOOGLE_API_USE_MTLS"] = "auto" options = client_options.ClientOptions( client_cert_source=client_cert_source_callback) with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: grpc_transport.return_value = None client = LookupServiceClient(client_options=options) grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_MTLS_ENDPOINT, client_cert_source=client_cert_source_callback, credentials=None, host=client.DEFAULT_MTLS_ENDPOINT, ) # Check the case api_endpoint is not provided, GOOGLE_API_USE_MTLS is # "auto", and default_client_cert_source is provided. os.environ["GOOGLE_API_USE_MTLS"] = "auto" with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: with mock.patch( "google.auth.transport.mtls.has_default_client_cert_source", return_value=True, ): grpc_transport.return_value = None client = LookupServiceClient() grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_MTLS_ENDPOINT, client_cert_source=None, credentials=None, host=client.DEFAULT_MTLS_ENDPOINT, ) # Check the case api_endpoint is not provided, GOOGLE_API_USE_MTLS is # "auto", but client_cert_source and default_client_cert_source are None. os.environ["GOOGLE_API_USE_MTLS"] = "auto" with mock.patch( "google.cloud.servicedirectory_v1beta1.services.lookup_service.transports.LookupServiceGrpcTransport.__init__" ) as grpc_transport: with mock.patch( "google.auth.transport.mtls.has_default_client_cert_source", return_value=False, ): grpc_transport.return_value = None client = LookupServiceClient() grpc_transport.assert_called_once_with( api_mtls_endpoint=client.DEFAULT_ENDPOINT, client_cert_source=None, credentials=None, host=client.DEFAULT_ENDPOINT, ) # Check the case api_endpoint is not provided and GOOGLE_API_USE_MTLS has # unsupported value. os.environ["GOOGLE_API_USE_MTLS"] = "Unsupported" with pytest.raises(MutualTLSChannelError): client = LookupServiceClient() del os.environ["GOOGLE_API_USE_MTLS"]