forked from mirror/ledisdb
145 lines
3.0 KiB
Go
145 lines
3.0 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.internal/re/ledisdb/config"
|
|
"github.com/siddontang/goredis"
|
|
)
|
|
|
|
func TestDumpRestore(t *testing.T) {
|
|
c := getTestConn()
|
|
defer c.Close()
|
|
|
|
var err error
|
|
_, err = c.Do("set", "mtest_a", "1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = c.Do("rpush", "mtest_la", "1", "2", "3")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = c.Do("hmset", "mtest_ha", "a", "1", "b", "2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = c.Do("sadd", "mtest_sa", "1", "2", "3")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = c.Do("zadd", "mtest_za", 1, "a", 2, "b", 3, "c")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
testDumpRestore(c, "dump", "mtest_a", t)
|
|
testDumpRestore(c, "ldump", "mtest_la", t)
|
|
testDumpRestore(c, "hdump", "mtest_ha", t)
|
|
testDumpRestore(c, "sdump", "mtest_sa", t)
|
|
testDumpRestore(c, "zdump", "mtest_za", t)
|
|
}
|
|
|
|
func testDumpRestore(c *goredis.PoolConn, dump string, key string, t *testing.T) {
|
|
if data, err := goredis.Bytes(c.Do(dump, key)); err != nil {
|
|
t.Fatal(err)
|
|
} else if _, err := c.Do("restore", key, 0, data); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
dataDir := "/tmp/test_migrate"
|
|
os.RemoveAll(dataDir)
|
|
|
|
s1Cfg := config.NewConfigDefault()
|
|
s1Cfg.DataDir = fmt.Sprintf("%s/s1", dataDir)
|
|
s1Cfg.Addr = "127.0.0.1:11185"
|
|
|
|
s2Cfg := config.NewConfigDefault()
|
|
s2Cfg.DataDir = fmt.Sprintf("%s/s2", dataDir)
|
|
s2Cfg.Addr = "127.0.0.1:11186"
|
|
|
|
s1, err := NewApp(s1Cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer s1.Close()
|
|
|
|
s2, err := NewApp(s2Cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer s2.Close()
|
|
|
|
go s1.Run()
|
|
|
|
go s2.Run()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
c1, _ := goredis.Connect(s1Cfg.Addr)
|
|
defer c1.Close()
|
|
|
|
c2, _ := goredis.Connect(s2Cfg.Addr)
|
|
defer c2.Close()
|
|
|
|
if _, err = c1.Do("set", "a", "1"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
timeout := 30000
|
|
if _, err = c1.Do("xmigrate", "127.0.0.1", 11186, "KV", "a", 0, timeout); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if s, err := goredis.String(c2.Do("get", "a")); err != nil {
|
|
t.Fatal(err)
|
|
} else if s != "1" {
|
|
t.Fatal(s, "must 1")
|
|
}
|
|
|
|
if s, err := goredis.String(c1.Do("get", "a")); err != nil && err != goredis.ErrNil {
|
|
t.Fatal(err)
|
|
} else if s != "" {
|
|
t.Fatal(s, "must empty")
|
|
}
|
|
|
|
if num, err := goredis.Int(c2.Do("xmigratedb", "127.0.0.1", 11185, "KV", 10, 0, timeout)); err != nil {
|
|
t.Fatal(err)
|
|
} else if num != 1 {
|
|
t.Fatal(num, "must number 1")
|
|
}
|
|
|
|
if s, err := goredis.String(c1.Do("get", "a")); err != nil {
|
|
t.Fatal(err)
|
|
} else if s != "1" {
|
|
t.Fatal(s, "must 1")
|
|
}
|
|
|
|
if s, err := goredis.String(c2.Do("get", "a")); err != nil && err != goredis.ErrNil {
|
|
t.Fatal(err)
|
|
} else if s != "" {
|
|
t.Fatal(s, "must empty")
|
|
}
|
|
|
|
if _, err = c1.Do("xmigrate", "127.0.0.1", 11186, "ALL", "a", 0, timeout); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if s, err := goredis.String(c2.Do("get", "a")); err != nil {
|
|
t.Fatal(err)
|
|
} else if s != "1" {
|
|
t.Fatal(s, "must 1")
|
|
}
|
|
|
|
if s, err := goredis.String(c1.Do("get", "a")); err != nil && err != goredis.ErrNil {
|
|
t.Fatal(err)
|
|
} else if s != "" {
|
|
t.Fatal(s, "must empty")
|
|
}
|
|
}
|