chore: add IS_GO_TEST

Change-Id: If8b912fca32c318f47972a4b424f39ab555f236c
andeya · Aug 18, 2021 · ed2b3cf · ed2b3cf
1 parent b50c8d7
commit ed2b3cf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gotest.go b/gotest.go
@@ -16,6 +16,8 @@ func init() {
 	isGoTest = checkGoTestEnv()
 }
 
+const IS_GO_TEST = "IS_GO_TEST"
+
 func checkGoTestEnv() bool {
 	for _, arg := range os.Args[1:] {
 		for _, s := range []string{
@@ -33,6 +35,7 @@ func checkGoTestEnv() bool {
 			}
 		}
 	}
-	return false
+	env := os.Getenv(IS_GO_TEST)
+	return env == "1" || env == "true"
 	// return strings.HasSuffix(os.Args[0], ".test")
 }