storage/mongodb/mongodb_test.go

package mongodb

import (
	"context"
	"github.com/gofiber/utils"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"os"
	"testing"
	"time"
)

const (
	dbName  = "fiberStorage"
	colName = "fiberStorage"
)

var uri = os.Getenv("MONGO_URI")

func Connect() (*mongo.Database, *mongo.Collection) {
	client, err := mongo.NewClient(options.Client().ApplyURI(uri))
	if err != nil {
		panic(err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer cancel()
	err = client.Connect(ctx)

	db := client.Database(dbName)

	return db, db.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) {
	db, col := Connect()
	store := New(col)
	defer func() {
		_ = 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) {
	db, col := Connect()
	store := New(col)
	defer func() {
		_ = 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) {
	db, col := Connect()
	store := New(col)
	defer func() {
		_ = db.Client().Disconnect(context.TODO())
	}()

	key := "example_key_2"
	value := []byte("123")

	_ = store.Set(key, value, 10)
	names, _ := db.ListCollectionNames(context.TODO(), bson.D{})

	utils.AssertEqual(t, true, contains(names, colName), "has collection")
	_ = store.Clear()

	names2, _ := db.ListCollectionNames(context.TODO(), bson.D{})
	utils.AssertEqual(t, false, contains(names2, colName), "do not have collection")
}