mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 00:42:43 +08:00
113 lines
4.3 KiB
Go
113 lines
4.3 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/hdt3213/godis/lib/utils"
|
|
"github.com/hdt3213/godis/redis/connection"
|
|
"github.com/hdt3213/godis/redis/protocol"
|
|
"github.com/hdt3213/godis/redis/protocol/asserts"
|
|
"testing"
|
|
)
|
|
|
|
func TestMultiExecOnSelf(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testCluster.db.Exec(conn, utils.ToCmdLine("FLUSHALL"))
|
|
result := testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
key := utils.RandString(10)
|
|
value := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key, value))
|
|
key2 := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("rpush", key2, value))
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("exec"))
|
|
asserts.AssertNotError(t, result)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("get", key))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("lrange", key2, "0", "-1"))
|
|
asserts.AssertMultiBulkReply(t, result, []string{value})
|
|
}
|
|
|
|
func TestEmptyMulti(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testCluster.db.Exec(conn, utils.ToCmdLine("FLUSHALL"))
|
|
result := testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("PING"))
|
|
asserts.AssertNotError(t, result)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("EXEC"))
|
|
asserts.AssertNotError(t, result)
|
|
mbr := result.(*protocol.MultiRawReply)
|
|
asserts.AssertStatusReply(t, mbr.Replies[0], "PONG")
|
|
}
|
|
|
|
func TestMultiExecOnOthers(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testCluster.db.Exec(conn, utils.ToCmdLine("FLUSHALL"))
|
|
result := testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
key := utils.RandString(10)
|
|
value := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("rpush", key, value))
|
|
testCluster.Exec(conn, utils.ToCmdLine("lrange", key, "0", "-1"))
|
|
|
|
cmdLines := conn.GetQueuedCmdLine()
|
|
rawResp := execMultiOnOtherNode(testCluster, conn, testCluster.self, nil, cmdLines)
|
|
rep := rawResp.(*protocol.MultiRawReply)
|
|
if len(rep.Replies) != 2 {
|
|
t.Errorf("expect 2 replies actual %d", len(rep.Replies))
|
|
}
|
|
asserts.AssertMultiBulkReply(t, rep.Replies[1], []string{value})
|
|
}
|
|
|
|
func TestWatch(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testCluster.db.Exec(conn, utils.ToCmdLine("FLUSHALL"))
|
|
key := utils.RandString(10)
|
|
value := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("watch", key))
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key, value))
|
|
result := testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
key2 := utils.RandString(10)
|
|
value2 := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key2, value2))
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("exec"))
|
|
asserts.AssertNotError(t, result)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("get", key2))
|
|
asserts.AssertNullBulk(t, result)
|
|
|
|
testCluster.Exec(conn, utils.ToCmdLine("watch", key))
|
|
result = testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key2, value2))
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("exec"))
|
|
asserts.AssertNotError(t, result)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("get", key2))
|
|
asserts.AssertBulkReply(t, result, value2)
|
|
}
|
|
|
|
func TestWatch2(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testCluster.db.Exec(conn, utils.ToCmdLine("FLUSHALL"))
|
|
key := utils.RandString(10)
|
|
value := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("watch", key))
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key, value))
|
|
result := testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
key2 := utils.RandString(10)
|
|
value2 := utils.RandString(10)
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key2, value2))
|
|
cmdLines := conn.GetQueuedCmdLine()
|
|
execMultiOnOtherNode(testCluster, conn, testCluster.self, conn.GetWatching(), cmdLines)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("get", key2))
|
|
asserts.AssertNullBulk(t, result)
|
|
|
|
testCluster.Exec(conn, utils.ToCmdLine("watch", key))
|
|
result = testCluster.Exec(conn, toArgs("MULTI"))
|
|
asserts.AssertNotError(t, result)
|
|
testCluster.Exec(conn, utils.ToCmdLine("set", key2, value2))
|
|
execMultiOnOtherNode(testCluster, conn, testCluster.self, conn.GetWatching(), cmdLines)
|
|
result = testCluster.Exec(conn, utils.ToCmdLine("get", key2))
|
|
asserts.AssertBulkReply(t, result, value2)
|
|
}
|