org.elasticsearch.test.transport

Class MockTransportService

java.lang.Object

org.elasticsearch.common.component.AbstractComponent

org.elasticsearch.common.component.AbstractLifecycleComponent

org.elasticsearch.transport.TransportService

org.elasticsearch.test.transport.MockTransportService

All Implemented Interfaces:

java.io.Closeable, java.lang.AutoCloseable, org.elasticsearch.common.component.LifecycleComponent, org.elasticsearch.common.lease.Releasable

public final class MockTransportService
extends org.elasticsearch.transport.TransportService

A mock transport service that allows to simulate different network topology failures. Internally it maps TransportAddress objects to rules that inject failures. Adding rules for a node is done by adding rules for all bound addresses of a node (and the publish address, if different). Matching requests to rules is based on the transport address associated with the discovery node of the request, namely by DiscoveryNode.getAddress(). This address is usually the publish address of the node but can also be a different one (for example, @see org.elasticsearch.discovery.zen.ping.unicast.UnicastZenPing, which constructs fake DiscoveryNode instances where the publish address is one of the bound addresses).

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MockTransportService.ClearableTransport The delegate transport instances defined in this class mock various kinds of disruption types.
static class	MockTransportService.DelegateTransport A pure delegate transport.
static class	MockTransportService.TestPlugin
static class	MockTransportService.Tracer

Nested classes/interfaces inherited from class org.elasticsearch.transport.TransportService

org.elasticsearch.transport.TransportService.Adapter, org.elasticsearch.transport.TransportService.ContextRestoreResponseHandler<T extends org.elasticsearch.transport.TransportResponse>, org.elasticsearch.transport.TransportService.HandshakeResponse

Field Summary

Fields inherited from class org.elasticsearch.transport.TransportService

clusterName, DIRECT_RESPONSE_PROFILE, HANDSHAKE_ACTION_NAME, NOOP_TRANSPORT_INTERCEPTOR, taskManager, threadPool, TRACE_LOG_EXCLUDE_SETTING, TRACE_LOG_INCLUDE_SETTING, transport

Fields inherited from class org.elasticsearch.common.component.AbstractLifecycleComponent

lifecycle

Fields inherited from class org.elasticsearch.common.component.AbstractComponent

deprecationLogger, logger, settings

Constructor Summary

Constructors

Constructor and Description
MockTransportService(org.elasticsearch.common.settings.Settings settings, org.elasticsearch.transport.Transport transport, org.elasticsearch.threadpool.ThreadPool threadPool, org.elasticsearch.transport.TransportInterceptor interceptor, org.elasticsearch.common.settings.ClusterSettings clusterSettings) Build the service.
MockTransportService(org.elasticsearch.common.settings.Settings settings, org.elasticsearch.transport.Transport transport, org.elasticsearch.threadpool.ThreadPool threadPool, org.elasticsearch.transport.TransportInterceptor interceptor, java.util.function.Function<org.elasticsearch.common.transport.BoundTransportAddress,org.elasticsearch.cluster.node.DiscoveryNode> localNodeFactory, org.elasticsearch.common.settings.ClusterSettings clusterSettings) Build the service.

Method Summary

