91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/alicebob/miniredis/v2"
|
|
)
|
|
|
|
func TestListGetDeleteRedisKeys(t *testing.T) {
|
|
mr, err := miniredis.Run()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer mr.Close()
|
|
|
|
_ = mr.Set("str:a", "hello")
|
|
mr.HSet("hash:b", "f1", "v1")
|
|
mr.Push("list:c", "x", "y")
|
|
mr.SAdd("set:d", "m1", "m2")
|
|
_, _ = mr.ZAdd("zset:e", 1.5, "zm")
|
|
|
|
host, portStr, err := net.SplitHostPort(mr.Addr())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
port, _ := strconv.Atoi(portStr)
|
|
base := ConnReq{Driver: "redis", Host: host, Port: port, Database: "0"}
|
|
|
|
list := ListRedisKeys(context.Background(), base)
|
|
if list.Error != "" {
|
|
t.Fatal(list.Error)
|
|
}
|
|
if len(list.Keys) < 5 {
|
|
t.Fatalf("want >=5 keys, got %d %#v", len(list.Keys), list.Keys)
|
|
}
|
|
byName := map[string]RedisKeyInfo{}
|
|
for _, k := range list.Keys {
|
|
byName[k.Name] = k
|
|
}
|
|
if byName["str:a"].Type != "string" {
|
|
t.Fatalf("str type %s", byName["str:a"].Type)
|
|
}
|
|
if byName["hash:b"].Type != "hash" {
|
|
t.Fatalf("hash type %s", byName["hash:b"].Type)
|
|
}
|
|
// Real Redis returns -1 (no expire); miniredis may report 0.
|
|
if byName["str:a"].TTL < -1 {
|
|
t.Fatalf("unexpected ttl %d", byName["str:a"].TTL)
|
|
}
|
|
|
|
gv := GetRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "str:a"})
|
|
if gv.Error != "" || gv.RowCount != 1 || gv.Rows[0]["value"] != "hello" {
|
|
t.Fatalf("get string %#v", gv)
|
|
}
|
|
gh := GetRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "hash:b"})
|
|
if gh.Error != "" || gh.RowCount < 1 || gh.Columns[0] != "field" {
|
|
t.Fatalf("get hash %#v", gh)
|
|
}
|
|
if gh.Rows[0]["field"] != "f1" {
|
|
t.Fatalf("hash fields should be sorted, got %#v", gh.Rows)
|
|
}
|
|
gl := GetRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "list:c"})
|
|
if gl.Error != "" || gl.RowCount < 2 {
|
|
t.Fatalf("get list %#v", gl)
|
|
}
|
|
gs := GetRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "set:d"})
|
|
if gs.Error != "" || gs.RowCount < 2 {
|
|
t.Fatalf("get set %#v", gs)
|
|
}
|
|
gz := GetRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "zset:e"})
|
|
if gz.Error != "" || gz.RowCount != 1 {
|
|
t.Fatalf("get zset %#v", gz)
|
|
}
|
|
|
|
pat := ListRedisKeys(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Pattern: "str:*"})
|
|
if pat.Error != "" || len(pat.Keys) != 1 || pat.Keys[0].Name != "str:a" {
|
|
t.Fatalf("pattern %#v", pat)
|
|
}
|
|
|
|
del := DeleteRedisKey(context.Background(), ConnReq{Driver: "redis", Host: host, Port: port, Database: "0", Key: "str:a"})
|
|
if del.Error != "" {
|
|
t.Fatal(del.Error)
|
|
}
|
|
if mr.Exists("str:a") {
|
|
t.Fatal("key should be deleted")
|
|
}
|
|
}
|