apps/redis-go
0
0
Fork 0
mirror of https://github.com/HDT3213/godis.git synced 2025-10-05 00:42:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize project structure

Browse Source
This commit is contained in:
hdt3213
2021-11-08 23:05:53 +08:00
parent 3c5a44b7c0
commit 4163a45278
34 changed files with 82 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

186
database/tx_utils_test.go Normal file
View File

@@ -0,0 +1,186 @@
package database
import (
"github.com/hdt3213/godis/lib/utils"
"github.com/hdt3213/godis/redis/reply/asserts"
"testing"
"time"
)
func TestRollbackGivenKeys(t *testing.T) {
testDB.Flush()
// rollback to string
key := utils.RandString(10)
value := utils.RandString(10)
testDB.Exec(nil, utils.ToCmdLine("SET", key, value, "EX", "200"))
undoCmdLines := rollbackGivenKeys(testDB, key)
rawExpire, _ := testDB.ttlMap.Get(key)
expireTime, _ := rawExpire.(time.Time)
// override given key
value2 := value + utils.RandString(5)
testDB.Exec(nil, utils.ToCmdLine("SET", key, value2, "EX", "1000"))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("GET", key))
asserts.AssertBulkReply(t, actual, value)
rawExpire2, _ := testDB.ttlMap.Get(key)
expireTime2, _ := rawExpire2.(time.Time)
timeDiff := expireTime.Sub(expireTime2)
if timeDiff < -time.Millisecond || timeDiff > time.Millisecond {
t.Error("rollback ttl failed")
}
}
func TestRollbackToList(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
value2 := utils.RandString(10)
testDB.Remove(key)
testDB.Exec(nil, utils.ToCmdLine("RPUSH", key, value, value2))
undoCmdLines := rollbackGivenKeys(testDB, key)
testDB.Exec(nil, utils.ToCmdLine("LREM", key, value2))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("LRANGE", key, "0", "-1"))
asserts.AssertMultiBulkReply(t, actual, []string{value, value2})
}
func TestRollbackToSet(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
value2 := utils.RandString(10)
testDB.Remove(key)
cmdLine := utils.ToCmdLine("SADD", key, value)
testDB.Exec(nil, cmdLine)
undoCmdLines := rollbackFirstKey(testDB, cmdLine[1:])
testDB.Exec(nil, utils.ToCmdLine("SADD", key, value2))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("SMembers", key))
asserts.AssertMultiBulkReply(t, actual, []string{value})
}
func TestRollbackSetMembers(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
testDB.Remove(key)
// undo srem
cmdLine := utils.ToCmdLine("SADD", key, value)
testDB.Exec(nil, cmdLine)
undoCmdLines := undoSetChange(testDB, cmdLine[1:])
testDB.Exec(nil, utils.ToCmdLine("SREM", key, value))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("SIsMember", key, value))
asserts.AssertIntReply(t, actual, 1)
// undo sadd
testDB.Remove(key)
value2 := utils.RandString(10)
testDB.Exec(nil, utils.ToCmdLine("SADD", key, value2))
cmdLine = utils.ToCmdLine("SADD", key, value)
undoCmdLines = undoSetChange(testDB, cmdLine[1:])
testDB.Exec(nil, cmdLine)
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual = testDB.Exec(nil, utils.ToCmdLine("SIsMember", key, value))
asserts.AssertIntReply(t, actual, 0)
// undo sadd, only member
testDB.Remove(key)
undoCmdLines = rollbackSetMembers(testDB, key, value)
testDB.Exec(nil, utils.ToCmdLine("SAdd", key, value))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual = testDB.Exec(nil, utils.ToCmdLine("type", key))
asserts.AssertStatusReply(t, actual, "none")
}
func TestRollbackToHash(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
value2 := utils.RandString(10)
testDB.Remove(key)
testDB.Exec(nil, utils.ToCmdLine("HSet", key, value, value))
undoCmdLines := rollbackGivenKeys(testDB, key)
testDB.Exec(nil, utils.ToCmdLine("HSet", key, value, value2))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("HGET", key, value))
asserts.AssertBulkReply(t, actual, value)
}
func TestRollbackHashFields(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
value2 := utils.RandString(10)
testDB.Remove(key)
testDB.Exec(nil, utils.ToCmdLine("HSet", key, value, value))
undoCmdLines := rollbackHashFields(testDB, key, value, value2)
testDB.Exec(nil, utils.ToCmdLine("HSet", key, value, value2, value2, value2))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("HGET", key, value))
asserts.AssertBulkReply(t, actual, value)
actual = testDB.Exec(nil, utils.ToCmdLine("HGET", key, value2))
asserts.AssertNullBulk(t, actual)
testDB.Remove(key)
undoCmdLines = rollbackHashFields(testDB, key, value)
testDB.Exec(nil, utils.ToCmdLine("HSet", key, value, value))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual = testDB.Exec(nil, utils.ToCmdLine("type", key))
asserts.AssertStatusReply(t, actual, "none")
}
func TestRollbackToZSet(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
testDB.Remove(key)
testDB.Exec(nil, utils.ToCmdLine("ZADD", key, "1", value))
undoCmdLines := rollbackGivenKeys(testDB, key)
testDB.Exec(nil, utils.ToCmdLine("ZADD", key, "2", value))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("ZSCORE", key, value))
asserts.AssertBulkReply(t, actual, "1")
}
func TestRollbackZSetFields(t *testing.T) {
key := utils.RandString(10)
value := utils.RandString(10)
value2 := utils.RandString(10)
testDB.Remove(key)
testDB.Exec(nil, utils.ToCmdLine("ZADD", key, "1", value))
undoCmdLines := rollbackZSetFields(testDB, key, value, value2)
testDB.Exec(nil, utils.ToCmdLine("ZADD", key, "2", value, "3", value2))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual := testDB.Exec(nil, utils.ToCmdLine("ZSCORE", key, value))
asserts.AssertBulkReply(t, actual, "1")
actual = testDB.Exec(nil, utils.ToCmdLine("ZSCORE", key, value2))
asserts.AssertNullBulk(t, actual)
testDB.Remove(key)
undoCmdLines = rollbackZSetFields(testDB, key, value)
testDB.Exec(nil, utils.ToCmdLine("ZADD", key, "1", value))
for _, cmdLine := range undoCmdLines {
testDB.Exec(nil, cmdLine)
}
actual = testDB.Exec(nil, utils.ToCmdLine("type", key))
asserts.AssertStatusReply(t, actual, "none")
}