Change Moved cluster test and add confirmation for jedis cluster tests

src/main/java/redis/clients/jedis/exceptions/JedisAskDataException.java

@@ -0,0 +1,17 @@
+package redis.clients.jedis.exceptions;
+
+public class JedisAskDataException extends JedisDataException {
+    private static final long serialVersionUID = 3878126572474819403L;
+
+    public JedisAskDataException(String message) {
+        super(message);
+    }
+
+    public JedisAskDataException(Throwable cause) {
+        super(cause);
+    }
+
+    public JedisAskDataException(String message, Throwable cause) {
+        super(message, cause);
+    }
+}

src/test/java/redis/clients/jedis/tests/JedisClusterTest.java

@@ -1,7 +1,5 @@
 package redis.clients.jedis.tests;
 
-import java.util.HashSet;
-
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
@@ -11,6 +9,7 @@ import redis.clients.jedis.HostAndPort;
 import redis.clients.jedis.Jedis;
 import redis.clients.jedis.JedisCluster;
 import redis.clients.jedis.Pipeline;
+import redis.clients.jedis.exceptions.JedisAskDataException;
 import redis.clients.jedis.exceptions.JedisMovedDataException;
 import redis.clients.jedis.tests.utils.RedisSlot;
 
@@ -24,55 +23,61 @@ public class JedisClusterTest extends Assert {
     private HostAndPort nodeInfo3 = HostAndPortUtil.getClusterServers().get(2);
 
     @Before
-    public void setUp() {
+    public void setUp() throws InterruptedException {
-	node1 = new Jedis(nodeInfo1.getHost(), nodeInfo1.getPort());
+		node1 = new Jedis(nodeInfo1.getHost(), nodeInfo1.getPort());
-	node1.connect();
+		node1.connect();
-	node1.flushAll();
+		node1.flushAll();
 	
-	node2 = new Jedis(nodeInfo2.getHost(), nodeInfo2.getPort());
+		node2 = new Jedis(nodeInfo2.getHost(), nodeInfo2.getPort());
-	node2.connect();
+		node2.connect();
-	node2.flushAll();
+		node2.flushAll();
 	
-	node3 = new Jedis(nodeInfo3.getHost(), nodeInfo3.getPort());
+		node3 = new Jedis(nodeInfo3.getHost(), nodeInfo3.getPort());
-	node3.connect();
+		node3.connect();
-	node3.flushAll();
+		node3.flushAll();
 	
-	// ---- configure cluster
+		// ---- configure cluster
 	
-	// add nodes to cluster
+		// add nodes to cluster
-	node1.clusterMeet("127.0.0.1", nodeInfo1.getPort());
+		node1.clusterMeet("127.0.0.1", nodeInfo1.getPort());
-	node1.clusterMeet("127.0.0.1", nodeInfo2.getPort());
+		node1.clusterMeet("127.0.0.1", nodeInfo2.getPort());
-	node1.clusterMeet("127.0.0.1", nodeInfo3.getPort());
+		node1.clusterMeet("127.0.0.1", nodeInfo3.getPort());
 	
-	// add all slots to node1
+		// add all slots to node1
-	Pipeline pipelined = node1.pipelined();
+		Pipeline pipelined = node1.pipelined();
-	for (int i = 0; i < JedisCluster.HASHSLOTS; i++) {
+		for (int i = 0; i < JedisCluster.HASHSLOTS; i++) {
-	    pipelined.clusterAddSlots(i);
+		    pipelined.clusterAddSlots(i);
-	}
+		}
-	pipelined.sync();
+		pipelined.sync();
+		
+		boolean clusterOk = false;
+		while (!clusterOk) {
+			if (node1.clusterInfo().split("\n")[0].contains("ok")) {
+				clusterOk = true;
+			}
+			Thread.sleep(100);
+		}
     }
 
     @After
     public void tearDown() {
-	// clear all slots of node1
+		// clear all slots of node1
-	Pipeline pipelined = node1.pipelined();
+		Pipeline pipelined = node1.pipelined();
-	for (int i = 0; i < JedisCluster.HASHSLOTS; i++) {
+		for (int i = 0; i < JedisCluster.HASHSLOTS; i++) {
-	    pipelined.clusterDelSlots(i);
+		    pipelined.clusterDelSlots(i);
-	}
+		}
-	pipelined.sync();
+		pipelined.sync();
     }
 
     @Test(expected=JedisMovedDataException.class)
     public void throwMovedExceptionTest() {
-    	JedisCluster jc = new JedisCluster(new HashSet<HostAndPort>(HostAndPortUtil.getClusterServers()));
+    	node1.set("foo", "bar");
-    	jc.set("foo", "bar");
+    	node2.get("foo");
-    	jc.get("foo");
     }
 
-//    @Test
+//    @Test(expected=JedisAskDataException.class)
 //    public void ask() {
-//    	JedisCluster jc = new JedisCluster(new HashSet<HostAndPort>(HostAndPortUtil.getClusterServers()));
+//    	node1.set("foo", "bar");
-//    	jc.set("foo", "bar");
 //    	int keySlot = RedisSlot.getSlot("foo");
 //    	String node2Id = getNodeId(node2.clusterNodes());
 //    	node1.clusterSetSlotMigrating(keySlot, node2Id);