Add check to skip test setup in TestMain

ubuntu · Jan 2, 2024 · 23d4351 · 23d4351
1 parent ba7a274
commit 23d4351
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pam/integration-tests/integration_test.go b/pam/integration-tests/integration_test.go
@@ -186,6 +186,11 @@ func saveArtifactsForDebug(t *testing.T, artifacts []string) {
 }
 
 func TestMain(m *testing.M) {
+	// Needed to skip the test setup when running the gpasswd mock.
+	if os.Getenv("GO_WANT_HELPER_PROCESS") != "" {
+		os.Exit(m.Run())
+	}
+
 	testutils.InstallUpdateFlag()
 	flag.Parse()