Modifier and Type	Method and Description
boolean	addDelegate(org.elasticsearch.common.transport.TransportAddress transportAddress, MockTransportService.DelegateTransport transport) Adds a new delegate transport that is used for communication with the given transport address.
boolean	addDelegate(org.elasticsearch.transport.TransportService transportService, MockTransportService.DelegateTransport transport) Adds a new delegate transport that is used for communication with the given transport service.
void	addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress) Adds a rule that will cause every send request to fail, and each new connect since the rule is added to fail as well.
void	addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress, java.util.Set<java.lang.String> blockedActions) Adds a rule that will cause matching operations to throw ConnectTransportExceptions
void	addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress, java.lang.String... blockedActions) Adds a rule that will cause matching operations to throw ConnectTransportExceptions
void	addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService) Adds a rule that will cause every send request to fail, and each new connect since the rule is added to fail as well.
void	addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService, java.util.Set<java.lang.String> blockedActions) Adds a rule that will cause matching operations to throw ConnectTransportExceptions
void	addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService, java.lang.String... blockedActions) Adds a rule that will cause matching operations to throw ConnectTransportExceptions
void	addTracer(MockTransportService.Tracer tracer)
void	addUnresponsiveRule(org.elasticsearch.common.transport.TransportAddress transportAddress) Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.
void	addUnresponsiveRule(org.elasticsearch.common.transport.TransportAddress transportAddress, org.elasticsearch.common.unit.TimeValue duration) Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.
void	addUnresponsiveRule(org.elasticsearch.transport.TransportService transportService) Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.
void	addUnresponsiveRule(org.elasticsearch.transport.TransportService transportService, org.elasticsearch.common.unit.TimeValue duration) Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.
void	clearAllRules() Clears all the registered rules.
void	clearRule(org.elasticsearch.common.transport.TransportAddress transportAddress) Clears the rule associated with the provided transport address.
void	clearRule(org.elasticsearch.transport.TransportService transportService) Clears the rule associated with the provided transport service.
void	clearTracers()
protected org.elasticsearch.transport.TransportService.Adapter	createAdapter()
static MockTransportService	createNewService(org.elasticsearch.common.settings.Settings settings, org.elasticsearch.transport.Transport transport, org.elasticsearch.Version version, org.elasticsearch.threadpool.ThreadPool threadPool, org.elasticsearch.common.settings.ClusterSettings clusterSettings)
static MockTransportService	createNewService(org.elasticsearch.common.settings.Settings settings, org.elasticsearch.Version version, org.elasticsearch.threadpool.ThreadPool threadPool, org.elasticsearch.common.settings.ClusterSettings clusterSettings)
protected org.elasticsearch.tasks.TaskManager	createTaskManager()
protected void	doClose()
static org.elasticsearch.common.transport.TransportAddress[]	extractTransportAddresses(org.elasticsearch.transport.TransportService transportService)
org.elasticsearch.cluster.node.DiscoveryNode	getLocalDiscoNode()
org.elasticsearch.transport.Transport	getOriginalTransport()
org.elasticsearch.transport.Transport.Connection	openConnection(org.elasticsearch.cluster.node.DiscoveryNode node, org.elasticsearch.transport.ConnectionProfile profile)
org.elasticsearch.transport.Transport	original() Returns the original Transport service wrapped by this mock transport service.
boolean	removeTracer(MockTransportService.Tracer tracer)

Methods inherited from class org.elasticsearch.transport.TransportService

acceptIncomingRequests, addConnectionListener, addressesFromString, boundAddress, connectToNode, connectToNode, disconnectFromNode, doStart, doStop, getConnection, getLocalAddresses, getLocalNode, getRemoteClusterService, getRequestHandler, getTaskManager, getThreadPool, handshake, handshake, info, nodeConnected, registerRequestHandler, registerRequestHandler, removeConnectionListener, sendChildRequest, sendChildRequest, sendChildRequest, sendRequest, sendRequest, sendRequest, stats, submitRequest, submitRequest

Methods inherited from class org.elasticsearch.common.component.AbstractLifecycleComponent

addLifecycleListener, close, lifecycleState, removeLifecycleListener, start, stop

Methods inherited from class org.elasticsearch.common.component.AbstractComponent

logDeprecatedSetting, logRemovedSetting, nodeName

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

MockTransportService

public MockTransportService(org.elasticsearch.common.settings.Settings settings,
                            org.elasticsearch.transport.Transport transport,
                            org.elasticsearch.threadpool.ThreadPool threadPool,
                            org.elasticsearch.transport.TransportInterceptor interceptor,
                            @Nullable
                            org.elasticsearch.common.settings.ClusterSettings clusterSettings)

Build the service.

Parameters:

clusterSettings - if non null the the TransportService will register with the ClusterSettings for settings updates for TransportService.TRACE_LOG_EXCLUDE_SETTING and TransportService.TRACE_LOG_INCLUDE_SETTING.

MockTransportService

public MockTransportService(org.elasticsearch.common.settings.Settings settings,
                            org.elasticsearch.transport.Transport transport,
                            org.elasticsearch.threadpool.ThreadPool threadPool,
                            org.elasticsearch.transport.TransportInterceptor interceptor,
                            java.util.function.Function<org.elasticsearch.common.transport.BoundTransportAddress,org.elasticsearch.cluster.node.DiscoveryNode> localNodeFactory,
                            @Nullable
                            org.elasticsearch.common.settings.ClusterSettings clusterSettings)

Build the service.

Parameters:

clusterSettings - if non null the the TransportService will register with the ClusterSettings for settings updates for TransportService.TRACE_LOG_EXCLUDE_SETTING and TransportService.TRACE_LOG_INCLUDE_SETTING.

Method Detail

createNewService

public static MockTransportService createNewService(org.elasticsearch.common.settings.Settings settings,
                                                    org.elasticsearch.Version version,
                                                    org.elasticsearch.threadpool.ThreadPool threadPool,
                                                    @Nullable
                                                    org.elasticsearch.common.settings.ClusterSettings clusterSettings)

