go-test-color/main.go

109 lines
2.0 KiB
Go

package main
import (
"bufio"
"fmt"
"io"
"log"
"os"
"os/exec"
"strings"
"github.com/fatih/color"
)
const cmdName = "go-test-color"
func main() {
code := runGoTest()
os.Exit(code)
}
// Run go test with args
func runGoTest() int {
// Pass all args
args := []string{"test"}
args = append(args, os.Args[1:]...)
cmd := exec.Command("go", args...)
// Read stdout and stderr
outReader, err := cmd.StdoutPipe()
if err != nil {
log.Printf("%s failed to get stdout pipe: %s", cmdName, err)
return 1
}
errReader, err := cmd.StderrPipe()
if err != nil {
log.Printf("%s failed to get stderr pipe: %s", cmdName, err)
return 1
}
// See https://stackoverflow.com/questions/8875038/redirect-stdout-pipe-of-child-process-in-go
if err := cmd.Start(); err != nil {
log.Printf("%s failed to start: %s", cmdName, err)
return 1
}
// Add color to both stdout and stderr
colorOutputReader(outReader)
colorErrorReader(errReader)
if err := cmd.Wait(); err != nil {
log.Printf("%s failed to wait: %s", cmdName, err)
return 1
}
return 0
}
func colorOutputReader(reader io.Reader) {
scanner := bufio.NewScanner(reader)
for scanner.Scan() {
line := scanner.Text()
line = strings.TrimSpace(line)
if strings.HasSuffix(line, "[no test files]") {
continue
}
if strings.HasPrefix(line, "--- PASS") ||
strings.HasPrefix(line, "PASS") ||
strings.HasPrefix(line, "ok") {
color.Green("%s\n", line)
continue
}
if strings.HasPrefix(line, "--- SKIP") {
color.Yellow("%s\n", line)
continue
}
if strings.HasPrefix(line, "--- FAIL") ||
strings.HasPrefix(line, "FAIL") {
color.Red("%s\n", line)
continue
}
fmt.Println(line)
}
if err := scanner.Err(); err != nil {
log.Printf("scanner error: %s", err)
}
}
func colorErrorReader(reader io.Reader) {
scanner := bufio.NewScanner(reader)
for scanner.Scan() {
line := scanner.Text()
color.Red("%s\n", line)
}
if err := scanner.Err(); err != nil {
log.Printf("scanner error: %s", err)
}
}