Function withDatabase

schema/database_test.go:78–117

(ctx context.Context, connstr string, tb TB, fn func(t TB, db *pgx.Conn))

}

func withDatabase[TB testing.TB](ctx context.Context, connstr string, tb TB, fn func(t TB, db *pgx.Conn)) {
	if connstr == "" {
		tb.Skip("connstr not defined")
	}

	var id [8]byte
	rand.Read(id[:])
	uniqueName := tb.Name() + "/" + hex.EncodeToString(id[:])

	// Create the first connection that we use to create the database.
	maindb, err := pgx.Connect(ctx, connstr)
	if err != nil {
		tb.Fatalf("Unable to connect to database: %v", err)
	}

	// Run the database creation query and defer the database cleanup query.
	if err := createDatabase(ctx, maindb, uniqueName); err != nil {
		tb.Fatalf("unable to create database: %v", err)
	}
	defer func() {
		if err := dropDatabase(ctx, maindb, uniqueName); err != nil {
			tb.Fatalf("unable to drop database: %v", err)
		}
	}()

	// Make a connection the new database.
	connstr, err = connstrWithDatabase(connstr, uniqueName)
	if err != nil {
		tb.Fatal(err)
	}

	db, err := pgx.Connect(ctx, connstr)
	if err != nil {
		tb.Fatalf("Unable to connect to database: %v", err)
	}
	defer func() { _ = db.Close(ctx) }()

	// Run our test code.
	fn(tb, db)
}

func sanitizeDatabaseName(schema string) string {
	return pgx.Identifier{schema}.Sanitize()