2_sample_index_test.go

//go:build integration

package migrate

import (
	"context"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

const globalTestIndexName = "test_idx_2"

func init() {
	MustRegister(func(ctx context.Context, db *mongo.Database) error {
		keys := bson.D{{"a", 1}}
		opt := options.Index().SetName(globalTestIndexName)
		model := mongo.IndexModel{Keys: keys, Options: opt}
		_, err := db.Collection(globalTestCollection).Indexes().CreateOne(ctx, model)
		if err != nil {
			return err
		}
		return nil
	}, func(ctx context.Context, db *mongo.Database) error {
		_, err := db.Collection(globalTestCollection).Indexes().DropOne(ctx, globalTestIndexName)
		if err != nil {
			return err
		}
		return nil
	})
}