[TEST] Verify expected number of nodes in cluster before issuing shar…

…dStores request
elastic · Jul 22, 2015 · ad59af4 · ad59af4
1 parent f0f5a1e
commit ad59af4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...est/java/org/elasticsearch/action/admin/indices/shards/IndicesShardStoreRequestTests.java b/...est/java/org/elasticsearch/action/admin/indices/shards/IndicesShardStoreRequestTests.java
@@ -45,6 +45,7 @@
 import java.util.concurrent.ExecutionException;
 
 import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
+import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertNoTimeout;
 import static org.hamcrest.Matchers.*;
 
 @ElasticsearchIntegrationTest.ClusterScope(scope = ElasticsearchIntegrationTest.Scope.TEST)
@@ -91,8 +92,9 @@ public void testBasic() throws Exception {
         logger.info("--> disable allocation");
         disableAllocation(index);
         logger.info("--> stop random node");
+        int num = client().admin().cluster().prepareState().get().getState().nodes().getSize();
         internalCluster().stopRandomNode(new IndexNodePredicate(index));
-        ensureYellow(index);
+        assertNoTimeout(client().admin().cluster().prepareHealth().setWaitForNodes("" + (num - 1)));
         ClusterState clusterState = client().admin().cluster().prepareState().get().getState();
         List<ShardRouting> unassignedShards = clusterState.routingTable().index(index).shardsWithState(ShardRoutingState.UNASSIGNED);
         response = client().admin().indices().shardStores(Requests.indicesShardStoresRequest(index)).get();