Fix TestMain cleanup issue scroll-tech#1542

Favy217 · Oct 17, 2024 · 3d40526 · 3d40526
1 parent 41ce22b
commit 3d40526
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/rollup/tests/bridge_test.go b/rollup/tests/bridge_test.go
@@ -63,15 +63,16 @@ func setupDB(t *testing.T) *gorm.DB {
 }
 
 func TestMain(m *testing.M) {
-	defer func() {
-		if testApps != nil {
-			testApps.Free()
-		}
-		if rollupApp != nil {
-			rollupApp.Free()
-		}
-	}()
-	m.Run()
+    code := m.Run()  // Run tests and capture exit code before defer
+    defer func() {
+        if testApps != nil {
+            testApps.Free()
+        }
+        if rollupApp != nil {
+            rollupApp.Free()
+        }
+    }()
+    os.Exit(code)  // Ensure proper exit with the test code
 }
 
 func setupEnv(t *testing.T) {