Refactor test result presentation and add short results table generation

Author: lukaszcl

tools/flakeguard/cmd/run.go

@@ -174,12 +174,6 @@ var RunTestsCmd = &cobra.Command{
 			flushSummaryAndExit(ErrorExitCode)
 		}
 
-		// Accumulate main summary into summaryBuffer
-		fmt.Fprint(&summaryBuffer, "\nFlakeguard Main Summary:\n")
-		fmt.Fprintf(&summaryBuffer, "-------------------------\n")
-		reports.RenderTestReport(&summaryBuffer, mainReport, false, false)
-		fmt.Fprintln(&summaryBuffer)
-
 		// Rerun failed tests
 		if rerunFailedCount > 0 {
 			failedTests := reports.FilterTests(mainReport.Results, func(tr reports.TestResult) bool {
@@ -210,11 +204,6 @@ var RunTestsCmd = &cobra.Command{
 				flushSummaryAndExit(ErrorExitCode)
 			}
 
-			fmt.Fprint(&summaryBuffer, "\nAll Tests That Were Rerun:\n")
-			fmt.Fprintf(&summaryBuffer, "---------------------------\n")
-			reports.PrintTestResultsTable(&summaryBuffer, rerunReport.Results, false, false)
-			fmt.Fprintln(&summaryBuffer)
-
 			// Save the rerun test report to file
 			if rerunResultsPath != "" && len(rerunResults) > 0 {
 				if err := reports.SaveTestResultsToFile(rerunResults, rerunResultsPath); err != nil {
@@ -230,13 +219,11 @@ var RunTestsCmd = &cobra.Command{
 			})
 
 			if len(failedAfterRerun) > 0 {
-				fmt.Fprint(&summaryBuffer, "\nTests That Have 0 Success Runs:\n")
-				fmt.Fprintf(&summaryBuffer, "--------------------------------\n")
-				reports.PrintTestResultsTable(&summaryBuffer, failedAfterRerun, false, false)
+				fmt.Fprint(&summaryBuffer, "\nTests That Have 0 Success Rate After Reruns:\n\n")
+				reports.PrintTestResultsTable(&summaryBuffer, failedAfterRerun, false, false, true)
 				fmt.Fprintln(&summaryBuffer)
 
-				fmt.Fprint(&summaryBuffer, "\nLogs From All Reruns:\n")
-				fmt.Fprintf(&summaryBuffer, "----------------------\n")
+				fmt.Fprint(&summaryBuffer, "\nLogs:\n\n")
 				err := rerunReport.PrintGotestsumOutput(&summaryBuffer, "pkgname")
 				if err != nil {
 					log.Error().Err(err).Msg("Error printing gotestsum output")

tools/flakeguard/reports/presentation.go

@@ -75,6 +75,52 @@ func generateTestResultsTable(
 	return table
 }
 
+func generateShortTestResultsTable(
+	results []TestResult,
+	markdown bool,
+	filter func(TestResult) bool,
+) [][]string {
+	p := message.NewPrinter(language.English)
+
+	// Define the columns you want
+	headers := []string{"Name", "Path", "Runs", "Failures", "Code Owners"}
+
+	// Optionally format the headers for Markdown
+	if markdown {
+		for i, header := range headers {
+			headers[i] = fmt.Sprintf("**%s**", header)
+		}
+	}
+
+	// Initialize table with headers
+	table := [][]string{headers}
+
+	// Fill the table rows
+	for _, r := range results {
+		if !filter(r) {
+			continue
+		}
+
+		// Format the Code Owners
+		owners := "Unknown"
+		if len(r.CodeOwners) > 0 {
+			owners = strings.Join(r.CodeOwners, ", ")
+		}
+
+		row := []string{
+			r.TestName,
+			r.TestPath,
+			p.Sprintf("%d", r.Runs),
+			p.Sprintf("%d", r.Failures),
+			owners,
+		}
+
+		table = append(table, row)
+	}
+
+	return table
+}
+
 // GenerateFlakyTestsTable returns a table with only the flaky tests.
 func GenerateFlakyTestsTable(
 	testReport TestReport,
@@ -90,10 +136,17 @@ func PrintTestResultsTable(
 	w io.Writer,
 	results []TestResult,
 	markdown bool,
-	collapsible bool) {
-	table := generateTestResultsTable(results, markdown, func(result TestResult) bool {
+	collapsible bool,
+	shortTable bool) {
+	filter := func(result TestResult) bool {
 		return true // Include all tests
-	})
+	}
+	var table [][]string
+	if shortTable {
+		table = generateShortTestResultsTable(results, markdown, filter)
+	} else {
+		table = generateTestResultsTable(results, markdown, filter)
+	}
 	printTable(w, table, collapsible)
 }