From e5ca0530116967d4d30d1fd9cc78b3a10e5d0044 Mon Sep 17 00:00:00 2001
From: marun <maru.newby@avalabs.org>
Date: Wed, 16 Oct 2024 12:44:12 -0700
Subject: [PATCH] [testing] Enable config of log format for bootstrap monitor
 (#3467)

---
 tests/fixture/bootstrapmonitor/cmd/main.go    | 24 +++++++++++++++----
 .../fixture/bootstrapmonitor/e2e/e2e_test.go  |  2 ++
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/tests/fixture/bootstrapmonitor/cmd/main.go b/tests/fixture/bootstrapmonitor/cmd/main.go
index 9dfe08279256..1e04fd882ad6 100644
--- a/tests/fixture/bootstrapmonitor/cmd/main.go
+++ b/tests/fixture/bootstrapmonitor/cmd/main.go
@@ -10,7 +10,6 @@ import (
 	"time"
 
 	"github.com/spf13/cobra"
-	"go.uber.org/zap"
 
 	"github.com/ava-labs/avalanchego/tests/fixture/bootstrapmonitor"
 	"github.com/ava-labs/avalanchego/utils/logging"
@@ -31,6 +30,7 @@ func main() {
 		podName           string
 		nodeContainerName string
 		dataDir           string
+		rawLogFormat      string
 	)
 	rootCmd := &cobra.Command{
 		Use:   commandName,
@@ -40,6 +40,7 @@ func main() {
 	rootCmd.PersistentFlags().StringVar(&podName, "pod-name", os.Getenv("POD_NAME"), "The name of the pod")
 	rootCmd.PersistentFlags().StringVar(&nodeContainerName, "node-container-name", "", "The name of the node container in the pod")
 	rootCmd.PersistentFlags().StringVar(&dataDir, "data-dir", "", "The path of the data directory used for the bootstrap job")
+	rootCmd.PersistentFlags().StringVar(&rawLogFormat, "log-format", "auto", "The structure of log format. Defaults to 'auto' which formats terminal-like logs, when the output is a terminal. Otherwise, should be one of {auto, plain, colors, json}")
 
 	versionCmd := &cobra.Command{
 		Use:   "version",
@@ -55,9 +56,6 @@ func main() {
 	}
 	rootCmd.AddCommand(versionCmd)
 
-	// Use avalanchego logger for consistency
-	log := logging.NewLogger("", logging.NewWrappedCore(logging.Verbo, os.Stdout, logging.Plain.ConsoleEncoder()))
-
 	initCmd := &cobra.Command{
 		Use:   "init",
 		Short: "Initialize a new bootstrap test",
@@ -65,6 +63,10 @@ func main() {
 			if err := checkArgs(namespace, podName, nodeContainerName, dataDir); err != nil {
 				return err
 			}
+			log, err := newLogger(rawLogFormat)
+			if err != nil {
+				return err
+			}
 			return bootstrapmonitor.InitBootstrapTest(log, namespace, podName, nodeContainerName, dataDir)
 		},
 	}
@@ -87,6 +89,10 @@ func main() {
 			if imageCheckInterval <= 0 {
 				return errors.New("--image-check-interval must be greater than 0")
 			}
+			log, err := newLogger(rawLogFormat)
+			if err != nil {
+				return err
+			}
 			return bootstrapmonitor.WaitForCompletion(log, namespace, podName, nodeContainerName, dataDir, healthCheckInterval, imageCheckInterval)
 		},
 	}
@@ -95,7 +101,6 @@ func main() {
 	rootCmd.AddCommand(waitCmd)
 
 	if err := rootCmd.Execute(); err != nil {
-		log.Error(commandName+" failed", zap.Error(err))
 		os.Exit(1)
 	}
 	os.Exit(0)
@@ -116,3 +121,12 @@ func checkArgs(namespace string, podName string, nodeContainerName string, dataD
 	}
 	return nil
 }
+
+func newLogger(rawLogFormat string) (logging.Logger, error) {
+	writeCloser := os.Stdout
+	logFormat, err := logging.ToFormat(rawLogFormat, writeCloser.Fd())
+	if err != nil {
+		return nil, err
+	}
+	return logging.NewLogger("", logging.NewWrappedCore(logging.Verbo, writeCloser, logFormat.ConsoleEncoder())), nil
+}
diff --git a/tests/fixture/bootstrapmonitor/e2e/e2e_test.go b/tests/fixture/bootstrapmonitor/e2e/e2e_test.go
index 99375c7667b8..7b1fc55fda9c 100644
--- a/tests/fixture/bootstrapmonitor/e2e/e2e_test.go
+++ b/tests/fixture/bootstrapmonitor/e2e/e2e_test.go
@@ -467,6 +467,7 @@ func createBootstrapTester(tc tests.TestContext, clientset *kubernetes.Clientset
 		"init",
 		"--node-container-name=" + nodeContainerName,
 		"--data-dir=" + dataDir,
+		"--log-format=json",
 	})
 	initContainer.VolumeMounts = []corev1.VolumeMount{
 		{
@@ -481,6 +482,7 @@ func createBootstrapTester(tc tests.TestContext, clientset *kubernetes.Clientset
 		"--data-dir=" + dataDir,
 		"--health-check-interval=1s",
 		"--image-check-interval=1s",
+		"--log-format=json",
 	})
 	monitorContainer.VolumeMounts = []corev1.VolumeMount{
 		{