debug

Author: wxiaoguang

modules/testlogger/testlogger.go

@@ -110,28 +110,24 @@ func PrintCurrentTest(t testing.TB, skip ...int) func() {
 
 	Printf("=== %s (%s:%d)\n", log.NewColoredValue(t.Name()), strings.TrimPrefix(filename, prefix), line)
 
+	getAllStack := func() string {
+		stackBuf := make([]byte, 1024*1024)
+		n := runtime.Stack(stackBuf, true)
+		return string(stackBuf[:n])
+	}
 	WriterCloser.pushT(t)
 	timeoutChecker := time.AfterFunc(TestTimeout, func() {
-		l := 128 * 1024
-		var stack []byte
-		for {
-			stack = make([]byte, l)
-			n := runtime.Stack(stack, true)
-			if n <= l {
-				stack = stack[:n]
-				break
-			}
-			l = n
-		}
-		Printf("!!! %s ... timeout: %v ... stacktrace:\n%s\n\n", log.NewColoredValue(t.Name(), log.Bold, log.FgRed), TestTimeout, string(stack))
+		Printf("!!! %s ... timeout: %v ... full stack dump:\n%s\n\n", log.NewColoredValue(t.Name(), log.Bold, log.FgRed), TestTimeout, getAllStack())
 	})
 	return func() {
 		flushStart := time.Now()
 		slowFlushChecker := time.AfterFunc(TestSlowFlush, func() {
 			Printf("+++ %s ... still flushing after %v ...\n", log.NewColoredValue(t.Name(), log.Bold, log.FgRed), TestSlowFlush)
 		})
 		if err := queue.GetManager().FlushAll(t.Context(), -1); err != nil {
-			t.Errorf("Flushing queues failed with error %v", err)
+			t.Errorf("Flushing queues failed with error: %v", err)
+			// usually it is caused by something gets stuck, so dump stacktrace
+			t.Errorf("Full stack dump:\n%s", getAllStack())
 		}
 		slowFlushChecker.Stop()
 		timeoutChecker.Stop()