mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 08:46:56 +08:00
83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
package dict
|
|
|
|
import (
|
|
"github.com/hdt3213/godis/lib/utils"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func TestSimpleDict_Keys(t *testing.T) {
|
|
d := MakeSimple()
|
|
size := 10
|
|
var expectKeys []string
|
|
for i := 0; i < size; i++ {
|
|
str := utils.RandString(5)
|
|
d.Put(str, str)
|
|
expectKeys = append(expectKeys, str)
|
|
}
|
|
sort.Slice(expectKeys, func(i, j int) bool {
|
|
return expectKeys[i] > expectKeys[j]
|
|
})
|
|
keys := d.Keys()
|
|
if len(keys) != size {
|
|
t.Errorf("expect %d keys, actual: %d", size, len(d.Keys()))
|
|
}
|
|
sort.Slice(keys, func(i, j int) bool {
|
|
return keys[i] > keys[j]
|
|
})
|
|
for i, k := range keys {
|
|
if k != expectKeys[i] {
|
|
t.Errorf("expect %s actual %s", expectKeys[i], k)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSimpleDict_PutIfExists(t *testing.T) {
|
|
d := MakeSimple()
|
|
key := utils.RandString(5)
|
|
val := key + "1"
|
|
ret := d.PutIfExists(key, val)
|
|
if ret != 0 {
|
|
t.Error("expect 0")
|
|
return
|
|
}
|
|
d.Put(key, val)
|
|
val = key + "2"
|
|
ret = d.PutIfExists(key, val)
|
|
if ret != 1 {
|
|
t.Error("expect 1")
|
|
return
|
|
}
|
|
if v, _ := d.Get(key); v != val {
|
|
t.Error("wrong value")
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestSimpleDict_Scan(t *testing.T) {
|
|
d := MakeSimple()
|
|
size := 10
|
|
for i := 0; i < size; i++ {
|
|
str := "a" + utils.RandString(5)
|
|
d.Put(str, []byte(str))
|
|
}
|
|
keys, nextCursor := d.DictScan(0, size, "*")
|
|
if len(keys) != size*2 {
|
|
t.Errorf("expect %d keys, actual: %d", size*2, len(keys))
|
|
return
|
|
}
|
|
if nextCursor != 0 {
|
|
t.Errorf("expect 0, actual: %d", nextCursor)
|
|
return
|
|
}
|
|
for i := 0; i < size; i++ {
|
|
str := "b" + utils.RandString(5)
|
|
d.Put(str, str)
|
|
}
|
|
keys, _ = d.DictScan(0, size*2, "a*")
|
|
if len(keys) != size*2 {
|
|
t.Errorf("expect %d keys, actual: %d", size*2, len(keys))
|
|
return
|
|
}
|
|
}
|