Use progressWriter.FormatAction

Author: masaru-iritani

internal/app/app.go

@@ -68,13 +68,14 @@ func Run(option Options) (int, error) {
 		defer option.FollowOutputWriter.Close()
 	}
 
+	progressWriter := newConsoleWriter(option.Output, option.Format, option.DisableColor)
 	summary, err := parse.Process(
 		reader,
 		parse.WithFollowOutput(option.FollowOutput),
 		parse.WithFollowVersboseOutput(option.FollowOutputVerbose),
 		parse.WithWriter(option.FollowOutputWriter),
 		parse.WithProgress(option.Progress),
-		parse.WithProgressOutput(option.ProgressOutput),
+		parse.WithProgressOutput(progressWriter),
 	)
 	if err != nil {
 		return 1, err

parse/process.go

@@ -142,7 +142,7 @@ func Process(r io.Reader, optionsFunc ...OptionsFunc) (*GoTestSummary, error) {
 
 // printProgress prints a single summary line for each PASS or FAIL package.
 // This is useful for long-running test suites.
-func printProgress(w io.Writer, e *Event, summary map[string]*Package) {
+func printProgress(w progressWriter, e *Event, summary map[string]*Package) {
 	if !e.LastLine() {
 		return
 	}
@@ -174,7 +174,7 @@ func printProgress(w io.Writer, e *Event, summary map[string]*Package) {
 	//
 	// We modify this output slightly so it's more consistent and easier to parse.
 	fmt.Fprintf(w, "[%s]\t%10s\t%s%s\n",
-		strings.ToUpper(action.String()),
+		w.FormatAction(action),
 		strconv.FormatFloat(e.Elapsed, 'f', 2, 64)+"s",
 		e.Package,
 		suffix,

parse/process_options.go

@@ -4,14 +4,19 @@ import (
 	"io"
 )
 
+type progressWriter interface {
+	io.Writer
+	FormatAction(Action) string
+}
+
 type options struct {
 	w             io.Writer
 	follow        bool
 	followVerbose bool
 	debug         bool
 
 	progress       bool
-	progressOutput io.Writer
+	progressOutput progressWriter
 }
 
 type OptionsFunc func(o *options)
@@ -36,6 +41,6 @@ func WithProgress(b bool) OptionsFunc {
 	return func(o *options) { o.progress = b }
 }
 
-func WithProgressOutput(w io.Writer) OptionsFunc {
+func WithProgressOutput(w progressWriter) OptionsFunc {
 	return func(o *options) { o.progressOutput = w }
 }