Whatever the reported TEST_TIMEOUT is, timeout 3 seconds before (#17)

Author: rickystewart

go/tools/bzltestutil/wrap.go

@@ -131,6 +131,18 @@ func Wrap(pkg string) error {
 	// and will be killed by Bazel after the grace period instead.
 	signal.Ignore(syscall.SIGTERM)
 
+	secsString, ok := os.LookupEnv("TEST_TIMEOUT")
+	if ok {
+		timeoutSecs, err := strconv.Atoi(secsString)
+		if err != nil {
+			return fmt.Errorf("could not parse TEST_TIMEOUT string %s: got error %w", secsString, err)
+		}
+		mungedTimeoutSecs := timeoutSecs - 3
+		if mungedTimeoutSecs <= 0 {
+			mungedTimeoutSecs = timeoutSecs
+		}
+		args = append([]string{"-test.timeout", fmt.Sprintf("%ds", mungedTimeoutSecs)}, args...)
+	}
 	cmd := exec.Command(exePath, args...)
 	cmd.Env = append(os.Environ(), "GO_TEST_WRAP=0")
 	cmd.Stderr = io.MultiWriter(os.Stderr, streamMerger.ErrW)