createNewService

public static MockTransportService createNewService(org.elasticsearch.common.settings.Settings settings,
                                                    org.elasticsearch.transport.Transport transport,
                                                    org.elasticsearch.Version version,
                                                    org.elasticsearch.threadpool.ThreadPool threadPool,
                                                    @Nullable
                                                    org.elasticsearch.common.settings.ClusterSettings clusterSettings)

extractTransportAddresses

public static org.elasticsearch.common.transport.TransportAddress[] extractTransportAddresses(org.elasticsearch.transport.TransportService transportService)

createTaskManager

protected org.elasticsearch.tasks.TaskManager createTaskManager()

Overrides:

createTaskManager in class org.elasticsearch.transport.TransportService

clearAllRules

public void clearAllRules()

Clears all the registered rules.

clearRule

public void clearRule(org.elasticsearch.transport.TransportService transportService)

Clears the rule associated with the provided transport service.

clearRule

public void clearRule(org.elasticsearch.common.transport.TransportAddress transportAddress)

Clears the rule associated with the provided transport address.

original

public org.elasticsearch.transport.Transport original()

Returns the original Transport service wrapped by this mock transport service.

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService)

Adds a rule that will cause every send request to fail, and each new connect since the rule is added to fail as well.

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress)

Adds a rule that will cause every send request to fail, and each new connect since the rule is added to fail as well.

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService,
                                       java.lang.String... blockedActions)

Adds a rule that will cause matching operations to throw ConnectTransportExceptions

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress,
                                       java.lang.String... blockedActions)

Adds a rule that will cause matching operations to throw ConnectTransportExceptions

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.transport.TransportService transportService,
                                       java.util.Set<java.lang.String> blockedActions)

Adds a rule that will cause matching operations to throw ConnectTransportExceptions

addFailToSendNoConnectRule

public void addFailToSendNoConnectRule(org.elasticsearch.common.transport.TransportAddress transportAddress,
                                       java.util.Set<java.lang.String> blockedActions)

Adds a rule that will cause matching operations to throw ConnectTransportExceptions

addUnresponsiveRule

public void addUnresponsiveRule(org.elasticsearch.transport.TransportService transportService)

Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.

addUnresponsiveRule

public void addUnresponsiveRule(org.elasticsearch.common.transport.TransportAddress transportAddress)

Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.

addUnresponsiveRule

public void addUnresponsiveRule(org.elasticsearch.transport.TransportService transportService,
                                org.elasticsearch.common.unit.TimeValue duration)

Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.

Parameters:

duration - the amount of time to delay sending and connecting.

addUnresponsiveRule

public void addUnresponsiveRule(org.elasticsearch.common.transport.TransportAddress transportAddress,
                                org.elasticsearch.common.unit.TimeValue duration)

Adds a rule that will cause ignores each send request, simulating an unresponsive node and failing to connect once the rule was added.

Parameters:

duration - the amount of time to delay sending and connecting.

addDelegate

public boolean addDelegate(org.elasticsearch.transport.TransportService transportService,
                           MockTransportService.DelegateTransport transport)

Adds a new delegate transport that is used for communication with the given transport service.

Returns:

true iff no other delegate was registered for any of the addresses bound by transport service.

addDelegate

public boolean addDelegate(org.elasticsearch.common.transport.TransportAddress transportAddress,
                           MockTransportService.DelegateTransport transport)

Adds a new delegate transport that is used for communication with the given transport address.

Returns:

true iff no other delegate was registered for this address before.

addTracer

public void addTracer(MockTransportService.Tracer tracer)

removeTracer

public boolean removeTracer(MockTransportService.Tracer tracer)

clearTracers

public void clearTracers()

createAdapter

protected org.elasticsearch.transport.TransportService.Adapter createAdapter()

Overrides:

createAdapter in class org.elasticsearch.transport.TransportService

getOriginalTransport

public org.elasticsearch.transport.Transport getOriginalTransport()

openConnection

public org.elasticsearch.transport.Transport.Connection openConnection(org.elasticsearch.cluster.node.DiscoveryNode node,
                                                                       org.elasticsearch.transport.ConnectionProfile profile)
                                                                throws java.io.IOException

Overrides:

openConnection in class org.elasticsearch.transport.TransportService

Throws:

java.io.IOException

doClose

protected void doClose()
                throws java.io.IOException

Overrides:

doClose in class org.elasticsearch.transport.TransportService

Throws:

java.io.IOException

getLocalDiscoNode

public org.elasticsearch.cluster.node.DiscoveryNode getLocalDiscoNode()