fix

Author: asmyasnikov

internal/xtest/leak.go

@@ -8,7 +8,7 @@ import (
 	"time"
 )
 
-func checkGoroutinesLeak(onLeak func(stacks []string)) {
+func checkGoroutinesLeak(onLeak func(goroutines []string)) {
 	var (
 		bb               = make([]byte, 2<<32)
 		currentGoroutine string

internal/xtest/leak_test.go

@@ -9,21 +9,33 @@ import (
 
 func TestCheckGoroutinesLeak(t *testing.T) {
 	t.Run("Leak", func(t *testing.T) {
-		require.Panics(t, func() {
-			defer checkGoroutinesLeak(func(stacks []string) {
-				panic("panic")
+		TestManyTimes(t, func(t testing.TB) {
+			ch := make(chan struct{})
+			require.Panics(t, func() {
+				defer checkGoroutinesLeak(func(goroutines []string) {
+					panic("panic")
+				})
+				go func() {
+					<-ch
+				}()
 			})
-			go func() {
-				time.Sleep(time.Second)
-			}()
-		})
+			close(ch)
+		}, StopAfter(13*time.Second))
 	})
 	t.Run("NoLeak", func(t *testing.T) {
-		require.NotPanics(t, func() {
-			defer checkGoroutinesLeak(func(stacks []string) {
-				panic("panic")
+		TestManyTimes(t, func(t testing.TB) {
+			ch := make(chan struct{})
+			require.NotPanics(t, func() {
+				defer checkGoroutinesLeak(func(goroutines []string) {
+					panic("panic")
+				})
+				defer func() {
+					<-ch
+				}()
+				go func() {
+					close(ch)
+				}()
 			})
-			time.Sleep(time.Second)
-		})
+		}, StopAfter(13*time.Second))
 	})
 }