Class DisruptableMockTransport

java.lang.Object

org.opensearch.test.transport.StubbableTransport

org.opensearch.test.transport.MockTransport

org.opensearch.test.disruption.DisruptableMockTransport

All Implemented Interfaces:

Closeable, AutoCloseable, org.opensearch.common.lease.Releasable, org.opensearch.common.lifecycle.LifecycleComponent, org.opensearch.transport.Transport

public abstract class DisruptableMockTransport
extends MockTransport

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	DisruptableMockTransport.ConnectionStatus	Response type from getConnectionStatus(DiscoveryNode) indicating whether, and how, messages should be disrupted on this transport.

Nested classes/interfaces inherited from class org.opensearch.test.transport.StubbableTransport

StubbableTransport.OpenConnectionBehavior, StubbableTransport.RequestHandlingBehavior<Request extends org.opensearch.transport.TransportRequest>, StubbableTransport.SendRequestBehavior, StubbableTransport.WrappedConnection

Nested classes/interfaces inherited from interface org.opensearch.transport.Transport

org.opensearch.transport.Transport.Connection, org.opensearch.transport.Transport.RequestHandlers, org.opensearch.transport.Transport.ResponseContext<T extends org.opensearch.core.transport.TransportResponse>, org.opensearch.transport.Transport.ResponseHandlers

Constructor Summary

Constructors

Constructor	Description
DisruptableMockTransport(org.opensearch.cluster.node.DiscoveryNode localNode, org.apache.logging.log4j.Logger logger, DeterministicTaskQueue deterministicTaskQueue)

Method Summary

Modifier and Type	Method	Description
org.opensearch.transport.TransportService	createTransportService(org.opensearch.common.settings.Settings settings, org.opensearch.threadpool.ThreadPool threadPool, org.opensearch.transport.TransportInterceptor interceptor, Function<org.opensearch.core.common.transport.BoundTransportAddress, org.opensearch.cluster.node.DiscoveryNode> localNodeFactory, org.opensearch.common.settings.ClusterSettings clusterSettings, Set<String> taskHeaders, org.opensearch.telemetry.tracing.Tracer tracer)
protected abstract void	execute(Runnable runnable)
protected abstract DisruptableMockTransport.ConnectionStatus	getConnectionStatus(org.opensearch.cluster.node.DiscoveryNode destination)
protected Runnable	getDisconnectException(long requestId, String action, org.opensearch.cluster.node.DiscoveryNode destination)
protected abstract Optional<DisruptableMockTransport>	getDisruptableMockTransport(org.opensearch.core.common.transport.TransportAddress address)
org.opensearch.cluster.node.DiscoveryNode	getLocalNode()
protected String	getRequestDescription(long requestId, String action, org.opensearch.cluster.node.DiscoveryNode destination)
protected void	onBlackholedDuringSend(long requestId, String action, DisruptableMockTransport destinationTransport)
protected void	onConnectedDuringSend(long requestId, String action, org.opensearch.transport.TransportRequest request, DisruptableMockTransport destinationTransport)
protected void	onDisconnectedDuringSend(long requestId, String action, DisruptableMockTransport destinationTransport)
protected void	onSendRequest(long requestId, String action, org.opensearch.transport.TransportRequest request, DisruptableMockTransport destinationTransport)
void	openConnection(org.opensearch.cluster.node.DiscoveryNode node, org.opensearch.transport.ConnectionProfile profile, org.opensearch.core.action.ActionListener<org.opensearch.transport.Transport.Connection> listener)

Methods inherited from class org.opensearch.test.transport.MockTransport

createConnection, handleError, handleLocalError, handleRemoteError, handleResponse, onSendRequest, onSendRequest, setMessageListener, writeableRegistry

Methods inherited from class org.opensearch.test.transport.StubbableTransport

addLifecycleListener, addressesFromString, boundAddress, close, getDefaultSeedAddresses, getRequestHandlers, getResponseHandlers, getStats, lifecycleState, profileBoundAddresses, removeLifecycleListener, setDefaultConnectBehavior, start, stop

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.opensearch.transport.Transport

isSecure, registerRequestHandler, setSlowLogThreshold

Constructor Details

DisruptableMockTransport

public DisruptableMockTransport(org.opensearch.cluster.node.DiscoveryNode localNode,
 org.apache.logging.log4j.Logger logger,
 DeterministicTaskQueue deterministicTaskQueue)

Method Details

getConnectionStatus

protected abstract DisruptableMockTransport.ConnectionStatus getConnectionStatus(org.opensearch.cluster.node.DiscoveryNode destination)

getDisruptableMockTransport

protected abstract Optional<DisruptableMockTransport> getDisruptableMockTransport(org.opensearch.core.common.transport.TransportAddress address)

execute

protected abstract void execute(Runnable runnable)

getLocalNode

public org.opensearch.cluster.node.DiscoveryNode getLocalNode()

createTransportService

public org.opensearch.transport.TransportService createTransportService(org.opensearch.common.settings.Settings settings,
 org.opensearch.threadpool.ThreadPool threadPool,
 org.opensearch.transport.TransportInterceptor interceptor,
 Function<org.opensearch.core.common.transport.BoundTransportAddress, org.opensearch.cluster.node.DiscoveryNode> localNodeFactory,
 @Nullable
 org.opensearch.common.settings.ClusterSettings clusterSettings,
 Set<String> taskHeaders,
 org.opensearch.telemetry.tracing.Tracer tracer)

Overrides:

createTransportService in class MockTransport

openConnection

public void openConnection(org.opensearch.cluster.node.DiscoveryNode node,
 org.opensearch.transport.ConnectionProfile profile,
 org.opensearch.core.action.ActionListener<org.opensearch.transport.Transport.Connection> listener)

Specified by:

openConnection in interface org.opensearch.transport.Transport

Overrides:

openConnection in class StubbableTransport

onSendRequest

protected void onSendRequest(long requestId,
 String action,
 org.opensearch.transport.TransportRequest request,
 DisruptableMockTransport destinationTransport)

getDisconnectException

protected Runnable getDisconnectException(long requestId,
 String action,
 org.opensearch.cluster.node.DiscoveryNode destination)

getRequestDescription

protected String getRequestDescription(long requestId,
 String action,
 org.opensearch.cluster.node.DiscoveryNode destination)

onBlackholedDuringSend

protected void onBlackholedDuringSend(long requestId,
 String action,
 DisruptableMockTransport destinationTransport)

onDisconnectedDuringSend

protected void onDisconnectedDuringSend(long requestId,
 String action,
 DisruptableMockTransport destinationTransport)

onConnectedDuringSend

protected void onConnectedDuringSend(long requestId,
 String action,
 org.opensearch.transport.TransportRequest request,
 DisruptableMockTransport destinationTransport)