tcc rename

database/cluster_helper_test.go

@@ -0,0 +1,58 @@
+package database
+
+import (
+	"fmt"
+	"github.com/hdt3213/godis/lib/utils"
+	"github.com/hdt3213/godis/redis/protocol"
+	"github.com/hdt3213/godis/redis/protocol/asserts"
+	"testing"
+)
+
+func TestExistIn(t *testing.T) {
+	testDB.Flush()
+	key := utils.RandString(10)
+	value := utils.RandString(10)
+	key2 := utils.RandString(10)
+	testDB.Exec(nil, utils.ToCmdLine("set", key, value))
+	result := testDB.Exec(nil, utils.ToCmdLine("ExistIn", key, key2))
+	asserts.AssertMultiBulkReply(t, result, []string{key})
+	key3 := utils.RandString(10)
+	result = testDB.Exec(nil, utils.ToCmdLine("ExistIn", key2, key3))
+	asserts.AssertMultiBulkReplySize(t, result, 0)
+}
+
+func TestDumpKeyAndRenameTo(t *testing.T) {
+	testDB.Flush()
+	key := utils.RandString(10)
+	value := utils.RandString(10)
+	newKey := key + utils.RandString(2)
+	testDB.Exec(nil, utils.ToCmdLine("set", key, value, "ex", "1000"))
+
+	result := testDB.Exec(nil, utils.ToCmdLine("DumpKey", key))
+	if protocol.IsErrorReply(result) {
+		t.Error("dump key error")
+		return
+	}
+	dumpResult := result.(*protocol.MultiBulkReply)
+	result = testDB.Exec(nil, utils.ToCmdLine("RenameTo", newKey,
+		string(dumpResult.Args[0]), string(dumpResult.Args[1])))
+	asserts.AssertNotError(t, result)
+	result = testDB.Exec(nil, utils.ToCmdLine("RenameFrom", key))
+	asserts.AssertNotError(t, result)
+
+	result = testDB.Exec(nil, utils.ToCmdLine("exists", key))
+	asserts.AssertIntReply(t, result, 0)
+	result = testDB.Exec(nil, utils.ToCmdLine("exists", newKey))
+	asserts.AssertIntReply(t, result, 1)
+	// check ttl
+	result = testDB.Exec(nil, utils.ToCmdLine("ttl", newKey))
+	intResult, ok := result.(*protocol.IntReply)
+	if !ok {
+		t.Error(fmt.Sprintf("expected int protocol, actually %s", result.ToBytes()))
+		return
+	}
+	if intResult.Code <= 0 {
+		t.Errorf("expected ttl more than 0, actual: %d", intResult.Code)
+		return
+	}
+}