Examples with StaleShard org.elasticsearch.cluster.routing.allocation.StaleShard used on opensource projects

Example 1 with StaleShard

use of org.elasticsearch.cluster.routing.allocation.StaleShard in project elasticsearch by elastic.

the class ShardFailedClusterStateTaskExecutorTests method testTriviallySuccessfulTasksBatchedWithFailingTasks.

public void testTriviallySuccessfulTasksBatchedWithFailingTasks() throws Exception {
    String reason = "test trivially successful tasks batched with failing tasks";
    ClusterState currentState = createClusterStateWithStartedShards(reason);
    List<ShardStateAction.ShardEntry> failingTasks = createExistingShards(currentState, reason);
    List<ShardStateAction.ShardEntry> nonExistentTasks = createNonExistentShards(currentState, reason);
    ShardStateAction.ShardFailedClusterStateTaskExecutor failingExecutor = new ShardStateAction.ShardFailedClusterStateTaskExecutor(allocationService, null, logger) {

        @Override
        ClusterState applyFailedShards(ClusterState currentState, List<FailedShard> failedShards, List<StaleShard> staleShards) {
            throw new RuntimeException("simulated applyFailedShards failure");
        }
    };
    List<ShardStateAction.ShardEntry> tasks = new ArrayList<>();
    tasks.addAll(failingTasks);
    tasks.addAll(nonExistentTasks);
    ClusterStateTaskExecutor.ClusterTasksResult<ShardStateAction.ShardEntry> result = failingExecutor.execute(currentState, tasks);
    Map<ShardStateAction.ShardEntry, ClusterStateTaskExecutor.TaskResult> taskResultMap = failingTasks.stream().collect(Collectors.toMap(Function.identity(), task -> ClusterStateTaskExecutor.TaskResult.failure(new RuntimeException("simulated applyFailedShards failure"))));
    taskResultMap.putAll(nonExistentTasks.stream().collect(Collectors.toMap(Function.identity(), task -> ClusterStateTaskExecutor.TaskResult.success())));
    assertTaskResults(taskResultMap, result, currentState, false);
}