package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-09-29 13:52:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update redis_test.go

Browse Source
This commit is contained in:
Fenny
2020-11-04 02:28:38 +01:00
parent 86441817d6
commit 148845bb8b
1 changed files with 7 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
redis/redis_test.go
View File

@@ -5,7 +5,6 @@ package redis
import (
"os"
"testing"
"time"
"github.com/gofiber/utils"
)
@@ -18,159 +17,20 @@ func init() {
}
}
func Test_Redis_Set(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
store.Set(id, value, 0)
var (
returnedValue []byte
exp int64
)
store.db.QueryRow(store.sqlSelect, id).Scan(&returnedValue, &exp)
utils.AssertEqual(t, returnedValue, value)
utils.AssertEqual(t, exp, int64(0))
}
func Test_Redis_SetExpiry(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
expiry := time.Second * 10
store.Set(id, value, expiry)
now := time.Now().Unix()
var (
returnedValue []byte
exp int64
)
store.db.QueryRow(store.sqlSelect, id).Scan(&returnedValue, &exp)
delta := exp - now
upperBound := int64(expiry.Seconds())
lowerBound := upperBound - 2
if !(delta <= upperBound && delta > lowerBound) {
t.Fatalf("Test_SetExpiry: expiry delta out of bounds (is %d, must be %d<x<=%d)", delta, lowerBound, upperBound)
}
}
func Test_Redis_Get(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
key := "john"
value := []byte("doe")
store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
returnedValue, err := store.Get(id)
result, err := store.Get(id)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, value, returnedValue)
utils.AssertEqual(t, nil, result)
}
func Test_Redis_Get_NoRows(t *testing.T) {
store := New(storeConfig)
id := "hello"
returnedValue, err := store.Get(id)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, 0, len(returnedValue))
}
func Test_Redis_Get_Expired(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), time.Now().Add(time.Minute*-1).Unix())
returnedValue, err := store.Get(id)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, 0, len(returnedValue))
}
func Test_Redis_Delete(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
err := store.Delete(id)
err = store.Set(key, value)
utils.AssertEqual(t, nil, err)
row := store.db.QueryRow(store.sqlSelect, id)
err = row.Scan()
utils.AssertEqual(t, noRows, err.Error())
}
func Test_Redis_Clear(t *testing.T) {
store := New(storeConfig)
id := "hello"
value := []byte("Hi there!")
store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
err := store.Clear()
result, err := store.Get(id)
utils.AssertEqual(t, nil, err)
row := store.db.QueryRow(store.sqlSelect, id)
err = row.Scan()
utils.AssertEqual(t, noRows, err.Error())
}
func Benchmark_Redis_Set(b *testing.B) {
store := New(storeConfig)
key := "aaaa"
val := []byte("This is a value")
expiry := time.Second * 60
b.ResetTimer()
for n := 0; n < b.N; n++ {
store.Set(key, val, expiry)
}
}
func Benchmark_Redis_Get(b *testing.B) {
store := New(storeConfig)
key := "aaaa"
val := []byte("This is a value")
store.Set(key, val, 0)
b.ResetTimer()
for n := 0; n < b.N; n++ {
store.Get(key)
}
utils.AssertEqual(t, value, result)
}