mirror of
https://github.com/golang/go
synced 2024-11-18 14:34:39 -07:00
dd5f9d1033
This adds a package for dealing with stacks in tests. The only function at this time is NoLeak which verifies that a test does not leak any goroutines, and prints a stack summary if it does. Change-Id: I284b846f422334b745bef0e96d84138295120a62 Reviewed-on: https://go-review.googlesource.com/c/tools/+/232681 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Robert Findley <rfindley@google.com>
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
// Copyright 2018 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package stacktest
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/tools/internal/stack"
|
|
)
|
|
|
|
//this is only needed to support pre 1.14 when testing.TB did not have Cleanup
|
|
type withCleanup interface {
|
|
Cleanup(func())
|
|
}
|
|
|
|
// the maximum amount of time to wait for goroutines to clean themselves up.
|
|
const maxWait = time.Second
|
|
|
|
// NoLeak checks that a test (or benchmark) does not leak any goroutines.
|
|
func NoLeak(t testing.TB) {
|
|
c, ok := t.(withCleanup)
|
|
if !ok {
|
|
return
|
|
}
|
|
before := stack.Capture()
|
|
c.Cleanup(func() {
|
|
var delta stack.Delta
|
|
start := time.Now()
|
|
delay := time.Millisecond
|
|
for {
|
|
after := stack.Capture()
|
|
delta = stack.Diff(before, after)
|
|
if len(delta.After) == 0 {
|
|
// no leaks
|
|
return
|
|
}
|
|
if time.Since(start) > maxWait {
|
|
break
|
|
}
|
|
time.Sleep(delay)
|
|
delay *= 2
|
|
}
|
|
// it's been long enough, and leaks are still present
|
|
summary := stack.Summarize(delta.After)
|
|
t.Errorf("goroutine leak detected:\n%+v", summary)
|
|
})
|
|
}
|