64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package mongodb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/vault/helper/testhelpers/docker"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"go.mongodb.org/mongo-driver/mongo/readpref"
|
|
)
|
|
|
|
// PrepareTestContainer calls PrepareTestContainerWithDatabase without a
|
|
// database name value, which results in configuring a database named "test"
|
|
func PrepareTestContainer(t *testing.T, version string) (cleanup func(), retURL string) {
|
|
return PrepareTestContainerWithDatabase(t, version, "")
|
|
}
|
|
|
|
// PrepareTestContainerWithDatabase configures a test container with a given
|
|
// database name, to test non-test/admin database configurations
|
|
func PrepareTestContainerWithDatabase(t *testing.T, version, dbName string) (func(), string) {
|
|
if os.Getenv("MONGODB_URL") != "" {
|
|
return func() {}, os.Getenv("MONGODB_URL")
|
|
}
|
|
|
|
runner, err := docker.NewServiceRunner(docker.RunOptions{
|
|
ContainerName: "mongo",
|
|
ImageRepo: "docker.mirror.hashicorp.services/library/mongo",
|
|
ImageTag: version,
|
|
Ports: []string{"27017/tcp"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("could not start docker mongo: %s", err)
|
|
}
|
|
|
|
svc, err := runner.StartService(context.Background(), func(ctx context.Context, host string, port int) (docker.ServiceConfig, error) {
|
|
connURL := fmt.Sprintf("mongodb://%s:%d", host, port)
|
|
if dbName != "" {
|
|
connURL = fmt.Sprintf("%s/%s", connURL, dbName)
|
|
}
|
|
|
|
ctx, _ = context.WithTimeout(context.Background(), 1*time.Minute)
|
|
client, err := mongo.Connect(ctx, options.Client().ApplyURI(connURL))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = client.Ping(ctx, readpref.Primary())
|
|
if err = client.Disconnect(ctx); err != nil {
|
|
t.Fatal()
|
|
}
|
|
|
|
return docker.NewServiceURLParse(connURL)
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("could not start docker mongo: %s", err)
|
|
}
|
|
|
|
return svc.Cleanup, svc.Config.URL().String()
|
|
}
|