Examples with CancelTasksResponse org.elasticsearch.action.admin.cluster.node.tasks.cancel.CancelTasksResponse used on opensource projects

Example 6 with CancelTasksResponse

use of org.elasticsearch.action.admin.cluster.node.tasks.cancel.CancelTasksResponse in project elasticsearch by elastic.

the class TasksIT method testTasksCancellation.

public void testTasksCancellation() throws Exception {
    // Start blocking test task
    // Get real client (the plugin is not registered on transport nodes)
    ListenableActionFuture<TestTaskPlugin.NodesResponse> future = TestTaskPlugin.TestTaskAction.INSTANCE.newRequestBuilder(client()).execute();
    logger.info("--> started test tasks");
    // Wait for the task to start on all nodes
    assertBusy(() -> assertEquals(internalCluster().size(), client().admin().cluster().prepareListTasks().setActions(TestTaskPlugin.TestTaskAction.NAME + "[n]").get().getTasks().size()));
    logger.info("--> cancelling the main test task");
    CancelTasksResponse cancelTasksResponse = client().admin().cluster().prepareCancelTasks().setActions(TestTaskPlugin.TestTaskAction.NAME).get();
    assertEquals(1, cancelTasksResponse.getTasks().size());
    future.get();
    logger.info("--> checking that test tasks are not running");
    assertEquals(0, client().admin().cluster().prepareListTasks().setActions(TestTaskPlugin.TestTaskAction.NAME + "*").get().getTasks().size());
}