[TEST] Fix SearchStatsTests to have all shards allocated

If randomization brings up a single shard per index in this test
we might run our searches on only one index which causes the assertions
to fail afterwards that's why we need to wait until everything is alloated.
This commit is contained in:
Simon Willnauer 2014-03-11 11:18:45 +01:00
parent 9624b215fb
commit bb83c823b6
1 changed files with 7 additions and 7 deletions

View File

@ -28,8 +28,6 @@ import org.elasticsearch.cluster.ClusterState;
import org.elasticsearch.cluster.routing.GroupShardsIterator; import org.elasticsearch.cluster.routing.GroupShardsIterator;
import org.elasticsearch.cluster.routing.ShardIterator; import org.elasticsearch.cluster.routing.ShardIterator;
import org.elasticsearch.cluster.routing.ShardRouting; import org.elasticsearch.cluster.routing.ShardRouting;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.unit.TimeValue; import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.search.stats.SearchStats.Stats; import org.elasticsearch.index.search.stats.SearchStats.Stats;
@ -47,10 +45,8 @@ import static org.hamcrest.Matchers.*;
public class SearchStatsTests extends ElasticsearchIntegrationTest { public class SearchStatsTests extends ElasticsearchIntegrationTest {
@Override @Override
public Settings indexSettings() { protected int numberOfReplicas() {
return ImmutableSettings.builder() return 0;
.put("index.number_of_replicas", 0)
.build();
} }
@Test @Test
@ -72,6 +68,10 @@ public class SearchStatsTests extends ElasticsearchIntegrationTest {
} }
} }
cluster().ensureAtMostNumNodes(numAssignedShards("test1", "test2")); cluster().ensureAtMostNumNodes(numAssignedShards("test1", "test2"));
assertThat(cluster().size(), greaterThanOrEqualTo(2));
assertThat(numAssignedShards("test1", "test2"), greaterThanOrEqualTo(2));
// THERE WILL BE AT LEAST 2 NODES HERE SO WE CAN WAIT FOR GREEN
ensureGreen();
for (int i = 0; i < 200; i++) { for (int i = 0; i < 200; i++) {
client().prepareSearch().setQuery(QueryBuilders.termQuery("field", "value")).setStats("group1", "group2").execute().actionGet(); client().prepareSearch().setQuery(QueryBuilders.termQuery("field", "value")).setStats("group1", "group2").execute().actionGet();
} }