Rename test : Shar<>edJedisPipelineTest to ShardedJedisPipelineTest

2011-09-16 17:23:59 +02:00
parent fd0c57599c
commit dcba7bd3b3
1 changed files with 0 additions and 0 deletions
--- a/src/test/java/redis/clients/jedis/tests/ShardedJedisPipelineTest.java
+++ b/src/test/java/redis/clients/jedis/tests/ShardedJedisPipelineTest.java
@@ -0,0 +1,116 @@
+package redis.clients.jedis.tests;
+
+import org.junit.Before;
+import org.junit.Test;
+import redis.clients.jedis.*;
+import redis.clients.jedis.exceptions.JedisDataException;
+
+import java.io.UnsupportedEncodingException;
+import java.util.*;
+
+import static junit.framework.Assert.assertEquals;
+import static junit.framework.Assert.assertNull;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+
+public class SharedJedisPipelineTest {
+    private static HostAndPortUtil.HostAndPort redis1 = HostAndPortUtil.getRedisServers()
+            .get(0);
+    private static HostAndPortUtil.HostAndPort redis2 = HostAndPortUtil.getRedisServers()
+            .get(1);
+
+    private ShardedJedis jedis;
+
+    @Before
+    public void setUp() throws Exception {
+        Jedis jedis = new Jedis(redis1.host, redis1.port);
+        jedis.auth("foobared");
+        jedis.flushAll();
+        jedis.disconnect();
+        jedis = new Jedis(redis2.host, redis2.port);
+        jedis.auth("foobared");
+        jedis.flushAll();
+        jedis.disconnect();
+
+        List<JedisShardInfo> shards = new ArrayList<JedisShardInfo>();
+        JedisShardInfo si1 = new JedisShardInfo(redis1.host, redis1.port);
+        si1.setPassword("foobared");
+        shards.add(si1);
+        JedisShardInfo si2 = new JedisShardInfo(redis2.host, redis2.port);
+        si2.setPassword("foobared");
+        shards.add(si2);
+        this.jedis = new ShardedJedis(shards);
+    }
+
+
+    @Test
+    public void pipeline() throws UnsupportedEncodingException {
+        ShardedJedisPipeline p = jedis.pipelined();
+        p.set("foo", "bar");
+        p.get("foo");
+        List<Object> results = p.syncAndReturnAll();
+
+        assertEquals(2, results.size());
+        assertEquals("OK", results.get(0));
+        assertEquals("bar", results.get(1));
+
+    }
+
+    @Test
+    public void pipelineResponse() {
+        jedis.set("string", "foo");
+        jedis.lpush("list", "foo");
+        jedis.hset("hash", "foo", "bar");
+        jedis.zadd("zset", 1, "foo");
+        jedis.sadd("set", "foo");
+
+        ShardedJedisPipeline p = jedis.pipelined();
+        Response<String> string = p.get("string");
+        Response<String> list = p.lpop("list");
+        Response<String> hash = p.hget("hash", "foo");
+        Response<Set<String>> zset = p.zrange("zset", 0, -1);
+        Response<String> set = p.spop("set");
+        Response<Boolean> blist = p.exists("list");
+        Response<Double> zincrby = p.zincrby("zset", 1, "foo");
+        Response<Long> zcard = p.zcard("zset");
+        p.lpush("list", "bar");
+        Response<List<String>> lrange = p.lrange("list", 0, -1);
+        Response<Map<String, String>> hgetAll = p.hgetAll("hash");
+        p.sadd("set", "foo");
+        Response<Set<String>> smembers = p.smembers("set");
+        Response<Set<Tuple>> zrangeWithScores = p.zrangeWithScores("zset", 0,
+                -1);
+        p.sync();
+
+        assertEquals("foo", string.get());
+        assertEquals("foo", list.get());
+        assertEquals("bar", hash.get());
+        assertEquals("foo", zset.get().iterator().next());
+        assertEquals("foo", set.get());
+        assertFalse(blist.get());
+        assertEquals(new Double(2), zincrby.get());
+        assertEquals(new Long(1), zcard.get());
+        assertEquals(1, lrange.get().size());
+        assertNotNull(hgetAll.get().get("foo"));
+        assertEquals(1, smembers.get().size());
+        assertEquals(1, zrangeWithScores.get().size());
+    }
+
+    @Test(expected = JedisDataException.class)
+    public void pipelineResponseWithinPipeline() {
+        jedis.set("string", "foo");
+
+        ShardedJedisPipeline p = jedis.pipelined();
+        Response<String> string = p.get("string");
+        string.get();
+        p.sync();
+    }
+
+    @Test
+    public void canRetrieveUnsetKey() {
+        ShardedJedisPipeline p = jedis.pipelined();
+        Response<String> shouldNotExist = p.get(UUID.randomUUID().toString());
+        p.sync();
+        assertNull(shouldNotExist.get());
+    }
+}