mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 00:42:43 +08:00
bf7f628810
wip: raft does not care about migrating wip: optimize code wip: raft election wip wip: fix raft leader missing log entries wip fix a dead lock batch set slot route wip: raft persist wip refactor cluster suite remove relay rename relay2 refactor: allow customizing client factory test raft refactor re-balance avoid errors caused by inconsistent status on follower nodes during raft commits test raft election
129 lines
5.6 KiB
Go
129 lines
5.6 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/hdt3213/godis/lib/utils"
|
|
"github.com/hdt3213/godis/redis/connection"
|
|
"github.com/hdt3213/godis/redis/protocol/asserts"
|
|
"testing"
|
|
)
|
|
|
|
func TestCopy(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testNodeA := testCluster[0]
|
|
testNodeB := testCluster[1]
|
|
testNodeA.Exec(conn, utils.ToCmdLine("FlushALL"))
|
|
testNodeB.Exec(conn, utils.ToCmdLine("FlushALL"))
|
|
|
|
// cross node copy
|
|
srcKey := "127.0.0.1:6399Bk2r3Sz0V5" // use fix key to ensure hashing to different node
|
|
destKey := "127.0.0.1:7379CcdC0QOopF"
|
|
value := utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value))
|
|
result := Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", srcKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
result = testNodeB.Exec(conn, utils.ToCmdLine("GET", destKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
// key exists
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey))
|
|
asserts.AssertIntReply(t, result, 0)
|
|
// replace
|
|
value = utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey, "REPLACE"))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", srcKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
result = testNodeB.Exec(conn, utils.ToCmdLine("GET", destKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
// test copy expire time
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value, "EX", "1000"))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey, "REPLACE"))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("TTL", srcKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
result = testNodeB.Exec(conn, utils.ToCmdLine("TTL", destKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
|
|
// same node copy
|
|
srcKey = "{" + testNodeA.self + "}" + utils.RandString(10)
|
|
destKey = "{" + testNodeA.self + "}" + utils.RandString(9)
|
|
value = utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", srcKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", destKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
// key exists
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey))
|
|
asserts.AssertIntReply(t, result, 0)
|
|
// replace
|
|
value = utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey, "REPLACE"))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", srcKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("GET", destKey))
|
|
asserts.AssertBulkReply(t, result, value)
|
|
// test copy expire time
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value, "EX", "1000"))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey, "REPLACE"))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("TTL", srcKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("TTL", destKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
}
|
|
|
|
func TestCopyTimeout(t *testing.T) {
|
|
conn := new(connection.FakeConn)
|
|
testNodeA := testCluster[0]
|
|
testNodeB := testCluster[1]
|
|
testNodeA.Exec(conn, utils.ToCmdLine("FlushALL"))
|
|
testNodeB.Exec(conn, utils.ToCmdLine("FlushALL"))
|
|
|
|
// test src prepare failed
|
|
timeoutFlags[0] = true
|
|
srcKey := "127.0.0.1:6399Bk2r3Sz0V5" // use fix key to ensure hashing to different node
|
|
destKey := "127.0.0.1:7379CcdC0QOopF"
|
|
value := utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value, "ex", "1000"))
|
|
result := Rename(testNodeB, conn, utils.ToCmdLine("RENAME", srcKey, destKey))
|
|
asserts.AssertErrReply(t, result, "ERR timeout")
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("EXISTS", srcKey))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("TTL", srcKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
result = testNodeB.Exec(conn, utils.ToCmdLine("EXISTS", destKey))
|
|
asserts.AssertIntReply(t, result, 0)
|
|
timeoutFlags[0] = false
|
|
|
|
// test dest prepare failed
|
|
timeoutFlags[1] = true
|
|
value = utils.RandString(10)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value, "ex", "1000"))
|
|
result = Rename(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey))
|
|
asserts.AssertErrReply(t, result, "ERR timeout")
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("EXISTS", srcKey))
|
|
asserts.AssertIntReply(t, result, 1)
|
|
result = testNodeA.Exec(conn, utils.ToCmdLine("TTL", srcKey))
|
|
asserts.AssertIntReplyGreaterThan(t, result, 0)
|
|
result = testNodeB.Exec(conn, utils.ToCmdLine("EXISTS", destKey))
|
|
asserts.AssertIntReply(t, result, 0)
|
|
timeoutFlags[1] = false
|
|
// Copying to another database
|
|
srcKey = testNodeA.self + utils.RandString(10)
|
|
value = utils.RandString(10)
|
|
destKey = srcKey + utils.RandString(2)
|
|
testNodeA.Exec(conn, utils.ToCmdLine("SET", srcKey, value))
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey, destKey, "db", "1"))
|
|
asserts.AssertErrReply(t, result, copyToAnotherDBErr)
|
|
|
|
result = Copy(testNodeA, conn, utils.ToCmdLine("COPY", srcKey))
|
|
asserts.AssertErrReply(t, result, "ERR wrong number of arguments for 'copy' command")
|
|
}
|