main_test.go: Add to detect goroutine leaks

Updates #330
nhooyr · Oct 19, 2023 · e6a7e0e · e6a7e0e
1 parent 108d137
commit e6a7e0e
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/main_test.go b/main_test.go
@@ -0,0 +1,17 @@
+package websocket_test
+
+import (
+	"fmt"
+	"os"
+	"runtime"
+	"testing"
+)
+
+func TestMain(m *testing.M) {
+	code := m.Run()
+	if runtime.NumGoroutine() != 1 {
+		fmt.Fprintf(os.Stderr, "goroutine leak detected, expected 1 but got %d goroutines\n", runtime.NumGoroutine())
+		os.Exit(1)
+	}
+	os.Exit(code)
+}