forked from mirror/ledisdb
139 lines
2.8 KiB
Go
139 lines
2.8 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/siddontang/ledisdb/client/go/ledis"
|
|
"github.com/siddontang/ledisdb/config"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestScan(t *testing.T) {
|
|
cfg := new(config.Config)
|
|
cfg.DataDir = "/tmp/test_scan"
|
|
cfg.Addr = "127.0.0.1:11185"
|
|
|
|
os.RemoveAll(cfg.DataDir)
|
|
|
|
s, err := NewApp(cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
go s.Run()
|
|
defer s.Close()
|
|
|
|
cc := new(ledis.Config)
|
|
cc.Addr = cfg.Addr
|
|
cc.MaxIdleConns = 1
|
|
c := ledis.NewClient(cc)
|
|
defer c.Close()
|
|
|
|
testKVScan(t, c)
|
|
testHashScan(t, c)
|
|
testListScan(t, c)
|
|
testZSetScan(t, c)
|
|
testSetScan(t, c)
|
|
testBitScan(t, c)
|
|
|
|
}
|
|
|
|
func checkScanValues(t *testing.T, ay interface{}, values ...int) {
|
|
a, err := ledis.Strings(ay, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(a) != len(values) {
|
|
t.Fatal(fmt.Sprintf("len %d != %d", len(a), len(values)))
|
|
}
|
|
|
|
for i, v := range a {
|
|
if string(v) != fmt.Sprintf("%d", values[i]) {
|
|
t.Fatal(fmt.Sprintf("%d %s != %d", string(v), values[i]))
|
|
}
|
|
}
|
|
}
|
|
|
|
func checkScan(t *testing.T, c *ledis.Client, cmd string) {
|
|
if ay, err := ledis.Values(c.Do(cmd, "", "count", 5)); err != nil {
|
|
t.Fatal(err)
|
|
} else if len(ay) != 2 {
|
|
t.Fatal(len(ay))
|
|
} else if n := ay[0].([]byte); string(n) != "4" {
|
|
t.Fatal(string(n))
|
|
} else {
|
|
checkScanValues(t, ay[1], 0, 1, 2, 3, 4)
|
|
}
|
|
|
|
if ay, err := ledis.Values(c.Do(cmd, "4", "count", 6)); err != nil {
|
|
t.Fatal(err)
|
|
} else if len(ay) != 2 {
|
|
t.Fatal(len(ay))
|
|
} else if n := ay[0].([]byte); string(n) != "" {
|
|
t.Fatal(string(n))
|
|
} else {
|
|
checkScanValues(t, ay[1], 5, 6, 7, 8, 9)
|
|
}
|
|
|
|
}
|
|
|
|
func testKVScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("set", fmt.Sprintf("%d", i), []byte("value")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "scan")
|
|
}
|
|
|
|
func testHashScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("hset", fmt.Sprintf("%d", i), fmt.Sprintf("%d", i), []byte("value")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "hscan")
|
|
}
|
|
|
|
func testListScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("lpush", fmt.Sprintf("%d", i), fmt.Sprintf("%d", i)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "lscan")
|
|
}
|
|
|
|
func testZSetScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("zadd", fmt.Sprintf("%d", i), i, []byte("value")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "zscan")
|
|
}
|
|
|
|
func testSetScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("sadd", fmt.Sprintf("%d", i), fmt.Sprintf("%d", i)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "sscan")
|
|
}
|
|
|
|
func testBitScan(t *testing.T, c *ledis.Client) {
|
|
for i := 0; i < 10; i++ {
|
|
if _, err := c.Do("bsetbit", fmt.Sprintf("%d", i), 1024, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
checkScan(t, c, "bscan")
|
|
}
|