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

Feat: Add "GetEX" and "GetDel" Commands

Browse Source
This commit is contained in:
Eriri
2022-06-04 23:09:27 +08:00
committed by finley
parent bde2c8ccbe
commit f3274fd8c8
4 changed files with 162 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
database/string_test.go
View File

@@ -283,6 +283,62 @@ func TestDecr(t *testing.T) {
}
}
func TestGetEX(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
value := utils.RandString(10)
ttl := "1000"
testDB.Exec(nil, utils.ToCmdLine("SET", key, value))
// Normal Get
actual := testDB.Exec(nil, utils.ToCmdLine("GETEX", key))
asserts.AssertBulkReply(t, actual, value)
// Test GetEX Key EX Seconds
actual = testDB.Exec(nil, utils.ToCmdLine("GETEX", key, "EX", ttl))
asserts.AssertBulkReply(t, actual, value)
actual = testDB.Exec(nil, utils.ToCmdLine("TTL", key))
intResult, ok := actual.(*protocol.IntReply)
if !ok {
t.Error(fmt.Sprintf("expected int protocol, actually %s", actual.ToBytes()))
return
}
if intResult.Code <= 0 || intResult.Code > 1000 {
t.Error(fmt.Sprintf("expected int between [0, 1000], actually %d", intResult.Code))
return
}
// Test GetEX Key Persist
actual = testDB.Exec(nil, utils.ToCmdLine("GETEX", key, "PERSIST"))
asserts.AssertBulkReply(t, actual, value)
actual = testDB.Exec(nil, utils.ToCmdLine("TTL", key))
intResult, ok = actual.(*protocol.IntReply)
if !ok {
t.Error(fmt.Sprintf("expected int protocol, actually %s", actual.ToBytes()))
return
}
if intResult.Code != -1 {
t.Error(fmt.Sprintf("expected int equals -1, actually %d", intResult.Code))
return
}
// Test GetEX Key NX Milliseconds
ttl = "1000000"
actual = testDB.Exec(nil, utils.ToCmdLine("GETEX", key, "PX", ttl))
asserts.AssertBulkReply(t, actual, value)
actual = testDB.Exec(nil, utils.ToCmdLine("TTL", key))
intResult, ok = actual.(*protocol.IntReply)
if !ok {
t.Error(fmt.Sprintf("expected int protocol, actually %s", actual.ToBytes()))
return
}
if intResult.Code <= 0 || intResult.Code > 1000000 {
t.Error(fmt.Sprintf("expected int between [0, 1000000], actually %d", intResult.Code))
return
}
}
func TestGetSet(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
@@ -300,6 +356,17 @@ func TestGetSet(t *testing.T) {
asserts.AssertBulkReply(t, actual, value)
actual = testDB.Exec(nil, utils.ToCmdLine("GET", key))
asserts.AssertBulkReply(t, actual, value2)
// Test GetDel
actual = testDB.Exec(nil, utils.ToCmdLine("GETDEL", key))
asserts.AssertBulkReply(t, actual, value2)
actual = testDB.Exec(nil, utils.ToCmdLine("GETDEL", key))
_, ok = actual.(*protocol.NullBulkReply)
if !ok {
t.Errorf("expect null bulk protocol, get: %s", string(actual.ToBytes()))
return
}
}
func TestMSetNX(t *testing.T) {