Function startIgnoringPostgresQueryCancel

cli/server_test.go:2673–2688

startIgnoringPostgresQueryCancel starts the Invocation, but excludes PostgreSQL query canceled and context cancellation errors. This prevents flakes in tests that only assert things that happen before PostgreSQL is fully initialized in the server.

(t *testing.T, inv *serpent.Invocation)

// cancellation errors. This prevents flakes in tests that only assert things that happen before PostgreSQL is fully
// initialized in the server.
func startIgnoringPostgresQueryCancel(t *testing.T, inv *serpent.Invocation) {
	t.Helper()
	clitest.StartWithAssert(t, inv, func(t *testing.T, err error) {
		if database.IsQueryCanceledError(err) {
			return
		}
		// specifically when making our initial connection to PostgreSQL, we ping the database.
		// Database driver.Conn instances can return driver.ErrBadConn on ping to remove the connection from the pool.
		// lib/pq does this no matter what the error is, including context.Canceled.
		// c.f. https://pkg.go.dev/database/sql/driver#Pinger
		if xerrors.Is(err, driver.ErrBadConn) {
			return
		}
		assert.NoError(t, err)
	})
}