Add consoleWriter.FormatAction

Author: masaru-iritani

internal/app/console_writer.go

@@ -2,9 +2,12 @@ package app
 
 import (
 	"io"
+	"strings"
 
 	"github.com/charmbracelet/lipgloss"
 	"github.com/muesli/termenv"
+
+	"github.com/mfridman/tparse/parse"
 )
 
 type OutputFormat int
@@ -81,3 +84,17 @@ func newConsoleWriter(w io.Writer, format OutputFormat, disableColor bool) *cons
 	}
 	return cw
 }
+
+func (w *consoleWriter) FormatAction(action parse.Action) string {
+	s := strings.ToUpper(action.String())
+	switch action {
+	case parse.ActionPass:
+		return w.green(s)
+	case parse.ActionSkip:
+		return w.yellow(s)
+	case parse.ActionFail:
+		return w.red(s)
+	default:
+		return s
+	}
+}

internal/app/table_summary.go

@@ -167,15 +167,7 @@ func (c *consoleWriter) summaryTable(
 			}
 		}
 
-		status := strings.ToUpper(pkg.Summary.Action.String())
-		switch pkg.Summary.Action {
-		case parse.ActionPass:
-			status = c.green(status)
-		case parse.ActionSkip:
-			status = c.yellow(status)
-		case parse.ActionFail:
-			status = c.red(status)
-		}
+		status := c.FormatAction(pkg.Summary.Action)
 
 		// Skip packages with no coverage to mimic nocoverageredesign behavior (changed in github.com/golang/go/issues/24570)
 		totalTests := len(pkg.TestsByAction(parse.ActionPass)) + len(pkg.TestsByAction(parse.ActionFail)) + len(pkg.TestsByAction(parse.ActionSkip))

internal/app/table_tests.go

@@ -87,16 +87,7 @@ func (c *consoleWriter) testsTable(packages []*parse.Package, option TestTableOp
 
 			testName := shortenTestName(t.Name, option.Trim, 32)
 
-			status := strings.ToUpper(t.Status().String())
-			switch t.Status() {
-			case parse.ActionPass:
-				status = c.green(status)
-			case parse.ActionSkip:
-				status = c.yellow(status)
-			case parse.ActionFail:
-				status = c.red(status)
-			}
-
+			status := c.FormatAction(t.Status())
 			packageName := shortenPackageName(t.Package, packagePrefix, 16, option.Trim, option.TrimPath)
 
 			row := testRow{
@@ -155,15 +146,7 @@ func (c *consoleWriter) testsTableMarkdown(packages []*parse.Package, option Tes
 
 			testName := shortenTestName(t.Name, option.Trim, 32)
 
-			status := strings.ToUpper(t.Status().String())
-			switch t.Status() {
-			case parse.ActionPass:
-				status = c.green(status)
-			case parse.ActionSkip:
-				status = c.yellow(status)
-			case parse.ActionFail:
-				status = c.red(status)
-			}
+			status := c.FormatAction(t.Status())
 			data.Append([]string{
 				status,
 				strconv.FormatFloat(t.Elapsed(), 'f', 2, 64),