mirror of
https://github.com/gofiber/storage.git
synced 2025-10-05 08:37:10 +08:00
87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
package mongodb
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gofiber/utils"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
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) {
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.DB.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key"
|
|
value := []byte("123")
|
|
|
|
_ = store.Set(key, value, 0)
|
|
|
|
getVal, _ := store.Get(key)
|
|
|
|
utils.AssertEqual(t, value, getVal, "correctly set and get value")
|
|
}
|
|
|
|
func TestMongoStore_Delete(t *testing.T) {
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.DB.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key_2"
|
|
value := []byte("123")
|
|
|
|
_ = store.Set(key, value, 10)
|
|
_ = store.Delete(key)
|
|
|
|
getVal, _ := store.Get(key)
|
|
|
|
utils.AssertEqual(t, []byte{}, getVal, "correctly delete value")
|
|
}
|
|
|
|
func TestMongoStore_Clear(t *testing.T) {
|
|
store := New(getConfig())
|
|
defer func() {
|
|
_ = store.DB.Client().Disconnect(context.TODO())
|
|
}()
|
|
|
|
key := "example_key_2"
|
|
value := []byte("123")
|
|
|
|
_ = store.Set(key, value, 10)
|
|
names, _ := store.DB.ListCollectionNames(context.TODO(), bson.D{})
|
|
|
|
utils.AssertEqual(t, true, contains(names, colName), "has collection")
|
|
_ = store.Clear()
|
|
|
|
names2, _ := store.DB.ListCollectionNames(context.TODO(), bson.D{})
|
|
utils.AssertEqual(t, false, contains(names2, colName), "do not have collection")
|
|
}
|