feat: add unit test for printer

Signed-off-by: baoyinghai_yewu <[email protected]>
kosmos-io · Nov 25, 2024 · 681d152 · 681d152
1 parent 9962047
commit 681d152
Show file tree

Hide file tree

Showing 3 changed files with 1,838 additions and 3 deletions.
diff --git a/pkg/command/share/printer.go b/pkg/command/share/printer.go
@@ -2,6 +2,7 @@ package share
 
 import (
 	"fmt"
+	"io"
 	"os"
 	"strconv"
 
@@ -19,13 +20,16 @@ type PrintCheckData struct {
 }
 
 func PrintResult(resultData []*PrintCheckData) {
-	table := tablewriter.NewWriter(os.Stdout)
+	PrintResultWithWriter(resultData, os.Stdout)
+}
+func PrintResultWithWriter(resultData []*PrintCheckData, writer io.Writer) {
+	table := tablewriter.NewWriter(writer)
 	table.SetHeader([]string{"S/N", "SRC_NODE_NAME", "DST_NODE_NAME", "TARGETP", "RESULT"})
 
-	tableException := tablewriter.NewWriter(os.Stdout)
+	tableException := tablewriter.NewWriter(writer)
 	tableException.SetHeader([]string{"S/N", "SRC_NODE_NAME", "DST_NODE_NAME", "TARGET", "RESULT", "LOG"})
 
-	tableFailed := tablewriter.NewWriter(os.Stdout)
+	tableFailed := tablewriter.NewWriter(writer)
 	tableFailed.SetHeader([]string{"S/N", "SRC_NODE_NAME", "DST_NODE_NAME", "TARGET", "RESULT", "LOG"})
 
 	resumeData := []*PrintCheckData{}