Function DBPoolClone

rivershared/riversharedtest/riversharedtest.go:88–103

DBPoolClone returns a disposable clone of DBPool. Share resources by using DBPool when possible, but this is useless for areas like stress tests where context cancellations are likely to end up closing the pool. Unlike DBPool, adds a test cleanup hook that closes the pool after run.

(ctx context.Context, tb testing.TB)

//
// Unlike DBPool, adds a test cleanup hook that closes the pool after run.
func DBPoolClone(ctx context.Context, tb testing.TB) *pgxpool.Pool {
	tb.Helper()

	dbPool := DBPool(ctx, tb)

	config := dbPool.Config()
	config.MaxConns = 4 // dramatically reduce max allowed conns for clones so we they don't clobber the database server

	var err error
	dbPool, err = pgxpool.NewWithConfig(ctx, config)
	require.NoError(tb, err)

	tb.Cleanup(dbPool.Close)

	return dbPool
}

// Gets an SQLite test directory at project root so it's invariant of which
// package tests are being run in.