mirror of https://github.com/go-redis/redis.git
1732 lines
44 KiB
Go
1732 lines
44 KiB
Go
package redis_test
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
. "launchpad.net/gocheck"
|
|
|
|
"github.com/vmihailenco/redis"
|
|
)
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
type RedisTest struct {
|
|
client, multiClient *redis.Client
|
|
}
|
|
|
|
var _ = Suite(&RedisTest{})
|
|
|
|
func Test(t *testing.T) { TestingT(t) }
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) SetUpTest(c *C) {
|
|
t.client = redis.NewTCPClient(":6379", "", -1)
|
|
_, err := t.client.Flushdb().Reply()
|
|
c.Check(err, IsNil)
|
|
|
|
t.multiClient = t.client.Multi()
|
|
}
|
|
|
|
func (t *RedisTest) TearDownTest(c *C) {
|
|
_, err := t.client.Flushdb().Reply()
|
|
c.Check(err, IsNil)
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestInitConn(c *C) {
|
|
openConn := func() (io.ReadWriter, error) {
|
|
return net.Dial("tcp", ":6379")
|
|
}
|
|
|
|
isInitConnCalled := false
|
|
initConn := func(client *redis.Client) error {
|
|
isInitConnCalled = true
|
|
return nil
|
|
}
|
|
|
|
client := redis.NewClient(openConn, nil, initConn)
|
|
pong, err := client.Ping().Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pong, Equals, "PONG")
|
|
c.Check(isInitConnCalled, Equals, true)
|
|
}
|
|
|
|
func (t *RedisTest) TestRunWithMissingReplyPart(c *C) {
|
|
req := t.client.Set("foo", "bar")
|
|
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
|
|
ok, err := req.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestAuth(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestEcho(c *C) {
|
|
echo, err := t.client.Echo("hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(echo, Equals, "hello")
|
|
}
|
|
|
|
func (t *RedisTest) TestPing(c *C) {
|
|
pong, err := t.client.Ping().Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pong, Equals, "PONG")
|
|
}
|
|
|
|
func (t *RedisTest) TestSelect(c *C) {
|
|
ok, err := t.client.Select(1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestDel(c *C) {
|
|
t.client.Set("key1", "Hello")
|
|
t.client.Set("key2", "World")
|
|
|
|
n, err := t.client.Del("key1", "key2", "key3").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestDump(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestExists(c *C) {
|
|
t.client.Set("key1", "Hello")
|
|
|
|
exists, err := t.client.Exists("key1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(exists, Equals, true)
|
|
|
|
exists, err = t.client.Exists("key2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(exists, Equals, false)
|
|
}
|
|
|
|
func (t *RedisTest) TestExpire(c *C) {
|
|
t.client.Set("key", "Hello").Reply()
|
|
|
|
isSet, err := t.client.Expire("key", 10).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
ttl, err := t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(10))
|
|
|
|
t.client.Set("key", "Hello World")
|
|
|
|
ttl, err = t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(-1))
|
|
}
|
|
|
|
func (t *RedisTest) TestExpireAt(c *C) {
|
|
t.client.Set("key", "Hello").Reply()
|
|
|
|
exists, err := t.client.Exists("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(exists, Equals, true)
|
|
|
|
isSet, err := t.client.ExpireAt("key", 1293840000).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
exists, err = t.client.Exists("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(exists, Equals, false)
|
|
}
|
|
|
|
func (t *RedisTest) TestKeys(c *C) {
|
|
t.client.MSet("one", "1", "two", "2", "three", "3", "four", "4").Reply()
|
|
|
|
keys, err := t.client.Keys("*o*").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(keys, DeepEquals, []interface{}{"four", "two", "one"})
|
|
|
|
keys, err = t.client.Keys("t??").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(keys, DeepEquals, []interface{}{"two"})
|
|
|
|
keys, err = t.client.Keys("*").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(keys, DeepEquals, []interface{}{"four", "three", "two", "one"})
|
|
}
|
|
|
|
func (t *RedisTest) TestMigrate(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestMove(c *C) {
|
|
isMoved, err := t.client.Move("foo", 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isMoved, Equals, false)
|
|
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
isMoved, err = t.client.Move("foo", 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isMoved, Equals, true)
|
|
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
|
|
t.client.Select(1)
|
|
|
|
v, err = t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestObject(c *C) {
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
n, err := t.client.ObjectRefCount("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
enc, err := t.client.ObjectEncoding("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(enc, Equals, "raw")
|
|
|
|
n, err = t.client.ObjectIdleTime("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
}
|
|
|
|
func (t *RedisTest) TestPersist(c *C) {
|
|
t.client.Set("key", "Hello").Reply()
|
|
t.client.Expire("key", 10).Reply()
|
|
|
|
ttl, err := t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(10))
|
|
|
|
isPersisted, err := t.client.Persist("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isPersisted, Equals, true)
|
|
|
|
ttl, err = t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(-1))
|
|
}
|
|
|
|
func (t *RedisTest) TestPExpire(c *C) {
|
|
c.Skip("not implemented")
|
|
|
|
t.client.Set("key", "Hello").Reply()
|
|
|
|
isSet, err := t.client.PExpire("key", 1500).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
ttl, err := t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, 1)
|
|
|
|
pttl, err := t.client.PTTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pttl, Equals, 1500)
|
|
}
|
|
|
|
func (t *RedisTest) TestPExpireAt(c *C) {
|
|
c.Skip("not implemented")
|
|
|
|
t.client.Set("key", "Hello").Reply()
|
|
|
|
isSet, err := t.client.PExpireAt("key", 1555555555005).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
ttl, err := t.client.TTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, 211915059)
|
|
|
|
pttl, err := t.client.PTTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pttl, Equals, int64(211915059461))
|
|
}
|
|
|
|
func (t *RedisTest) TestPTTL(c *C) {
|
|
c.Skip("not implemented")
|
|
|
|
t.client.Set("key", "Hello").Reply()
|
|
t.client.Expire("key", 1).Reply()
|
|
|
|
pttl, err := t.client.PTTL("key").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pttl, Equals, int64(999))
|
|
}
|
|
|
|
func (t *RedisTest) TestRandomKey(c *C) {
|
|
key, err := t.client.RandomKey().Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(key, Equals, "")
|
|
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
key, err = t.client.RandomKey().Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(key, Equals, "foo")
|
|
}
|
|
|
|
func (t *RedisTest) TestRename(c *C) {
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
status, err := t.client.Rename("foo", "foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(status, Equals, "OK")
|
|
|
|
v, err := t.client.Get("foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestRenameNX(c *C) {
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
renamed, err := t.client.RenameNX("foo", "foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(renamed, Equals, true)
|
|
|
|
v, err := t.client.Get("foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestRestore(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestSort(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestTTL(c *C) {
|
|
ttl, err := t.client.TTL("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(-1))
|
|
|
|
t.client.Set("foo", "bar").Reply()
|
|
t.client.Expire("foo", 60)
|
|
|
|
ttl, err = t.client.TTL("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(60))
|
|
}
|
|
|
|
func (t *RedisTest) Type(c *C) {
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
type_, err := t.client.Type("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(type_, Equals, "string")
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestAppend(c *C) {
|
|
l, err := t.client.Append("foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(l, Equals, int64(3))
|
|
}
|
|
|
|
func (t *RedisTest) TestDecr(c *C) {
|
|
n, err := t.client.Decr("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(-1))
|
|
}
|
|
|
|
func (t *RedisTest) TestDecrBy(c *C) {
|
|
n, err := t.client.DecrBy("foo", 10).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(-10))
|
|
}
|
|
|
|
func (t *RedisTest) TestGet(c *C) {
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
}
|
|
|
|
func (t *RedisTest) TestSetGetBig(c *C) {
|
|
v, err := t.client.GetBit("foo", 5).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, int64(0))
|
|
|
|
v, err = t.client.SetBit("foo", 5, 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, int64(0))
|
|
|
|
v, err = t.client.GetBit("foo", 5).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, int64(1))
|
|
}
|
|
|
|
func (t *RedisTest) TestGetRange(c *C) {
|
|
t.client.Set("foo", "hello")
|
|
|
|
v, err := t.client.GetRange("foo", 0, 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "he")
|
|
}
|
|
|
|
func (t *RedisTest) TestGetSet(c *C) {
|
|
t.client.Set("foo", "bar")
|
|
|
|
v, err := t.client.GetSet("foo", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
|
|
v, err = t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar2")
|
|
}
|
|
|
|
func (t *RedisTest) TestIncr(c *C) {
|
|
n, err := t.client.Incr("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
}
|
|
|
|
func (t *RedisTest) TestIncrBy(c *C) {
|
|
n, err := t.client.IncrBy("foo", 10).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(10))
|
|
}
|
|
|
|
func (t *RedisTest) TestMsetMget(c *C) {
|
|
ok, err := t.client.MSet("foo1", "bar1", "foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
values, err := t.client.MGet("foo1", "foo2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"bar1", "bar2"})
|
|
}
|
|
|
|
func (t *RedisTest) MSetNX(c *C) {
|
|
isSet, err := t.client.MSetNX("foo1", "bar1", "foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
isSet, err = t.client.MSetNX("foo1", "bar1", "foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, false)
|
|
}
|
|
|
|
func (t *RedisTest) PSetEx(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestSetGet(c *C) {
|
|
ok, err := t.client.Set("foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestSetEx(c *C) {
|
|
ok, err := t.client.SetEx("foo", 10, "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
ttl, err := t.client.TTL("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ttl, Equals, int64(10))
|
|
}
|
|
|
|
func (t *RedisTest) TestSetNx(c *C) {
|
|
isSet, err := t.client.SetNx("foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
isSet, err = t.client.SetNx("foo", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, false)
|
|
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestSetRange(c *C) {
|
|
t.client.Set("foo", "Hello World").Reply()
|
|
|
|
n, err := t.client.SetRange("foo", 6, "Redis").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(11))
|
|
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "Hello Redis")
|
|
}
|
|
|
|
func (t *RedisTest) TestStrLen(c *C) {
|
|
t.client.Set("foo", "bar").Reply()
|
|
|
|
n, err := t.client.StrLen("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(3))
|
|
|
|
n, err = t.client.StrLen("_").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestHDel(c *C) {
|
|
t.client.HSet("myhash", "foo", "bar").Reply()
|
|
|
|
n, err := t.client.HDel("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.HDel("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
}
|
|
|
|
func (t *RedisTest) TestHExists(c *C) {
|
|
t.client.HSet("myhash", "foo", "bar").Reply()
|
|
|
|
n, err := t.client.HExists("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, true)
|
|
|
|
n, err = t.client.HExists("myhash", "foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, false)
|
|
}
|
|
|
|
func (t *RedisTest) TestHGet(c *C) {
|
|
t.client.HSet("myhash", "foo", "bar").Reply()
|
|
|
|
v, err := t.client.HGet("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
|
|
v, err = t.client.HGet("myhash", "foo1").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
}
|
|
|
|
func (t *RedisTest) TestHGetAll(c *C) {
|
|
t.client.HSet("myhash", "foo1", "bar1").Reply()
|
|
t.client.HSet("myhash", "foo2", "bar2").Reply()
|
|
|
|
values, err := t.client.HGetAll("myhash").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"foo1", "bar1", "foo2", "bar2"})
|
|
}
|
|
|
|
func (t *RedisTest) TestHIncrBy(c *C) {
|
|
t.client.HSet("myhash", "foo", "5").Reply()
|
|
|
|
n, err := t.client.HIncrBy("myhash", "foo", 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(6))
|
|
|
|
n, err = t.client.HIncrBy("myhash", "foo", -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(5))
|
|
|
|
n, err = t.client.HIncrBy("myhash", "foo", -10).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(-5))
|
|
}
|
|
|
|
func (t *RedisTest) TestHIncrByFloat(c *C) {
|
|
c.Skip("not implemented")
|
|
}
|
|
|
|
func (t *RedisTest) TestHKeys(c *C) {
|
|
t.client.HSet("myhash", "foo1", "bar1").Reply()
|
|
t.client.HSet("myhash", "foo2", "bar2").Reply()
|
|
|
|
keys, err := t.client.HKeys("myhash").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(keys, DeepEquals, []interface{}{"foo1", "foo2"})
|
|
}
|
|
|
|
func (t *RedisTest) TestHLen(c *C) {
|
|
t.client.HSet("myhash", "foo1", "bar1").Reply()
|
|
t.client.HSet("myhash", "foo2", "bar2").Reply()
|
|
|
|
n, err := t.client.HLen("myhash").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestHMGet(c *C) {
|
|
t.client.HSet("myhash", "foo1", "bar1").Reply()
|
|
t.client.HSet("myhash", "foo2", "bar2").Reply()
|
|
|
|
values, err := t.client.HMGet("myhash", "foo1", "foo2", "_").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"bar1", "bar2", nil})
|
|
}
|
|
|
|
func (t *RedisTest) TestHMSet(c *C) {
|
|
ok, err := t.client.HMSet("myhash", "foo1", "bar1", "foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
v1, err := t.client.HGet("myhash", "foo1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v1, Equals, "bar1")
|
|
|
|
v2, err := t.client.HGet("myhash", "foo2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v2, Equals, "bar2")
|
|
}
|
|
|
|
func (t *RedisTest) TestHSet(c *C) {
|
|
isNew, err := t.client.HSet("myhash", "foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isNew, Equals, true)
|
|
|
|
v, err := t.client.HGet("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestHSetNX(c *C) {
|
|
isSet, err := t.client.HSetNX("myhash", "foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, true)
|
|
|
|
isSet, err = t.client.HSetNX("myhash", "foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isSet, Equals, false)
|
|
|
|
v, err := t.client.HGet("myhash", "foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestHVals(c *C) {
|
|
t.client.HSet("myhash", "foo1", "bar1").Reply()
|
|
t.client.HSet("myhash", "foo2", "bar2").Reply()
|
|
|
|
vals, err := t.client.HVals("myhash").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(vals, DeepEquals, []interface{}{"bar1", "bar2"})
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestBLPop(c *C) {
|
|
t.client.RPush("list1", "a", "b", "c").Reply()
|
|
|
|
values, err := t.client.BLPop(0, "list1", "list2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"list1", "a"})
|
|
}
|
|
|
|
func (t *RedisTest) TestBrPop(c *C) {
|
|
t.client.RPush("list1", "a", "b", "c").Reply()
|
|
|
|
values, err := t.client.BRPop(0, "list1", "list2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"list1", "c"})
|
|
}
|
|
|
|
func (t *RedisTest) TestBRPopLPush(c *C) {
|
|
t.client.RPush("list1", "a", "b", "c").Reply()
|
|
|
|
v, err := t.client.BRPopLPush("list1", "list2", 0).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "c")
|
|
}
|
|
|
|
func (t *RedisTest) TestLIndex(c *C) {
|
|
t.client.LPush("list", "World")
|
|
t.client.LPush("list", "Hello")
|
|
|
|
v, err := t.client.LIndex("list", 0).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "Hello")
|
|
|
|
v, err = t.client.LIndex("list", -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "World")
|
|
|
|
v, err = t.client.LIndex("list", 3).Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
}
|
|
|
|
func (t *RedisTest) TestLInsert(c *C) {
|
|
t.client.RPush("list", "Hello").Reply()
|
|
t.client.RPush("list", "World").Reply()
|
|
|
|
n, err := t.client.LInsert("list", "BEFORE", "World", "There").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(3))
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"Hello", "There", "World"})
|
|
}
|
|
|
|
func (t *RedisTest) TestLLen(c *C) {
|
|
t.client.LPush("list", "World").Reply()
|
|
t.client.LPush("list", "Hello").Reply()
|
|
|
|
n, err := t.client.LLen("list").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestLPop(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
v, err := t.client.LPop("list").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "one")
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "three"})
|
|
}
|
|
|
|
func (t *RedisTest) TestLPush(c *C) {
|
|
t.client.LPush("list", "World").Reply()
|
|
t.client.LPush("list", "Hello").Reply()
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"Hello", "World"})
|
|
}
|
|
|
|
func (t *RedisTest) TestLPushX(c *C) {
|
|
t.client.LPush("list", "World").Reply()
|
|
|
|
n, err := t.client.LPushX("list", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
n, err = t.client.LPushX("list2", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"Hello", "World"})
|
|
|
|
values, err = t.client.LRange("list2", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{})
|
|
}
|
|
|
|
func (t *RedisTest) TestLRange(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
values, err := t.client.LRange("list", 0, 0).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one"})
|
|
|
|
values, err = t.client.LRange("list", -3, 2).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two", "three"})
|
|
|
|
values, err = t.client.LRange("list", -100, 100).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two", "three"})
|
|
|
|
values, err = t.client.LRange("list", 5, 10).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{})
|
|
}
|
|
|
|
func (t *RedisTest) TestLRem(c *C) {
|
|
t.client.RPush("list", "hello").Reply()
|
|
t.client.RPush("list", "hello").Reply()
|
|
t.client.RPush("list", "foo").Reply()
|
|
t.client.RPush("list", "hello").Reply()
|
|
|
|
n, err := t.client.LRem("list", -2, "hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"hello", "foo"})
|
|
}
|
|
|
|
func (t *RedisTest) TestLSet(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
ok, err := t.client.LSet("list", 0, "four").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
ok, err = t.client.LSet("list", -2, "five").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"four", "five", "three"})
|
|
}
|
|
|
|
func (t *RedisTest) TestLTrim(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
ok, err := t.client.LTrim("list", 1, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "three"})
|
|
}
|
|
|
|
func (t *RedisTest) TestRPop(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
v, err := t.client.RPop("list").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "three")
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two"})
|
|
}
|
|
|
|
func (t *RedisTest) TestRPopLPush(c *C) {
|
|
t.client.RPush("list", "one").Reply()
|
|
t.client.RPush("list", "two").Reply()
|
|
t.client.RPush("list", "three").Reply()
|
|
|
|
v, err := t.client.RPopLPush("list", "list2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "three")
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two"})
|
|
|
|
values, err = t.client.LRange("list2", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three"})
|
|
}
|
|
|
|
func (t *RedisTest) TestRPush(c *C) {
|
|
n, err := t.client.RPush("list", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.RPush("list", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"Hello", "World"})
|
|
}
|
|
|
|
func (t *RedisTest) TestRPushX(c *C) {
|
|
n, err := t.client.RPush("list", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.RPushX("list", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
n, err = t.client.RPushX("list2", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
|
|
values, err := t.client.LRange("list", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"Hello", "World"})
|
|
|
|
values, err = t.client.LRange("list2", 0, -1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{})
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestSAdd(c *C) {
|
|
n, err := t.client.SAdd("set", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.SAdd("set", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.SAdd("set", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
|
|
members, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(members, DeepEquals, []interface{}{"World", "Hello"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSCard(c *C) {
|
|
n, err := t.client.SAdd("set", "Hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.SAdd("set", "World").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
card, err := t.client.SCard("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(card, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestSDiff(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
values, err := t.client.SDiff("set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"a", "b"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSDiffStore(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
n, err := t.client.SDiffStore("set", "set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"a", "b"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSInter(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
values, err := t.client.SInter("set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"c"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSInterStore(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
n, err := t.client.SInterStore("set", "set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"c"})
|
|
}
|
|
|
|
func (t *RedisTest) TestIsMember(c *C) {
|
|
t.client.SAdd("set", "one").Reply()
|
|
|
|
isMember, err := t.client.SIsMember("set", "one").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isMember, Equals, true)
|
|
|
|
isMember, err = t.client.SIsMember("set", "two").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isMember, Equals, false)
|
|
}
|
|
|
|
func (t *RedisTest) TestSMembers(c *C) {
|
|
t.client.SAdd("set", "Hello").Reply()
|
|
t.client.SAdd("set", "World").Reply()
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"World", "Hello"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSMove(c *C) {
|
|
t.client.SAdd("set1", "one").Reply()
|
|
t.client.SAdd("set1", "two").Reply()
|
|
|
|
t.client.SAdd("set2", "three").Reply()
|
|
|
|
isMoved, err := t.client.SMove("set1", "set2", "two").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(isMoved, Equals, true)
|
|
|
|
values, err := t.client.SMembers("set1").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one"})
|
|
|
|
values, err = t.client.SMembers("set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three", "two"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSPop(c *C) {
|
|
t.client.SAdd("set", "one").Reply()
|
|
t.client.SAdd("set", "two").Reply()
|
|
t.client.SAdd("set", "three").Reply()
|
|
|
|
v, err := t.client.SPop("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Not(Equals), "")
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, HasLen, 2)
|
|
}
|
|
|
|
func (t *RedisTest) TestSRandMember(c *C) {
|
|
t.client.SAdd("set", "one").Reply()
|
|
t.client.SAdd("set", "two").Reply()
|
|
t.client.SAdd("set", "three").Reply()
|
|
|
|
v, err := t.client.SRandMember("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Not(Equals), "")
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, HasLen, 3)
|
|
}
|
|
|
|
func (t *RedisTest) TestSRem(c *C) {
|
|
t.client.SAdd("set", "one").Reply()
|
|
t.client.SAdd("set", "two").Reply()
|
|
t.client.SAdd("set", "three").Reply()
|
|
|
|
n, err := t.client.SRem("set", "one").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.SRem("set", "four").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three", "two"})
|
|
}
|
|
|
|
func (t *RedisTest) TestSUnion(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
values, err := t.client.SUnion("set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, HasLen, 5)
|
|
}
|
|
|
|
func (t *RedisTest) TestSUnionStore(c *C) {
|
|
t.client.SAdd("set1", "a").Reply()
|
|
t.client.SAdd("set1", "b").Reply()
|
|
t.client.SAdd("set1", "c").Reply()
|
|
|
|
t.client.SAdd("set2", "c").Reply()
|
|
t.client.SAdd("set2", "d").Reply()
|
|
t.client.SAdd("set2", "e").Reply()
|
|
|
|
n, err := t.client.SUnionStore("set", "set1", "set2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(5))
|
|
|
|
values, err := t.client.SMembers("set").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, HasLen, 5)
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestZAdd(c *C) {
|
|
n, err := t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.ZAdd("zset", redis.NewZMember(1, "uno")).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.ZAdd("zset", redis.NewZMember(3, "two")).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(0))
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "1", "uno", "1", "two", "3"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZCard(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
|
|
n, err := t.client.ZCard("zset").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestZCount(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZCount("zset", "-inf", "+inf").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(3))
|
|
|
|
n, err = t.client.ZCount("zset", "(1", "3").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
}
|
|
|
|
func (t *RedisTest) TestZIncrBy(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZIncrBy("zset", 2, "one").Reply()
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "2", "one", "3"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZInterStore(c *C) {
|
|
t.client.ZAdd("zset1", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset1", redis.NewZMember(2, "two")).Reply()
|
|
|
|
t.client.ZAdd("zset2", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset2", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset3", redis.NewZMember(3, "two")).Reply()
|
|
|
|
n, err := t.client.ZInterStore(
|
|
"out",
|
|
2,
|
|
[]string{"zset1", "zset2"},
|
|
[]int64{2, 3},
|
|
"",
|
|
).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
values, err := t.client.ZRange("out", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "5", "two", "10"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRange(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two", "three"})
|
|
|
|
values, err = t.client.ZRange("zset", 2, 3, false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three"})
|
|
|
|
values, err = t.client.ZRange("zset", -2, -1, false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "three"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRangeByScore(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
values, err := t.client.ZRangeByScore("zset", "-inf", "+inf", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two", "three"})
|
|
|
|
values, err = t.client.ZRangeByScore("zset", "1", "2", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "two"})
|
|
|
|
values, err = t.client.ZRangeByScore("zset", "(1", "2", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two"})
|
|
|
|
values, err = t.client.ZRangeByScore("zset", "(1", "(2", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRank(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZRank("zset", "three").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
n, err = t.client.ZRank("zset", "four").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(n, Equals, int64(0))
|
|
}
|
|
|
|
func (t *RedisTest) TestZRem(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZRem("zset", "two").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "1", "three", "3"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRemRangeByRank(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZRemRangeByRank("zset", 0, 1).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three", "3"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRemRangeByScore(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZRemRangeByScore("zset", "-inf", "(2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
values, err := t.client.ZRange("zset", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "2", "three", "3"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRevRange(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
values, err := t.client.ZRevRange("zset", "0", "-1", false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three", "two", "one"})
|
|
|
|
values, err = t.client.ZRevRange("zset", "2", "3", false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one"})
|
|
|
|
values, err = t.client.ZRevRange("zset", "-2", "-1", false).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two", "one"})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRevRangeByScore(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
values, err := t.client.ZRevRangeByScore("zset", "+inf", "-inf", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"three", "two", "one"})
|
|
|
|
values, err = t.client.ZRevRangeByScore("zset", "2", "(1", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"two"})
|
|
|
|
values, err = t.client.ZRevRangeByScore("zset", "(2", "(1", false, nil).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{})
|
|
}
|
|
|
|
func (t *RedisTest) TestZRevRank(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZRevRank("zset", "one").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(2))
|
|
|
|
n, err = t.client.ZRevRank("zset", "four").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(n, Equals, int64(0))
|
|
}
|
|
|
|
func (t *RedisTest) TestZScore(c *C) {
|
|
t.client.ZAdd("zset", redis.NewZMember(1.001, "one")).Reply()
|
|
|
|
score, err := t.client.ZScore("zset", "one").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(score, Equals, float64(1.001))
|
|
}
|
|
|
|
func (t *RedisTest) TestZUnionStore(c *C) {
|
|
t.client.ZAdd("zset1", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset1", redis.NewZMember(2, "two")).Reply()
|
|
|
|
t.client.ZAdd("zset2", redis.NewZMember(1, "one")).Reply()
|
|
t.client.ZAdd("zset2", redis.NewZMember(2, "two")).Reply()
|
|
t.client.ZAdd("zset2", redis.NewZMember(3, "three")).Reply()
|
|
|
|
n, err := t.client.ZUnionStore(
|
|
"out",
|
|
2,
|
|
[]string{"zset1", "zset2"},
|
|
[]int64{2, 3},
|
|
"",
|
|
).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(3))
|
|
|
|
values, err := t.client.ZRange("out", 0, -1, true).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"one", "5", "three", "9", "two", "10"})
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestPubSub(c *C) {
|
|
pubsub, err := t.client.PubSubClient()
|
|
c.Check(err, IsNil)
|
|
|
|
ch, err := pubsub.Subscribe("mychannel")
|
|
c.Check(err, IsNil)
|
|
c.Check(ch, Not(Equals), nil)
|
|
|
|
ch, err = pubsub.Subscribe("mychannel2")
|
|
c.Check(err, IsNil)
|
|
c.Check(ch, Not(Equals), nil)
|
|
|
|
n, err := t.client.Publish("mychannel", "hello").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
n, err = t.client.Publish("mychannel2", "hello2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, int64(1))
|
|
|
|
err = pubsub.Unsubscribe("mychannel")
|
|
c.Check(err, IsNil)
|
|
|
|
err = pubsub.Unsubscribe("mychannel2")
|
|
c.Check(err, IsNil)
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "subscribe")
|
|
c.Check(msg.Channel, Equals, "mychannel")
|
|
c.Check(msg.Number, Equals, int64(1))
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "subscribe")
|
|
c.Check(msg.Channel, Equals, "mychannel2")
|
|
c.Check(msg.Number, Equals, int64(2))
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "message")
|
|
c.Check(msg.Channel, Equals, "mychannel")
|
|
c.Check(msg.Message, Equals, "hello")
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "message")
|
|
c.Check(msg.Channel, Equals, "mychannel2")
|
|
c.Check(msg.Message, Equals, "hello2")
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "unsubscribe")
|
|
c.Check(msg.Channel, Equals, "mychannel")
|
|
c.Check(msg.Number, Equals, int64(1))
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
|
|
select {
|
|
case msg := <-ch:
|
|
c.Check(msg.Err, Equals, nil)
|
|
c.Check(msg.Name, Equals, "unsubscribe")
|
|
c.Check(msg.Channel, Equals, "mychannel2")
|
|
c.Check(msg.Number, Equals, int64(0))
|
|
case <-time.After(time.Second):
|
|
c.Error("Channel is empty.")
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestPipelining(c *C) {
|
|
_, err := t.client.Set("foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
|
|
setReq := t.multiClient.Set("foo1", "bar1")
|
|
getReq := t.multiClient.Get("foo2")
|
|
|
|
reqs, err := t.multiClient.RunQueued()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 2)
|
|
|
|
ok, err := setReq.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
|
|
v, err := getReq.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar2")
|
|
}
|
|
|
|
func (t *RedisTest) TestRunQueuedOnEmptyQueue(c *C) {
|
|
reqs, err := t.client.RunQueued()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 0)
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestDiscard(c *C) {
|
|
multiC := t.client.Multi()
|
|
|
|
multiC.Set("foo1", "bar1")
|
|
multiC.Discard()
|
|
multiC.Set("foo2", "bar2")
|
|
|
|
reqs, err := multiC.Exec()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 1)
|
|
|
|
v, err := t.client.Get("foo1").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
|
|
v, err = t.client.Get("foo2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar2")
|
|
}
|
|
|
|
func (t *RedisTest) TestMultiExec(c *C) {
|
|
multiC := t.client.Multi()
|
|
|
|
setR := multiC.Set("foo", "bar")
|
|
getR := multiC.Get("foo")
|
|
|
|
reqs, err := multiC.Exec()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 2)
|
|
|
|
_, err = setR.Reply()
|
|
c.Check(err, IsNil)
|
|
|
|
v, err := getR.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
func (t *RedisTest) TestExecOnEmptyQueue(c *C) {
|
|
reqs, err := t.client.Exec()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 0)
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) TestEchoFromGoroutines(c *C) {
|
|
for i := int64(0); i < 1000; i++ {
|
|
go func() {
|
|
msg := "echo" + strconv.FormatInt(i, 10)
|
|
echo, err := t.client.Echo(msg).Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(echo, Equals, msg)
|
|
}()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) TestPipeliningFromGoroutines(c *C) {
|
|
multiClient := t.client.Multi()
|
|
|
|
for i := int64(0); i < 1000; i += 2 {
|
|
go func() {
|
|
msg1 := "echo" + strconv.FormatInt(i, 10)
|
|
msg2 := "echo" + strconv.FormatInt(i+1, 10)
|
|
|
|
echo1Req := multiClient.Echo(msg1)
|
|
echo2Req := multiClient.Echo(msg2)
|
|
|
|
reqs, err := multiClient.RunQueued()
|
|
c.Check(reqs, HasLen, 2)
|
|
c.Check(err, IsNil)
|
|
|
|
echo1, err := echo1Req.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(echo1, Equals, msg1)
|
|
|
|
echo2, err := echo2Req.Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(echo2, Equals, msg2)
|
|
}()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) TestIncrFromGoroutines(c *C) {
|
|
wg := &sync.WaitGroup{}
|
|
for i := int64(0); i < 20000; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
_, err := t.client.Incr("TestIncrFromGoroutinesKey").Reply()
|
|
c.Check(err, IsNil)
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
|
|
n, err := t.client.Get("TestIncrFromGoroutinesKey").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, "20000")
|
|
}
|
|
|
|
func (t *RedisTest) TestIncrPipeliningFromGoroutines(c *C) {
|
|
multiClient := t.client.Multi()
|
|
|
|
wg := &sync.WaitGroup{}
|
|
for i := int64(0); i < 20000; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
multiClient.Incr("TestIncrPipeliningFromGoroutinesKey")
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
|
|
reqs, err := multiClient.RunQueued()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 20000)
|
|
|
|
n, err := t.client.Get("TestIncrPipeliningFromGoroutinesKey").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, "20000")
|
|
}
|
|
|
|
func (t *RedisTest) TestIncrTransaction(c *C) {
|
|
multiClient := t.client.Multi()
|
|
|
|
wg := &sync.WaitGroup{}
|
|
for i := int64(0); i < 20000; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
multiClient.Incr("TestIncrTransactionKey")
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
|
|
reqs, err := multiClient.Exec()
|
|
c.Check(err, IsNil)
|
|
c.Check(reqs, HasLen, 20000)
|
|
|
|
n, err := t.client.Get("TestIncrTransactionKey").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(n, Equals, "20000")
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
func (t *RedisTest) BenchmarkRedisPing(c *C) {
|
|
c.StopTimer()
|
|
|
|
for i := 0; i < 10; i++ {
|
|
pong, err := t.client.Ping().Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(pong, Equals, "PONG")
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.Ping().Reply()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) BenchmarkRedisSet(c *C) {
|
|
c.StopTimer()
|
|
|
|
for i := 0; i < 10; i++ {
|
|
ok, err := t.client.Set("foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(ok, Equals, "OK")
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.Set("foo", "bar").Reply()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) BenchmarkRedisGetNil(c *C) {
|
|
c.StopTimer()
|
|
|
|
for i := 0; i < 10; i++ {
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, Equals, redis.Nil)
|
|
c.Check(v, Equals, "")
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.Get("foo").Reply()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) BenchmarkRedisGet(c *C) {
|
|
c.StopTimer()
|
|
|
|
_, err := t.client.Set("foo", "bar").Reply()
|
|
c.Check(err, IsNil)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
v, err := t.client.Get("foo").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(v, Equals, "bar")
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.Get("foo").Reply()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) BenchmarkRedisMGet(c *C) {
|
|
c.StopTimer()
|
|
|
|
_, err := t.client.MSet("foo1", "bar1", "foo2", "bar2").Reply()
|
|
c.Check(err, IsNil)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
values, err := t.client.MGet("foo1", "foo2").Reply()
|
|
c.Check(err, IsNil)
|
|
c.Check(values, DeepEquals, []interface{}{"bar1", "bar2"})
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.MGet("foo1", "foo2").Reply()
|
|
}
|
|
}
|
|
|
|
func (t *RedisTest) BenchmarkRedisWriteRead(c *C) {
|
|
c.StopTimer()
|
|
|
|
req := []byte("PING\r\n")
|
|
conn, _, err := t.client.ConnPool.Get()
|
|
c.Check(err, IsNil)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
err := t.client.WriteRead(req, conn)
|
|
c.Check(err, IsNil)
|
|
c.Check(conn.Rd.Bytes(), DeepEquals, []byte("+PONG\r\n"))
|
|
}
|
|
|
|
c.StartTimer()
|
|
|
|
for i := 0; i < c.N; i++ {
|
|
t.client.WriteRead(req, conn)
|
|
}
|
|
|
|
c.StopTimer()
|
|
t.client.ConnPool.Add(conn)
|
|
c.StartTimer()
|
|
}
|