Function createTestDatabase

pkg/database/postgres/repair_test.go:240–315

Helper functions for integration tests createTestDatabase creates a test database instance with container

()

// createTestDatabase creates a test database instance with container
func createTestDatabase() (*Postgres, testcontainers.Container, error) {
	ctx := context.Background()

	image := "postgres:15-alpine"

	container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: testcontainers.ContainerRequest{
			Image:        image,
			ExposedPorts: []string{"5432/tcp"},
			Env: map[string]string{
				"POSTGRES_USER":     "postgres",
				"POSTGRES_PASSWORD": "postgres",
				"POSTGRES_DB":       "permify",
			},
			WaitingFor: wait.ForAll(
				wait.ForLog("database system is ready to accept connections"),
				wait.ForListeningPort("5432/tcp"),
			),
		},
		Started: true,
	})
	if err != nil {
		return nil, nil, err
	}

	// Enable track_commit_timestamp for XID tracking
	_, _, execErr := container.Exec(ctx, []string{"psql", "-U", "postgres", "-c", "ALTER SYSTEM SET track_commit_timestamp = on;"})
	if execErr != nil {
		container.Terminate(ctx)
		return nil, nil, execErr
	}

	// Restart container to apply the setting
	stopTimeout := 2 * time.Second
	err = container.Stop(context.Background(), &stopTimeout)
	if err != nil {
		container.Terminate(ctx)
		return nil, nil, err
	}

	err = container.Start(context.Background())
	if err != nil {
		container.Terminate(ctx)
		return nil, nil, err
	}

	// Get connection details
	host, err := container.Host(ctx)
	if err != nil {
		container.Terminate(ctx)
		return nil, nil, err
	}

	port, err := container.MappedPort(ctx, "5432")
	if err != nil {
		container.Terminate(ctx)
		return nil, nil, err
	}