96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
// Copyright 2013 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 integration
|
|
|
|
import (
|
|
"net/http"
|
|
"runtime"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func interestingGoroutines() (gs []string) {
|
|
buf := make([]byte, 2<<20)
|
|
buf = buf[:runtime.Stack(buf, true)]
|
|
for _, g := range strings.Split(string(buf), "\n\n") {
|
|
sl := strings.SplitN(g, "\n", 2)
|
|
if len(sl) != 2 {
|
|
continue
|
|
}
|
|
stack := strings.TrimSpace(sl[1])
|
|
if stack == "" ||
|
|
strings.Contains(stack, "created by testing.RunTests") ||
|
|
strings.Contains(stack, "testing.Main(") ||
|
|
strings.Contains(stack, "runtime.goexit") ||
|
|
strings.Contains(stack, "created by runtime.gc") ||
|
|
strings.Contains(stack, "runtime.MHeap_Scavenger") {
|
|
continue
|
|
}
|
|
gs = append(gs, stack)
|
|
}
|
|
sort.Strings(gs)
|
|
return
|
|
}
|
|
|
|
// Verify the other tests didn't leave any goroutines running.
|
|
// This is in a file named z_last_test.go so it sorts at the end.
|
|
func TestGoroutinesRunning(t *testing.T) {
|
|
t.Skip("TODO: etcdserver.Sender may still dial closed remote endpoint and need some time to timeout.")
|
|
if testing.Short() {
|
|
t.Skip("not counting goroutines for leakage in -short mode")
|
|
}
|
|
gs := interestingGoroutines()
|
|
|
|
n := 0
|
|
stackCount := make(map[string]int)
|
|
for _, g := range gs {
|
|
stackCount[g]++
|
|
n++
|
|
}
|
|
|
|
t.Logf("num goroutines = %d", n)
|
|
if n > 0 {
|
|
t.Error("Too many goroutines.")
|
|
for stack, count := range stackCount {
|
|
t.Logf("%d instances of:\n%s", count, stack)
|
|
}
|
|
}
|
|
}
|
|
|
|
func afterTest(t *testing.T) {
|
|
http.DefaultTransport.(*http.Transport).CloseIdleConnections()
|
|
if testing.Short() {
|
|
return
|
|
}
|
|
var bad string
|
|
badSubstring := map[string]string{
|
|
").readLoop(": "a Transport",
|
|
").writeLoop(": "a Transport",
|
|
"created by net/http/httptest.(*Server).Start": "an httptest.Server",
|
|
"timeoutHandler": "a TimeoutHandler",
|
|
"net.(*netFD).connect(": "a timing out dial",
|
|
").noteClientGone(": "a closenotifier sender",
|
|
}
|
|
var stacks string
|
|
for i := 0; i < 6; i++ {
|
|
bad = ""
|
|
stacks = strings.Join(interestingGoroutines(), "\n\n")
|
|
for substr, what := range badSubstring {
|
|
if strings.Contains(stacks, substr) {
|
|
bad = what
|
|
}
|
|
}
|
|
if bad == "" {
|
|
return
|
|
}
|
|
// Bad stuff found, but goroutines might just still be
|
|
// shutting down, so give it some time.
|
|
time.Sleep(50 * time.Millisecond)
|
|
}
|
|
t.Errorf("Test appears to have leaked %s:\n%s", bad, stacks)
|
|
}
|