package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-05 08:37:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🎄 xmas present

Browse Source
This commit is contained in:
Fenny
2020-11-05 04:17:05 +01:00
parent a6058cffb3
commit b4a848ca29
26 changed files with 585 additions and 750 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
mongodb/mongodb_test.go
View File

@@ -3,127 +3,107 @@
package mongodb
import (
"os"
"testing"
"time"
"github.com/gofiber/utils"
)
const (
dbName = "fiberStorage"
colName = "fiberStorage"
)
var testStore = New(Config{
Clear: true,
})
var uri = os.Getenv("MONGO_URI")
func getConfig() Config {
return Config{
URI: uri,
Database: dbName,
Collection: colName,
}
}
func Test_Redis_Set(t *testing.T) {
func Test_MongoDB_Set(t *testing.T) {
var (
store = testStore
key = "john"
val = []byte("doe")
key = "john"
val = []byte("doe")
)
err := store.Set(key, val, 0)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
}
func Test_Redis_Get(t *testing.T) {
func Test_MongoDB_Get(t *testing.T) {
var (
store = testStore
key = "john"
val = []byte("doe")
key = "john"
val = []byte("doe")
)
err := store.Set(key, val, 0)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
result, err := store.Get(key)
result, err := testStore.Get(key)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, val, result)
}
func Test_Redis_Set_Expiration(t *testing.T) {
func Test_MongoDB_Set_Expiration(t *testing.T) {
var (
store = testStore
key = "john"
val = []byte("doe")
exp = 500 * time.Millisecond
key = "john"
val = []byte("doe")
exp = 500 * time.Millisecond
)
err := store.Set(key, val, exp)
err := testStore.Set(key, val, exp)
utils.AssertEqual(t, nil, err)
time.Sleep(1 * time.Second)
}
func Test_Redis_Get_Expired(t *testing.T) {
func Test_MongoDB_Get_Expired(t *testing.T) {
var (
store = testStore
key = "john"
key = "john"
)
result, err := store.Get(key)
result, err := testStore.Get(key)
utils.AssertEqual(t, ErrNotExist, err)
utils.AssertEqual(t, true, len(result) == 0)
}
func Test_Redis_Get_NotExist(t *testing.T) {
var store = testStore
func Test_MongoDB_Get_NotExist(t *testing.T) {
result, err := store.Get("notexist")
result, err := testStore.Get("notexist")
utils.AssertEqual(t, ErrNotExist, err)
utils.AssertEqual(t, true, len(result) == 0)
}
func Test_Redis_Delete(t *testing.T) {
func Test_MongoDB_Delete(t *testing.T) {
var (
store = testStore
key = "john"
val = []byte("doe")
key = "john"
val = []byte("doe")
)
err := store.Set(key, val, 0)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
err = store.Delete(key)
err = testStore.Delete(key)
utils.AssertEqual(t, nil, err)
result, err := store.Get(key)
result, err := testStore.Get(key)
utils.AssertEqual(t, ErrNotExist, err)
utils.AssertEqual(t, true, len(result) == 0)
}
func Test_Redis_Clear(t *testing.T) {
func Test_MongoDB_Clear(t *testing.T) {
var (
store = testStore
val = []byte("doe")
val = []byte("doe")
)
err := store.Set("john1", val, 0)
err := testStore.Set("john1", val, 0)
utils.AssertEqual(t, nil, err)
err = store.Set("john2", val, 0)
err = testStore.Set("john2", val, 0)
utils.AssertEqual(t, nil, err)
err = store.Clear()
err = testStore.Clear()
utils.AssertEqual(t, nil, err)
result, err := store.Get("john1")
result, err := testStore.Get("john1")
utils.AssertEqual(t, ErrNotExist, err)
utils.AssertEqual(t, true, len(result) == 0)
result, err = store.Get("john2")
result, err = testStore.Get("john2")
utils.AssertEqual(t, ErrNotExist, err)
utils.AssertEqual(t, true, len(result) == 0)
}