mongox-go-driver/mongox/database/index_test.go

package database_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/mainnika/mongox-go-driver/v2/mongox-testing/database"
	"github.com/mainnika/mongox-go-driver/v2/mongox/base/oidbased"
)

func TestDatabase_Ensure(t *testing.T) {

	db, err := database.NewEphemeral("mongodb://localhost")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

	testvalues := []struct {
		doc      interface{}
		settings map[string]interface{}
		index    map[string]interface{}
	}{
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"1"`

				Foobar int `bson:"foobar" json:"foobar" index:"-,unique,allowNull,expireAfter=86400"`
				Foo    int `bson:"foo" json:"foo"`
				Bar    int `bson:"bar" json:"bar"`
			}{},
			index: map[string]interface{}{
				"background":         false,
				"expireAfterSeconds": int32(86400),
				"key": map[string]interface{}{
					"foobar": int32(-1),
				},
				"name": "-,unique,allowNull,expireAfter=86400_foobar_",
				"partialFilterExpression": map[string]interface{}{
					"foobar": map[string]interface{}{"$exists": true},
				},
				"unique": true,
			},
		},
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"2"`

				Foobar int `bson:"foobar" json:"foobar" index:",unique"`
			}{},
			index: map[string]interface{}{
				"background": false,
				"key": map[string]interface{}{
					"foobar": int32(1),
				},
				"name":   ",unique_foobar_",
				"unique": true,
			},
		},
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"3"`

				Foobar int `bson:"foobar" json:"foobar" index:"-,+foo,+-bar,unique,allowNull"`
				Foo    int `bson:"foo" json:"foo"`
				Bar    int `bson:"bar" json:"bar"`
			}{},
			index: map[string]interface{}{
				"background": false,
				"key": map[string]interface{}{
					"foobar": int32(-1),
					"foo":    int32(1),
					"bar":    int32(-1),
				},
				"name": "-,+foo,+-bar,unique,allowNull_foobar_",
				"partialFilterExpression": map[string]interface{}{
					"foobar": map[string]interface{}{"$exists": true},
				},
				"unique": true,
			},
		},
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"4"`

				Foobar int `bson:"foobar" json:"foobar" index:""`
				Foo    int `bson:"foo" json:"foo"`
				Bar    int `bson:"bar" json:"bar"`
			}{},
			index: map[string]interface{}{
				"background": false,
				"key": map[string]interface{}{
					"foobar": int32(1),
				},
				"name": "_foobar_",
			},
		},
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"5"`

				Foobar int `bson:"foobar" json:"foobar" index:"-"`
				Foo    int `bson:"foo" json:"foo"`
				Bar    int `bson:"bar" json:"bar"`
			}{},
			index: map[string]interface{}{
				"background": false,
				"key": map[string]interface{}{
					"foobar": int32(-1),
				},
				"name": "-_foobar_",
			},
		},
		{
			doc: &struct {
				oidbased.Primary `bson:",inline" json:",inline" collection:"1"`

				Foobar int `bson:"foobar" json:"foobar" index:"-,unique,allowNull,expireAfter={{.Expire}}"`
				Foo    int `bson:"foo" json:"foo"`
				Bar    int `bson:"bar" json:"bar"`
			}{},
			settings: map[string]interface{}{
				"Expire": 86400,
			},
			index: map[string]interface{}{
				"background":         false,
				"expireAfterSeconds": int32(86400),
				"key": map[string]interface{}{
					"foobar": int32(-1),
				},
				"name": "-,unique,allowNull,expireAfter=86400_foobar_",
				"partialFilterExpression": map[string]interface{}{
					"foobar": map[string]interface{}{"$exists": true},
				},
				"unique": true,
			},
		},
	}

	for _, tt := range testvalues {
		err = db.IndexEnsure(tt.settings, tt.doc)
		assert.NoError(t, err)

		indexes, _ := db.GetCollectionOf(tt.doc).Indexes().List(db.Context())
		index := new(map[string]interface{})

		indexes.Next(db.Context()) // skip _id_
		indexes.Next(db.Context())
		indexes.Decode(&index)

		for k, v := range tt.index {
			assert.Equal(t, v, (*index)[k])
		}
	}
}