Function tGoContext

cli/ssh_test.go:2458–2473

tGoContext runs fn in a goroutine passing a context that will be canceled on test completion and wait until fn has finished executing. Done and cancel are returned for optionally waiting until completion or early cancellation. NOTE(mafredri): This could be moved to a helper library.

(t *testing.T, fn func(context.Context))

//
// NOTE(mafredri): This could be moved to a helper library.
func tGoContext(t *testing.T, fn func(context.Context)) (done <-chan struct{}, cancel context.CancelFunc) {
	t.Helper()

	ctx, cancel := context.WithCancel(context.Background())
	doneC := make(chan struct{})
	t.Cleanup(func() {
		cancel()
		<-done
	})
	go func() {
		fn(ctx)
		close(doneC)
	}()

	return doneC, cancel
}

// tGo runs fn in a goroutine and waits until fn has completed before
// test completion. Done is returned for optionally waiting for fn to