fix(container): test failing due to lack of init scheme initialization

Author: shamikkarkhanis

internal/testutils/container.go

@@ -46,18 +46,37 @@ func SetupTestPostgres(t *testing.T) string {
 	return connStr
 }
 
-// SetupTestDB creates a fresh Postgres container, applies migrations, and returns the connection pool.
+// SetupTestDB creates a fresh Postgres container, initializes schema.sql, and returns the connection pool.
 func SetupTestDB(t *testing.T) *pgxpool.Pool {
 	ctx := context.Background()
-	connStr := SetupTestPostgres(t)
-
-	// Get repo root for migrations/
 	_, filename, _, _ := runtime.Caller(0)
 	projectRoot := filepath.Join(filepath.Dir(filename), "../..")
-	migrationsPath := filepath.Join(projectRoot, "migrations")
+	schemaPath := filepath.Join(projectRoot, "schema.sql")
+
+	pgContainer, err := postgres.Run(ctx,
+		"postgres:16-alpine",
+		postgres.WithInitScripts(schemaPath),
+		postgres.WithDatabase("test_db"),
+		postgres.WithUsername("test"),
+		postgres.WithPassword("test"),
+		testcontainers.WithWaitStrategy(
+			wait.ForLog("database system is ready to accept connections").
+				WithOccurrence(2).
+				WithStartupTimeout(30*time.Second)),
+	)
+	if err != nil {
+		t.Fatalf("failed to start postgres container: %v", err)
+	}
+
+	t.Cleanup(func() {
+		if err := pgContainer.Terminate(ctx); err != nil {
+			t.Fatalf("failed to terminate container: %v", err)
+		}
+	})
 
-	if err := database.RunMigrations(ctx, connStr, migrationsPath); err != nil {
-		t.Fatalf("failed to run migrations: %v", err)
+	connStr, err := pgContainer.ConnectionString(ctx, "sslmode=disable")
+	if err != nil {
+		t.Fatalf("failed to get connection string: %v", err)
 	}
 
 	pool, err := database.NewPool(ctx, connStr)

tests/benchmarks/suite_test.go

@@ -43,12 +43,13 @@ func TestMain(m *testing.M) {
 
 	_, filename, _, _ := runtime.Caller(0)
 	projectRoot := filepath.Join(filepath.Dir(filename), "../..")
-	migrationsPath := filepath.Join(projectRoot, "migrations")
+	schemaPath := filepath.Join(projectRoot, "schema.sql")
 
-	log.Printf("Using migrations from: %s", migrationsPath)
+	log.Printf("Using schema from: %s", schemaPath)
 
 	pgContainer, err := postgres.Run(ctx,
 		"postgres:16-alpine",
+		postgres.WithInitScripts(schemaPath),
 		postgres.WithDatabase("bench_db"),
 		postgres.WithUsername("bench"),
 		postgres.WithPassword("bench"),
@@ -76,10 +77,6 @@ func TestMain(m *testing.M) {
 		log.Fatalf("failed to get connection string: %v", err)
 	}
 
-	if err := database.RunMigrations(ctx, connStr, migrationsPath); err != nil {
-		log.Fatalf("failed to run migrations: %v", err)
-	}
-
 	benchDB, err = database.NewPool(ctx, connStr)
 	if err != nil {
 		log.Fatalf("failed to connect to database: %v", err)