WIP

asticode · Jan 24, 2024 · c8c2af6 · c8c2af6
1 parent 3cd8177
commit c8c2af6
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/astiav_test.go b/astiav_test.go
@@ -14,14 +14,17 @@ import (
 var globalHelper = newHelper()
 
 func TestMain(m *testing.M) {
-	// Run
-	m.Run()
+	// Make sure to exit with the proper code
+	var code int
+	defer func(code *int) {
+		os.Exit(*code)
+	}(&code)
 
 	// Make sure to close global helper
-	globalHelper.close()
+	defer globalHelper.close()
 
-	// Exit
-	os.Exit(0)
+	// Run
+	code = m.Run()
 }
 
 type helper struct {