Function TimeoutTest

internal/testutil/timeout.go:43–67

TimeoutTest returns a test function that executes f with a timeout, If the test does not complete in time the test is failed. This lets us set a per-test timeout instead of the global `go test -timeout` coarse timeout.

(timeout time.Duration, f func(t *testing.T))

// test does not complete in time the test is failed.  This lets us set a
// per-test timeout instead of the global `go test -timeout` coarse timeout.
func TimeoutTest(timeout time.Duration, f func(t *testing.T)) func(t *testing.T) {
	// Raise the timeout if we're run under the race detector.
	timeout *= RaceDetectorMultiplier
	// If we're in a CI environment, raise the timeout by 10x.  This mimics the
	// timeout global flag set in the Makefile.
	if os.Getenv("CI") == "true" {
		timeout *= 10
	}
	return func(t *testing.T) {
		t.Helper()
		done := make(chan bool)
		go func() {
			t.Helper()
			defer close(done)
			f(t)
		}()
		select {
		case <-time.After(timeout):
			buf := make([]byte, 1<<20)
			stacklen := runtime.Stack(buf, true)
			t.Fatalf("timed out after %s\n%s", timeout, buf[:stacklen])
		case <-done:
		}
	}
}