Class MockTaskManager

java.lang.Object

org.opensearch.tasks.TaskManager

org.opensearch.test.tasks.MockTaskManager

All Implemented Interfaces:

org.opensearch.cluster.ClusterStateApplier

public class MockTaskManager
extends org.opensearch.tasks.TaskManager

A mock task manager that allows adding listeners for events

Nested Class Summary

Nested classes/interfaces inherited from class org.opensearch.tasks.TaskManager

org.opensearch.tasks.TaskManager.TaskEventListeners

Field Summary

Fields

Modifier and Type	Field	Description
static final org.opensearch.common.settings.Setting<Boolean>	USE_MOCK_TASK_MANAGER_SETTING

Fields inherited from class org.opensearch.tasks.TaskManager

TASK_RESOURCE_CONSUMERS_ATTRIBUTES, TASK_RESOURCE_CONSUMERS_ENABLED

Constructor Summary

Constructors

Constructor	Description
MockTaskManager(org.opensearch.common.settings.Settings settings, org.opensearch.threadpool.ThreadPool threadPool, Set<String> taskHeaders)

Method Summary

Modifier and Type	Method	Description
void	addListener(MockTaskManagerListener listener)
org.opensearch.tasks.Task	register(String type, String action, org.opensearch.tasks.TaskAwareRequest request)
void	removeListener(MockTaskManagerListener listener)
org.opensearch.common.util.concurrent.ThreadContext.StoredContext	taskExecutionStarted(org.opensearch.tasks.Task task)
org.opensearch.tasks.Task	unregister(org.opensearch.tasks.Task task)
void	waitForTaskCompletion(org.opensearch.tasks.Task task, long untilInNanos)

Methods inherited from class org.opensearch.tasks.TaskManager

addTaskEventListeners, applyClusterState, cancel, cancelTaskAndDescendants, createTaskManagerWithClusterSettings, getBanCount, getBannedTaskIds, getCancellableTask, getCancellableTasks, getTask, getTasks, localNode, registerChildNode, registerTaskResourceConsumer, removeBan, setBan, setTaskCancellationService, setTaskResourceConsumersEnabled, setTaskResourceTrackingService, setTaskResultsService, startBanOnChildrenNodes, startBanOnChildrenNodes, startTrackingCancellableChannelTask, storeResult, storeResult

Methods inherited from class java.lang.Object

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

Field Details

USE_MOCK_TASK_MANAGER_SETTING

public static final org.opensearch.common.settings.Setting<Boolean> USE_MOCK_TASK_MANAGER_SETTING

Constructor Details

MockTaskManager

public MockTaskManager(org.opensearch.common.settings.Settings settings,
 org.opensearch.threadpool.ThreadPool threadPool,
 Set<String> taskHeaders)

Method Details

register

public org.opensearch.tasks.Task register(String type,
 String action,
 org.opensearch.tasks.TaskAwareRequest request)

Overrides:

register in class org.opensearch.tasks.TaskManager

unregister

public org.opensearch.tasks.Task unregister(org.opensearch.tasks.Task task)

Overrides:

unregister in class org.opensearch.tasks.TaskManager

waitForTaskCompletion

public void waitForTaskCompletion(org.opensearch.tasks.Task task,
 long untilInNanos)

Overrides:

waitForTaskCompletion in class org.opensearch.tasks.TaskManager

taskExecutionStarted

public org.opensearch.common.util.concurrent.ThreadContext.StoredContext taskExecutionStarted(org.opensearch.tasks.Task task)

Overrides:

taskExecutionStarted in class org.opensearch.tasks.TaskManager

addListener

public void addListener(MockTaskManagerListener listener)

removeListener

public void removeListener(MockTaskManagerListener listener)