singlestore-go-bookstore service testmain

main_test.go

package service

import (
	"bookstore/config"
	"bookstore/database"
	"context"
	"log"
	"os"
	"testing"

	"github.com/testcontainers/testcontainers-go/modules/mysql"
)

func TestMain(m *testing.M) {
	ctx := context.Background()
	// set database config
	config.Database.Host = "localhost"
	config.Database.Username = "root"
	config.Database.Password = "password"
	config.Database.Database = "bookstore"

	// start mysql testcontainer
	mysqlContainer, err := mysql.Run(ctx,
		"mysql:9.0",
		mysql.WithDatabase(config.Database.Database),
		mysql.WithUsername(config.Database.Username),
		mysql.WithPassword(config.Database.Password),
	)
	if err != nil {
		log.Fatalf("failed to start container: %s", err)
	}

	// get mapped port
	p, err := mysqlContainer.MappedPort(ctx, "3306")
	if err != nil {
		log.Fatalf("failed to get mapped port: %s", err)
	}
	config.Database.Port = p.Port()

	// initialize database
	database.InitializeDB()

	// Clean up the container
	defer func() {
		if err := mysqlContainer.Terminate(ctx); err != nil {
			log.Fatalf("failed to terminate container: %s", err)
		}
	}()

	// run tests
	exitVal := m.Run()
	os.Exit(exitVal)
}