From e488efb2956683bd4efaf820c833f096f5ddd65e Mon Sep 17 00:00:00 2001
From: Hau Nguyen <hauvipapro@gmail.com>
Date: Sat, 23 Jul 2022 23:26:44 +0700
Subject: [PATCH] refactor: replace cmd.Run with cmd.Start and cmd.Wait

See https://stackoverflow.com/questions/8875038/redirect-stdout-pipe-of-child-process-in-go
---
 main.go | 26 ++++++++++++++++----------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/main.go b/main.go
index 5d8990e..57293fb 100644
--- a/main.go
+++ b/main.go
@@ -17,34 +17,40 @@ func main() {
 	os.Exit(code)
 }
 
+// Run go test with args
 func runGoTest() int {
-	// Run go test
-
+	// Pass all args
 	args := []string{"test"}
 	args = append(args, os.Args[1:]...)
-
 	cmd := exec.Command("go", args...)
-	cmd.Env = os.Environ()
-
-	// Output pipe and error pipe
 
+	// Output pipe
 	outReader, outWriter := io.Pipe()
 	defer outReader.Close()
 	defer outWriter.Close()
 
+	// Error pipe
 	errReader, errWriter := io.Pipe()
 	defer errReader.Close()
 	defer errWriter.Close()
 
+	// Redirect cmd pipes to our pipes
 	cmd.Stdout = outWriter
 	cmd.Stderr = errWriter
 
-	go colorOutputReader(outReader)
-	go colorErrorReader(errReader)
-
-	if err := cmd.Run(); err != nil {
+	// See https://stackoverflow.com/questions/8875038/redirect-stdout-pipe-of-child-process-in-go
+	if err := cmd.Start(); err != nil {
+		log.Printf("Failed to start: %s", err)
 		return 1
 	}
+	defer func() {
+		if err := cmd.Wait(); err != nil {
+			log.Printf("Failed to wait: %s", err)
+		}
+	}()
+
+	go colorOutputReader(outReader)
+	go colorErrorReader(errReader)
 
 	return 0
 }