1 files changed, 16 insertions, 0 deletions
diff --git a/test/integration/container.go b/test/integration/container.go
index 67d7603..c95bdfd 100644
--- a/test/integration/container.go
+++ b/test/integration/container.go
@@ -10,6 +10,7 @@ import (
 	"github.com/stretchr/testify/require"
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/log"
+	"github.com/testcontainers/testcontainers-go/modules/postgres"
 	"github.com/testcontainers/testcontainers-go/wait"
 	"github.com/xlgmokha/x/pkg/env"
 )
@@ -34,3 +35,18 @@ func NewContainer(t *testing.T, ctx context.Context, envVars map[string]string)
 	require.NoError(t, err)
 	return container
 }
+
+func NewPgContainer(ctx context.Context, t *testing.T) *postgres.PostgresContainer {
+	container, err := postgres.Run(ctx, "postgres:17",
+		postgres.WithDatabase("sparkle_test"),
+		postgres.WithUsername("postgres"),
+		postgres.WithPassword("secret"),
+		testcontainers.WithLogConsumers(&Logger{TB: t}),
+		testcontainers.WithLogger(log.TestLogger(t)),
+		testcontainers.WithWaitStrategy(
+			wait.ForListeningPort("5432/tcp"),
+		),
+	)
+	require.NoError(t, err)
+	return container
+}