def test_completion_client_get_transport_class(): transport = CompletionClient.get_transport_class() available_transports = [ transports.CompletionGrpcTransport, ] assert transport in available_transports transport = CompletionClient.get_transport_class("grpc") assert transport == transports.CompletionGrpcTransport
def test_client_withDEFAULT_CLIENT_INFO(): client_info = gapic_v1.client_info.ClientInfo() with mock.patch.object( transports.CompletionTransport, "_prep_wrapped_messages" ) as prep: client = CompletionClient( credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info) with mock.patch.object( transports.CompletionTransport, "_prep_wrapped_messages" ) as prep: transport_class = CompletionClient.get_transport_class() transport = transport_class( credentials=ga_credentials.AnonymousCredentials(), client_info=client_info, ) prep.assert_called_once_with(client_info)
def test_completion_client_get_transport_class(): transport = CompletionClient.get_transport_class() assert transport == transports.CompletionGrpcTransport transport = CompletionClient.get_transport_class("grpc") assert transport == transports.CompletionGrpcTransport