From bb6fb6e08341d7c2156327ed80354ca827f19a1a Mon Sep 17 00:00:00 2001
From: Shay Banon <kimchy@gmail.com>
Date: Sat, 28 Jan 2012 00:26:53 +0200
Subject: [PATCH] improve test to wait for 2 nodes

---
 .../test/integration/gateway/local/QuorumLocalGatewayTests.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test/java/org/elasticsearch/test/integration/gateway/local/QuorumLocalGatewayTests.java b/src/test/java/org/elasticsearch/test/integration/gateway/local/QuorumLocalGatewayTests.java
index ca568feb289..89c1fe314f5 100644
--- a/src/test/java/org/elasticsearch/test/integration/gateway/local/QuorumLocalGatewayTests.java
+++ b/src/test/java/org/elasticsearch/test/integration/gateway/local/QuorumLocalGatewayTests.java
@@ -144,7 +144,7 @@ public class QuorumLocalGatewayTests extends AbstractNodesTests {
         closeNode("node1");
 
         logger.info("--> running cluster_health (wait for the shards to startup)");
-        clusterHealth = client("node2").admin().cluster().health(clusterHealthRequest().waitForYellowStatus().waitForActiveShards(4)).actionGet();
+        clusterHealth = client("node2").admin().cluster().health(clusterHealthRequest().waitForYellowStatus().waitForNodes("2").waitForActiveShards(4)).actionGet();
         logger.info("--> done cluster_health, status " + clusterHealth.status());
         assertThat(clusterHealth.timedOut(), equalTo(false));
         assertThat(clusterHealth.status(), equalTo(ClusterHealthStatus.YELLOW));