package redis.clients.jedis.tests.commands;

import java.util.List;

import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.Test;

import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.tests.HostAndPortUtil;
import redis.clients.jedis.tests.JedisTestBase;

public class ClusterCommandsTest extends JedisTestBase {
    private static Jedis node1;
    private static Jedis node2;

    private HostAndPort nodeInfo1 = HostAndPortUtil.getClusterServers().get(0);
    private HostAndPort nodeInfo2 = HostAndPortUtil.getClusterServers().get(1);

    @Before
    public void setUp() throws Exception {

	node1 = new Jedis(nodeInfo1.getHost(), nodeInfo1.getPort());
	node1.connect();
	node1.flushAll();

	node2 = new Jedis(nodeInfo2.getHost(), nodeInfo2.getPort());
	node2.connect();
	node2.flushAll();
    }

    @After
    public void tearDown() {
	node1.disconnect();
	node2.disconnect();
    }

    @AfterClass
    public static void removeSlots() throws InterruptedException {
	// This is to wait for gossip to replicate data.
	waitForEqualClusterSize();
	String[] nodes = node1.clusterNodes().split("\n");
	String node1Id = nodes[0].split(" ")[0];
	node1.clusterDelSlots(1, 2, 3, 4, 5, 500);
	node1.clusterSetSlotNode(5000, node1Id);
	node1.clusterDelSlots(5000, 10000);
	node2.clusterDelSlots(6000, 1, 2, 3, 4, 5, 500, 5000);
	node1.clusterAddSlots(6000);
	node1.clusterDelSlots(6000);
    }

    private static void waitForEqualClusterSize() throws InterruptedException {
	boolean notEqualSize = true;
	while (notEqualSize) {
	    notEqualSize = getClusterAttribute(node1.clusterInfo(),
		    "cluster_known_nodes") == getClusterAttribute(
		    node2.clusterInfo(), "cluster_size") ? false : true;
	}
    }

    private static int getClusterAttribute(String clusterInfo,
	    String attributeName) {
	for (String infoElement : clusterInfo.split("\n")) {
	    if (infoElement.contains(attributeName)) {
		return Integer.valueOf(infoElement.split(":")[1].trim());
	    }
	}
	return 0;
    }
    
    @Test
    public void clusterSetSlotImporting() {
	node2.clusterAddSlots(6000);
	String[] nodes = node1.clusterNodes().split("\n");
	String nodeId = nodes[0].split(" ")[0];
	String status = node1.clusterSetSlotImporting(6000, nodeId);
	assertEquals("OK", status);
    }

    @Test
    public void clusterNodes() {
	String nodes = node1.clusterNodes();
	assertTrue(nodes.split("\n").length > 0);
    }

    @Test
    public void clusterMeet() {
	String status = node1.clusterMeet("127.0.0.1", nodeInfo2.getPort());
	assertEquals("OK", status);
    }

    @Test
    public void clusterAddSlots() {
	String status = node1.clusterAddSlots(1, 2, 3, 4, 5);
	assertEquals("OK", status);
    }

    @Test
    public void clusterDelSlots() {
	node1.clusterAddSlots(900);
	String status = node1.clusterDelSlots(900);
	assertEquals("OK", status);
    }

    @Test
    public void clusterInfo() {
	String info = node1.clusterInfo();
	assertNotNull(info);
    }

    @Test
    public void clusterGetKeysInSlot() {
	node1.clusterAddSlots(500);
	List<String> keys = node1.clusterGetKeysInSlot(500, 1);
	assertEquals(0, keys.size());
    }

    @Test
    public void clusterSetSlotNode() {
	String[] nodes = node1.clusterNodes().split("\n");
	String nodeId = nodes[0].split(" ")[0];
	String status = node1.clusterSetSlotNode(10000, nodeId);
	assertEquals("OK", status);
    }

    @Test
    public void clusterSetSlotMigrating() {
	node1.clusterAddSlots(5000);
	String[] nodes = node1.clusterNodes().split("\n");
	String nodeId = nodes[0].split(" ")[0];
	String status = node1.clusterSetSlotMigrating(5000, nodeId);
	assertEquals("OK", status);
    }

}