mirror of https://github.com/ledisdb/ledisdb.git
fullfill test case for various data type
This commit is contained in:
silentsai
parent
68c63c416a
commit
778f8f4a2b
|
|
@ -9,67 +9,165 @@ import (
|
||||||
|
|
||||||
var m sync.Mutex
|
var m sync.Mutex
|
||||||
|
|
||||||
func TestKvExpire(t *testing.T) {
|
type adaptor struct {
|
||||||
db := getTestDB()
|
set func([]byte, []byte) (int64, error)
|
||||||
m.Lock()
|
del func([]byte) (int64, error)
|
||||||
defer m.Unlock()
|
exists func([]byte) (int64, error)
|
||||||
|
|
||||||
k := []byte("ttl_a")
|
expire func([]byte, int64) (int64, error)
|
||||||
ek := []byte("ttl_b")
|
expireAt func([]byte, int64) (int64, error)
|
||||||
db.Set(k, []byte("1"))
|
ttl func([]byte) (int64, error)
|
||||||
|
|
||||||
if ok, _ := db.Expire(k, 10); ok != 1 {
|
showIdent func() string
|
||||||
t.Fatal(ok)
|
|
||||||
}
|
|
||||||
|
|
||||||
// err - expire on an inexisting key
|
|
||||||
if ok, _ := db.Expire(ek, 10); ok != 0 {
|
|
||||||
t.Fatal(ok)
|
|
||||||
}
|
|
||||||
|
|
||||||
// err - duration is zero
|
|
||||||
if ok, err := db.Expire(k, 0); err == nil || ok != 0 {
|
|
||||||
t.Fatal(fmt.Sprintf("res = %d, err = %s", ok, err))
|
|
||||||
}
|
|
||||||
|
|
||||||
// err - duration is negative
|
|
||||||
if ok, err := db.Expire(k, -10); err == nil || ok != 0 {
|
|
||||||
t.Fatal(fmt.Sprintf("res = %d, err = %s", ok, err))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestKvExpireAt(t *testing.T) {
|
func kvAdaptor(db *DB) *adaptor {
|
||||||
db := getTestDB()
|
adp := new(adaptor)
|
||||||
m.Lock()
|
adp.showIdent = func() string {
|
||||||
defer m.Unlock()
|
return "kv-adptor"
|
||||||
|
|
||||||
k := []byte("ttl_a")
|
|
||||||
ek := []byte("ttl_b")
|
|
||||||
db.Set(k, []byte("1"))
|
|
||||||
|
|
||||||
now := time.Now().Unix()
|
|
||||||
|
|
||||||
if ok, _ := db.ExpireAt(k, now+5); ok != 1 {
|
|
||||||
t.Fatal(ok)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// err - expire on an inexisting key
|
adp.set = db.SetNX
|
||||||
if ok, _ := db.ExpireAt(ek, now+5); ok != 0 {
|
adp.exists = db.Exists
|
||||||
t.Fatal(ok)
|
adp.del = func(k []byte) (int64, error) {
|
||||||
|
return db.Del(k)
|
||||||
}
|
}
|
||||||
|
|
||||||
// err - expire with the current time
|
adp.expire = db.Expire
|
||||||
if ok, err := db.ExpireAt(k, now); err == nil || ok != 0 {
|
adp.expireAt = db.ExpireAt
|
||||||
t.Fatal(fmt.Sprintf("res = %d, err = %s", ok, err))
|
adp.ttl = db.TTL
|
||||||
}
|
|
||||||
|
|
||||||
// err - expire with the time before
|
return adp
|
||||||
if ok, err := db.ExpireAt(k, now-5); err == nil || ok != 0 {
|
|
||||||
t.Fatal(fmt.Sprintf("res = %d, err = %s", ok, err))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestKvTTL(t *testing.T) {
|
func listAdaptor(db *DB) *adaptor {
|
||||||
|
adp := new(adaptor)
|
||||||
|
adp.showIdent = func() string {
|
||||||
|
return "list-adptor"
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.set = func(k []byte, v []byte) (int64, error) {
|
||||||
|
eles := make([][]byte, 0)
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
e := []byte(String(v) + fmt.Sprintf("_%d", i))
|
||||||
|
eles = append(eles, e)
|
||||||
|
}
|
||||||
|
|
||||||
|
if n, err := db.LPush(k, eles...); err != nil {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return n, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.exists = func(k []byte) (int64, error) {
|
||||||
|
if llen, err := db.LLen(k); err != nil || llen <= 0 {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return 1, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.del = db.LClear
|
||||||
|
adp.expire = db.LExpire
|
||||||
|
adp.expireAt = db.LExpireAt
|
||||||
|
adp.ttl = db.LTTL
|
||||||
|
|
||||||
|
return adp
|
||||||
|
}
|
||||||
|
|
||||||
|
func hashAdaptor(db *DB) *adaptor {
|
||||||
|
adp := new(adaptor)
|
||||||
|
adp.showIdent = func() string {
|
||||||
|
return "hash-adptor"
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.set = func(k []byte, v []byte) (int64, error) {
|
||||||
|
datas := make([]FVPair, 0)
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
suffix := fmt.Sprintf("_%d", i)
|
||||||
|
pair := FVPair{
|
||||||
|
Field: []byte(String(k) + suffix),
|
||||||
|
Value: []byte(String(v) + suffix)}
|
||||||
|
|
||||||
|
datas = append(datas, pair)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := db.HMset(k, datas...); err != nil {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return int64(len(datas)), nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.exists = func(k []byte) (int64, error) {
|
||||||
|
if hlen, err := db.HLen(k); err != nil || hlen <= 0 {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return 1, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.del = db.HClear
|
||||||
|
adp.expire = db.HExpire
|
||||||
|
adp.expireAt = db.HExpireAt
|
||||||
|
adp.ttl = db.HTTL
|
||||||
|
|
||||||
|
return adp
|
||||||
|
}
|
||||||
|
|
||||||
|
func zsetAdaptor(db *DB) *adaptor {
|
||||||
|
adp := new(adaptor)
|
||||||
|
adp.showIdent = func() string {
|
||||||
|
return "zset-adptor"
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.set = func(k []byte, v []byte) (int64, error) {
|
||||||
|
datas := make([]ScorePair, 0)
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
memb := []byte(String(k) + fmt.Sprintf("_%d", i))
|
||||||
|
pair := ScorePair{
|
||||||
|
Score: int64(i),
|
||||||
|
Member: memb}
|
||||||
|
|
||||||
|
datas = append(datas, pair)
|
||||||
|
}
|
||||||
|
|
||||||
|
if n, err := db.ZAdd(k, datas...); err != nil {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return n, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.exists = func(k []byte) (int64, error) {
|
||||||
|
if cnt, err := db.ZCard(k); err != nil || cnt <= 0 {
|
||||||
|
return 0, err
|
||||||
|
} else {
|
||||||
|
return 1, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
adp.del = db.ZClear
|
||||||
|
adp.expire = db.ZExpire
|
||||||
|
adp.expireAt = db.ZExpireAt
|
||||||
|
adp.ttl = db.ZTTL
|
||||||
|
|
||||||
|
return adp
|
||||||
|
}
|
||||||
|
|
||||||
|
func allAdaptors(db *DB) []*adaptor {
|
||||||
|
adps := make([]*adaptor, 4)
|
||||||
|
adps[0] = kvAdaptor(db)
|
||||||
|
adps[1] = listAdaptor(db)
|
||||||
|
adps[2] = hashAdaptor(db)
|
||||||
|
adps[3] = zsetAdaptor(db)
|
||||||
|
return adps
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
func TestExpire(t *testing.T) {
|
||||||
db := getTestDB()
|
db := getTestDB()
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
|
|
@ -77,25 +175,101 @@ func TestKvTTL(t *testing.T) {
|
||||||
k := []byte("ttl_a")
|
k := []byte("ttl_a")
|
||||||
ek := []byte("ttl_b")
|
ek := []byte("ttl_b")
|
||||||
|
|
||||||
db.Set(k, []byte("1"))
|
dbEntrys := allAdaptors(db)
|
||||||
db.Expire(k, 2)
|
for _, entry := range dbEntrys {
|
||||||
|
ident := entry.showIdent()
|
||||||
|
|
||||||
if tRemain, _ := db.TTL(k); tRemain != 2 {
|
entry.set(k, []byte("1"))
|
||||||
t.Fatal(tRemain)
|
|
||||||
}
|
|
||||||
|
|
||||||
// err - check ttl on an inexisting key
|
if ok, _ := entry.expire(k, 10); ok != 1 {
|
||||||
if tRemain, _ := db.TTL(ek); tRemain != -1 {
|
t.Fatal(ident, ok)
|
||||||
t.Fatal(tRemain)
|
}
|
||||||
}
|
|
||||||
|
|
||||||
db.Del(k)
|
// err - expire on an inexisting key
|
||||||
if tRemain, _ := db.TTL(k); tRemain != -1 {
|
if ok, _ := entry.expire(ek, 10); ok != 0 {
|
||||||
t.Fatal(tRemain)
|
t.Fatal(ident, ok)
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - duration is zero
|
||||||
|
if ok, err := entry.expire(k, 0); err == nil || ok != 0 {
|
||||||
|
t.Fatal(ident, fmt.Sprintf("res = %d, err = %s", ok, err))
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - duration is negative
|
||||||
|
if ok, err := entry.expire(k, -10); err == nil || ok != 0 {
|
||||||
|
t.Fatal(ident, fmt.Sprintf("res = %d, err = %s", ok, err))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestKvExpCompose(t *testing.T) {
|
func TestExpireAt(t *testing.T) {
|
||||||
|
db := getTestDB()
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
k := []byte("ttl_a")
|
||||||
|
ek := []byte("ttl_b")
|
||||||
|
|
||||||
|
dbEntrys := allAdaptors(db)
|
||||||
|
for _, entry := range dbEntrys {
|
||||||
|
ident := entry.showIdent()
|
||||||
|
now := time.Now().Unix()
|
||||||
|
|
||||||
|
entry.set(k, []byte("1"))
|
||||||
|
|
||||||
|
if ok, _ := entry.expireAt(k, now+5); ok != 1 {
|
||||||
|
t.Fatal(ident, ok)
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - expire on an inexisting key
|
||||||
|
if ok, _ := entry.expireAt(ek, now+5); ok != 0 {
|
||||||
|
t.Fatal(ident, ok)
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - expire with the current time
|
||||||
|
if ok, err := entry.expireAt(k, now); err == nil || ok != 0 {
|
||||||
|
t.Fatal(ident, fmt.Sprintf("res = %d, err = %s", ok, err))
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - expire with the time before
|
||||||
|
if ok, err := entry.expireAt(k, now-5); err == nil || ok != 0 {
|
||||||
|
t.Fatal(ident, fmt.Sprintf("res = %d, err = %s", ok, err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTTL(t *testing.T) {
|
||||||
|
db := getTestDB()
|
||||||
|
m.Lock()
|
||||||
|
defer m.Unlock()
|
||||||
|
|
||||||
|
k := []byte("ttl_a")
|
||||||
|
ek := []byte("ttl_b")
|
||||||
|
|
||||||
|
dbEntrys := allAdaptors(db)
|
||||||
|
for _, entry := range dbEntrys {
|
||||||
|
ident := entry.showIdent()
|
||||||
|
|
||||||
|
entry.set(k, []byte("1"))
|
||||||
|
entry.expire(k, 2)
|
||||||
|
|
||||||
|
if tRemain, _ := entry.ttl(k); tRemain != 2 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// err - check ttl on an inexisting key
|
||||||
|
if tRemain, _ := entry.ttl(ek); tRemain != -1 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
|
||||||
|
entry.del(k)
|
||||||
|
if tRemain, _ := entry.ttl(k); tRemain != -1 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExpCompose(t *testing.T) {
|
||||||
db := getTestDB()
|
db := getTestDB()
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
|
|
@ -104,68 +278,84 @@ func TestKvExpCompose(t *testing.T) {
|
||||||
k1 := []byte("ttl_b")
|
k1 := []byte("ttl_b")
|
||||||
k2 := []byte("ttl_c")
|
k2 := []byte("ttl_c")
|
||||||
|
|
||||||
db.Set(k0, k0)
|
dbEntrys := allAdaptors(db)
|
||||||
db.Set(k1, k1)
|
|
||||||
db.Set(k2, k2)
|
|
||||||
|
|
||||||
db.Expire(k0, 5)
|
for _, entry := range dbEntrys {
|
||||||
db.Expire(k1, 2)
|
ident := entry.showIdent()
|
||||||
db.Expire(k2, 60)
|
|
||||||
|
|
||||||
if tRemain, _ := db.TTL(k0); tRemain != 5 {
|
entry.set(k0, k0)
|
||||||
t.Fatal(tRemain)
|
entry.set(k1, k1)
|
||||||
}
|
entry.set(k2, k2)
|
||||||
if tRemain, _ := db.TTL(k1); tRemain != 2 {
|
|
||||||
t.Fatal(tRemain)
|
entry.expire(k0, 5)
|
||||||
}
|
entry.expire(k1, 2)
|
||||||
if tRemain, _ := db.TTL(k2); tRemain != 60 {
|
entry.expire(k2, 60)
|
||||||
t.Fatal(tRemain)
|
|
||||||
|
if tRemain, _ := entry.ttl(k0); tRemain != 5 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
if tRemain, _ := entry.ttl(k1); tRemain != 2 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
if tRemain, _ := entry.ttl(k2); tRemain != 60 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// after 1 sec
|
// after 1 sec
|
||||||
time.Sleep(1 * time.Second)
|
time.Sleep(1 * time.Second)
|
||||||
if tRemain, _ := db.TTL(k0); tRemain != 4 {
|
|
||||||
t.Fatal(tRemain)
|
for _, entry := range dbEntrys {
|
||||||
}
|
ident := entry.showIdent()
|
||||||
if tRemain, _ := db.TTL(k1); tRemain != 1 {
|
|
||||||
t.Fatal(tRemain)
|
if tRemain, _ := entry.ttl(k0); tRemain != 4 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
if tRemain, _ := entry.ttl(k1); tRemain != 1 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// after 2 sec
|
// after 2 sec
|
||||||
time.Sleep(2 * time.Second)
|
time.Sleep(2 * time.Second)
|
||||||
if tRemain, _ := db.TTL(k1); tRemain != -1 {
|
|
||||||
t.Fatal(tRemain)
|
|
||||||
}
|
|
||||||
if v, _ := db.Get(k1); v != nil {
|
|
||||||
t.Fatal(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
if tRemain, _ := db.TTL(k0); tRemain != 2 {
|
for _, entry := range dbEntrys {
|
||||||
t.Fatal(tRemain)
|
ident := entry.showIdent()
|
||||||
}
|
|
||||||
if v, _ := db.Get(k0); v == nil {
|
|
||||||
t.Fatal(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// refresh the expiration of key
|
if tRemain, _ := entry.ttl(k1); tRemain != -1 {
|
||||||
if tRemain, _ := db.TTL(k2); !(0 < tRemain && tRemain < 60) {
|
t.Fatal(ident, tRemain)
|
||||||
t.Fatal(tRemain)
|
}
|
||||||
}
|
if exist, _ := entry.exists(k1); exist > 0 {
|
||||||
|
t.Fatal(ident, false)
|
||||||
|
}
|
||||||
|
|
||||||
if ok, _ := db.Expire(k2, 100); ok != 1 {
|
if tRemain, _ := entry.ttl(k0); tRemain != 2 {
|
||||||
t.Fatal(false)
|
t.Fatal(ident, tRemain)
|
||||||
}
|
}
|
||||||
|
if exist, _ := entry.exists(k0); exist <= 0 {
|
||||||
|
t.Fatal(ident, false)
|
||||||
|
}
|
||||||
|
|
||||||
if tRemain, _ := db.TTL(k2); tRemain != 100 {
|
// refresh the expiration of key
|
||||||
t.Fatal(tRemain)
|
if tRemain, _ := entry.ttl(k2); !(0 < tRemain && tRemain < 60) {
|
||||||
}
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
|
||||||
// expire an inexisting key
|
if ok, _ := entry.expire(k2, 100); ok != 1 {
|
||||||
if ok, _ := db.Expire(k1, 10); ok == 1 {
|
t.Fatal(ident, false)
|
||||||
t.Fatal(false)
|
}
|
||||||
}
|
|
||||||
if tRemain, _ := db.TTL(k1); tRemain != -1 {
|
if tRemain, _ := entry.ttl(k2); tRemain != 100 {
|
||||||
t.Fatal(tRemain)
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// expire an inexisting key
|
||||||
|
if ok, _ := entry.expire(k1, 10); ok == 1 {
|
||||||
|
t.Fatal(ident, false)
|
||||||
|
}
|
||||||
|
if tRemain, _ := entry.ttl(k1); tRemain != -1 {
|
||||||
|
t.Fatal(ident, tRemain)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue