database.go

package testenv

import (
	"cmp"
	"context"
	"fmt"

	"github.com/docker/docker/api/types/container"
	"github.com/docker/go-connections/nat"
	"github.com/testcontainers/testcontainers-go"
	"github.com/testcontainers/testcontainers-go/wait"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

func (b *Builder) DBEnable(v bool) *Builder {
	b.dbEnable = v
	return b
}

func (b *Builder) DBUser(v string) *Builder {
	b.dbUser = v
	return b
}

func (b *Builder) DBPass(v string) *Builder {
	b.dbPass = v
	return b
}

func (b *Builder) DBName(v string) *Builder {
	b.dbName = v
	return b
}

func (b *Builder) DBPort(v string) *Builder {
	b.dbPort = v
	return b
}

func setupDatabase(ctx context.Context, dbUser, dbPass, dbName, hostPort string) (_ *gorm.DB, _ func() error, xerr error) {
	container, err := testcontainers.GenericContainer(ctx,
		testcontainers.GenericContainerRequest{
			ContainerRequest: testcontainers.ContainerRequest{
				Image:        "postgres:16.3-alpine",
				ExposedPorts: []string{"5432/tcp"},
				Env: map[string]string{
					"POSTGRES_USER":     dbUser,
					"POSTGRES_PASSWORD": dbPass,
					"POSTGRES_DB":       dbName,
				},
				WaitingFor: wait.ForLog("database system is ready to accept connections").WithOccurrence(2),
				HostConfigModifier: func(hostConfig *container.HostConfig) {
					hostConfig.PortBindings = map[nat.Port][]nat.PortBinding{
						"5432/tcp": {
							{
								HostIP:   "0.0.0.0",
								HostPort: hostPort,
							},
						},
					}
				},
			},
			Started: true,
		},
	)
	if err != nil {
		return nil, nil, fmt.Errorf("fail to start container: %w", err)
	}
	defer func() {
		if xerr != nil {
			container.Terminate(context.Background())
		}
	}()

	endpoint, err := container.Endpoint(ctx, "")
	if err != nil {
		return nil, nil, fmt.Errorf("fail to get endpoint: %w", err)
	}
	dsn := fmt.Sprintf("postgres://%s:%s@%s/%s?sslmode=disable", dbUser, dbPass, endpoint, dbName)

	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
	if err != nil {
		return nil, nil, err
	}

	sqlDB, err := db.DB()
	if err != nil {
		return nil, nil, fmt.Errorf("no underlying sqlDB: %w", err)
	}

	return db, func() error {
		return cmp.Or(
			sqlDB.Close(),
			container.Terminate(context.Background()),
		)
	}, nil
}