mirror of
https://github.com/gofiber/storage.git
synced 2025-10-02 23:32:43 +08:00
122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package mongodb
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/gofiber/utils"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
const (
|
|
dbName = "fiberStorage"
|
|
colName = "fiberStorage"
|
|
)
|
|
|
|
var uri = os.Getenv("MONGO_URI")
|
|
|
|
func getConfig() Config {
|
|
|
|
return Config{
|
|
URI: uri,
|
|
Database: dbName,
|
|
Collection: colName,
|
|
}
|
|
}
|
|
|
|
func contains(arr []string, item string) bool {
|
|
for _, i := range arr {
|
|
if i == item {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func TestMongoStore_Set_Get(t *testing.T) {
|
|
if uri == "" {
|
|
t.Skip()
|
|
}
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.db.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key"
|
|
value := []byte("123")
|
|
|
|
err := store.Set(key, value, 0)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
getVal, getErr := store.Get(key)
|
|
|
|
utils.AssertEqual(t, nil, getErr)
|
|
utils.AssertEqual(t, value, getVal, "correctly set and get value")
|
|
}
|
|
|
|
func TestMongoStore_Get_Invalid(t *testing.T) {
|
|
if uri == "" {
|
|
t.Skip()
|
|
}
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.db.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "random_invalid_key"
|
|
|
|
getVal, getErr := store.Get(key)
|
|
|
|
utils.AssertEqual(t, true, getErr != nil)
|
|
utils.AssertEqual(t, true, getVal == nil, "get nil if key not found")
|
|
}
|
|
|
|
func TestMongoStore_Delete(t *testing.T) {
|
|
if uri == "" {
|
|
t.Skip()
|
|
}
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.db.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key_2"
|
|
value := []byte("123")
|
|
|
|
err := store.Set(key, value, 0)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
err = store.Delete(key)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
getVal, getErr := store.Get(key)
|
|
utils.AssertEqual(t, true, getErr != nil)
|
|
utils.AssertEqual(t, true, getVal == nil, "correctly delete value")
|
|
}
|
|
|
|
func TestMongoStore_Clear(t *testing.T) {
|
|
if uri == "" {
|
|
t.Skip()
|
|
}
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.db.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key_2"
|
|
value := []byte("123")
|
|
|
|
err := store.Set(key, value, 10)
|
|
names, _ := store.db.ListCollectionNames(context.TODO(), bson.D{})
|
|
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, true, contains(names, colName), "has collection")
|
|
|
|
cErr := store.Clear()
|
|
|
|
names2, _ := store.db.ListCollectionNames(context.TODO(), bson.D{})
|
|
utils.AssertEqual(t, nil, cErr)
|
|
utils.AssertEqual(t, false, contains(names2, colName), "do not have collection")
|
|
}
|