Added unit tests for ClusterIndexHealth and ClusterHealthResponse
Relates to #4528
This commit is contained in:
Boaz Leskes
parent
0bef2c66a9
commit
be27ed3a25
|
|
@ -317,7 +317,7 @@ public class IndexRoutingTable implements Iterable<IndexShardRoutingTable> {
|
||||||
public void validate() throws RoutingValidationException {
|
public void validate() throws RoutingValidationException {
|
||||||
}
|
}
|
||||||
|
|
||||||
public Builder builder(String index) {
|
public static Builder builder(String index) {
|
||||||
return new Builder(index);
|
return new Builder(index);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,192 @@
|
||||||
|
package org.elasticsearch.cluster;
|
||||||
|
/*
|
||||||
|
* Licensed to ElasticSearch under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. ElasticSearch licenses this
|
||||||
|
* file to you under the Apache License, Version 2.0 (the
|
||||||
|
* "License"); you may not use this file except in compliance
|
||||||
|
* with the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing,
|
||||||
|
* software distributed under the License is distributed on an
|
||||||
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
* KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations
|
||||||
|
* under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
import org.elasticsearch.ElasticSearchException;
|
||||||
|
import org.elasticsearch.action.admin.cluster.health.ClusterHealthResponse;
|
||||||
|
import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus;
|
||||||
|
import org.elasticsearch.action.admin.cluster.health.ClusterIndexHealth;
|
||||||
|
import org.elasticsearch.action.admin.cluster.health.ClusterShardHealth;
|
||||||
|
import org.elasticsearch.cluster.metadata.IndexMetaData;
|
||||||
|
import org.elasticsearch.cluster.metadata.MetaData;
|
||||||
|
import org.elasticsearch.cluster.routing.*;
|
||||||
|
import org.elasticsearch.index.shard.ShardId;
|
||||||
|
import org.elasticsearch.test.ElasticsearchTestCase;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import static org.hamcrest.CoreMatchers.equalTo;
|
||||||
|
|
||||||
|
public class ClusterHealthResponsesTests extends ElasticsearchTestCase {
|
||||||
|
|
||||||
|
|
||||||
|
private void assertIndexHealth(ClusterIndexHealth indexHealth, ShardCounter counter, IndexMetaData indexMetaData) {
|
||||||
|
assertThat(indexHealth.getStatus(), equalTo(counter.status()));
|
||||||
|
assertThat(indexHealth.getNumberOfShards(), equalTo(indexMetaData.getNumberOfShards()));
|
||||||
|
assertThat(indexHealth.getNumberOfReplicas(), equalTo(indexMetaData.getNumberOfReplicas()));
|
||||||
|
assertThat(indexHealth.getActiveShards(), equalTo(counter.active));
|
||||||
|
assertThat(indexHealth.getRelocatingShards(), equalTo(counter.relocating));
|
||||||
|
assertThat(indexHealth.getInitializingShards(), equalTo(counter.initializing));
|
||||||
|
assertThat(indexHealth.getUnassignedShards(), equalTo(counter.unassigned));
|
||||||
|
assertThat(indexHealth.getShards().size(), equalTo(indexMetaData.getNumberOfShards()));
|
||||||
|
int totalShards = 0;
|
||||||
|
for (ClusterShardHealth shardHealth : indexHealth.getShards().values()) {
|
||||||
|
totalShards += shardHealth.getActiveShards() + shardHealth.getInitializingShards() + shardHealth.getUnassignedShards();
|
||||||
|
}
|
||||||
|
|
||||||
|
assertThat(totalShards, equalTo(indexMetaData.getNumberOfShards() * (1 + indexMetaData.getNumberOfReplicas())));
|
||||||
|
}
|
||||||
|
|
||||||
|
protected class ShardCounter {
|
||||||
|
public int active;
|
||||||
|
public int relocating;
|
||||||
|
public int initializing;
|
||||||
|
public int unassigned;
|
||||||
|
public int primaryActive;
|
||||||
|
public int primaryInactive;
|
||||||
|
|
||||||
|
public ClusterHealthStatus status() {
|
||||||
|
if (primaryInactive > 0) {
|
||||||
|
return ClusterHealthStatus.RED;
|
||||||
|
}
|
||||||
|
if (unassigned > 0 || initializing > 0) {
|
||||||
|
return ClusterHealthStatus.YELLOW;
|
||||||
|
}
|
||||||
|
return ClusterHealthStatus.GREEN;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void update(ShardRouting shardRouting) {
|
||||||
|
if (shardRouting.active()) {
|
||||||
|
active++;
|
||||||
|
if (shardRouting.primary()) {
|
||||||
|
primaryActive++;
|
||||||
|
}
|
||||||
|
if (shardRouting.relocating()) {
|
||||||
|
relocating++;
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (shardRouting.primary()) {
|
||||||
|
primaryInactive++;
|
||||||
|
}
|
||||||
|
if (shardRouting.initializing()) {
|
||||||
|
initializing++;
|
||||||
|
} else {
|
||||||
|
unassigned++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static int node_id = 1;
|
||||||
|
|
||||||
|
private ImmutableShardRouting genShardRouting(String index, int shardId, boolean primary) {
|
||||||
|
|
||||||
|
ShardRoutingState state;
|
||||||
|
|
||||||
|
int i = randomInt(40);
|
||||||
|
if (i > 5) {
|
||||||
|
state = ShardRoutingState.STARTED;
|
||||||
|
} else if (i > 3) {
|
||||||
|
state = ShardRoutingState.RELOCATING;
|
||||||
|
} else if (i > 1) {
|
||||||
|
state = ShardRoutingState.INITIALIZING;
|
||||||
|
} else {
|
||||||
|
state = ShardRoutingState.UNASSIGNED;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (state) {
|
||||||
|
case UNASSIGNED:
|
||||||
|
return new MutableShardRouting(index, shardId, null, primary, ShardRoutingState.UNASSIGNED, 1);
|
||||||
|
case STARTED:
|
||||||
|
return new MutableShardRouting(index, shardId, "node_" + Integer.toString(node_id++), primary, ShardRoutingState.STARTED, 1);
|
||||||
|
case INITIALIZING:
|
||||||
|
return new MutableShardRouting(index, shardId, "node_" + Integer.toString(node_id++), primary, ShardRoutingState.INITIALIZING, 1);
|
||||||
|
case RELOCATING:
|
||||||
|
return new MutableShardRouting(index, shardId, "node_" + Integer.toString(node_id++), "node_" + Integer.toString(node_id++), primary, ShardRoutingState.RELOCATING, 1);
|
||||||
|
default:
|
||||||
|
throw new ElasticSearchException("Unknown state: " + state.name());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private IndexShardRoutingTable genShardRoutingTable(String index, int shardId, int replicas, ShardCounter counter) {
|
||||||
|
IndexShardRoutingTable.Builder builder = new IndexShardRoutingTable.Builder(new ShardId(index, shardId), true);
|
||||||
|
ImmutableShardRouting shardRouting = genShardRouting(index, shardId, true);
|
||||||
|
counter.update(shardRouting);
|
||||||
|
builder.addShard(shardRouting);
|
||||||
|
for (; replicas > 0; replicas--) {
|
||||||
|
shardRouting = genShardRouting(index, shardId, false);
|
||||||
|
counter.update(shardRouting);
|
||||||
|
builder.addShard(shardRouting);
|
||||||
|
}
|
||||||
|
|
||||||
|
return builder.build();
|
||||||
|
}
|
||||||
|
|
||||||
|
IndexRoutingTable genIndexRoutingTable(IndexMetaData indexMetaData, ShardCounter counter) {
|
||||||
|
IndexRoutingTable.Builder builder = IndexRoutingTable.builder(indexMetaData.index());
|
||||||
|
for (int shard = 0; shard < indexMetaData.numberOfShards(); shard++) {
|
||||||
|
builder.addIndexShard(genShardRoutingTable(indexMetaData.index(), shard, indexMetaData.getNumberOfReplicas(), counter));
|
||||||
|
}
|
||||||
|
return builder.build();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testClusterIndexHealth() {
|
||||||
|
int numberOfShards = randomInt(3) + 1;
|
||||||
|
int numberOfReplicas = randomInt(4);
|
||||||
|
IndexMetaData indexMetaData = IndexMetaData.builder("test1").numberOfShards(numberOfShards).numberOfReplicas(numberOfReplicas).build();
|
||||||
|
ShardCounter counter = new ShardCounter();
|
||||||
|
IndexRoutingTable indexRoutingTable = genIndexRoutingTable(indexMetaData, counter);
|
||||||
|
|
||||||
|
ClusterIndexHealth indexHealth = new ClusterIndexHealth(indexMetaData, indexRoutingTable);
|
||||||
|
logger.info("index status: {}, expected {}", indexHealth.getStatus(), counter.status());
|
||||||
|
assertIndexHealth(indexHealth, counter, indexMetaData);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void assertClusterHealth(ClusterHealthResponse clusterHealth, ShardCounter counter) {
|
||||||
|
assertThat(clusterHealth.getStatus(), equalTo(counter.status()));
|
||||||
|
assertThat(clusterHealth.getActiveShards(), equalTo(counter.active));
|
||||||
|
assertThat(clusterHealth.getActivePrimaryShards(), equalTo(counter.primaryActive));
|
||||||
|
assertThat(clusterHealth.getInitializingShards(), equalTo(counter.initializing));
|
||||||
|
assertThat(clusterHealth.getRelocatingShards(), equalTo(counter.relocating));
|
||||||
|
assertThat(clusterHealth.getUnassignedShards(), equalTo(counter.unassigned));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testClusterHealth() {
|
||||||
|
ShardCounter counter = new ShardCounter();
|
||||||
|
RoutingTable.Builder routingTable = RoutingTable.builder();
|
||||||
|
MetaData.Builder metaData = MetaData.builder();
|
||||||
|
for (int i = randomInt(4); i >= 0; i--) {
|
||||||
|
int numberOfShards = randomInt(3) + 1;
|
||||||
|
int numberOfReplicas = randomInt(4);
|
||||||
|
IndexMetaData indexMetaData = IndexMetaData.builder("test_" + Integer.toString(i)).numberOfShards(numberOfShards).numberOfReplicas(numberOfReplicas).build();
|
||||||
|
IndexRoutingTable indexRoutingTable = genIndexRoutingTable(indexMetaData, counter);
|
||||||
|
metaData.put(indexMetaData, true);
|
||||||
|
routingTable.add(indexRoutingTable);
|
||||||
|
}
|
||||||
|
ClusterState clusterState = ClusterState.builder().metaData(metaData).routingTable(routingTable).build();
|
||||||
|
ClusterHealthResponse clusterHealth = new ClusterHealthResponse("bla", clusterState.metaData().concreteIndices(null), clusterState);
|
||||||
|
logger.info("cluster status: {}, expected {}", clusterHealth.getStatus(), counter.status());
|
||||||
|
|
||||||
|
assertClusterHealth(clusterHealth, counter);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue