Function StartWithAssert

cli/clitest/clitest.go:179–213

StartWithAssert starts the given invocation and calls assertCallback with the resulting error when the invocation completes. If assertCallback is nil, expected shutdown errors are silently tolerated.

(t *testing.T, inv *serpent.Invocation, assertCallback func(t *testing.T, err error))

// with the resulting error when the invocation completes. If assertCallback
// is nil, expected shutdown errors are silently tolerated.
func StartWithAssert(t *testing.T, inv *serpent.Invocation, assertCallback func(t *testing.T, err error)) {
	t.Helper()

	closeCh := make(chan struct{})
	// StartWithWaiter adds its own `t.Cleanup`, so we need to be sure it's added
	// before ours.
	waiter := StartWithWaiter(t, inv)
	t.Cleanup(func() {
		waiter.Cancel()
		<-closeCh
	})

	go func() {
		defer close(closeCh)
		err := waiter.Wait()

		if assertCallback != nil {
			assertCallback(t, err)
			return
		}

		switch {
		case errors.Is(err, context.Canceled):
			return
		case err != nil && strings.Contains(err.Error(), "driver: bad connection"):
			// When we cancel the context on a query that's being executed within
			// a transaction, sometimes, instead of a context.Canceled error we get
			// a "driver: bad connection" error.
			// https://github.com/lib/pq/issues/1137
			return
		default:
			assert.NoError(t, err)
		}
	}()
}

// Run runs the command and asserts that there is no error.
func Run(t *testing.T, inv *serpent.Invocation) {