logging and middleware chaining

internal/test/test_database.go

@@ -0,0 +1,62 @@
+package test
+
+import (
+	"context"
+	"database/sql"
+	"log/slog"
+	"testing"
+	"time"
+
+	"gitea.michaelthomson.dev/mthomson/habits/internal/migrate"
+	"github.com/testcontainers/testcontainers-go"
+	"github.com/testcontainers/testcontainers-go/modules/postgres"
+	"github.com/testcontainers/testcontainers-go/wait"
+)
+
+type TestDatabase struct {
+	Db        *sql.DB
+	container testcontainers.Container
+}
+
+func NewTestDatabase(tb testing.TB) *TestDatabase {
+	tb.Helper()
+	ctx := context.Background()
+	// create container
+	postgresContainer, err := postgres.Run(ctx,
+		"postgres:16-alpine",
+		postgres.WithDatabase("todo"),
+		postgres.WithUsername("todo"),
+		postgres.WithPassword("password"),
+		testcontainers.WithWaitStrategy(
+			wait.ForLog("database system is ready to accept connections").
+				WithOccurrence(2).
+				WithStartupTimeout(5*time.Second)),
+	)
+
+	if err != nil {
+		tb.Fatalf("Failed to create postgres container, %v", err)
+	}
+
+	connectionString, err := postgresContainer.ConnectionString(ctx)
+	if err != nil {
+		tb.Fatalf("Failed to get connection string: %v", err)
+	}
+
+	// create db pool
+	db, err := sql.Open("pgx", connectionString)
+	if err != nil {
+		tb.Fatalf("Failed to open db pool: %v", err)
+	}
+
+	migrate.Migrate(slog.Default(), db)
+
+	return &TestDatabase{
+		Db:        db,
+		container: postgresContainer,
+	}
+}
+
+func (tdb *TestDatabase) TearDown() {
+	tdb.Db.Close()
+	_ = tdb.container.Terminate(context.Background())
+}