From 312b6256c3f61dc8e3e95152784a1602b5ad1234 Mon Sep 17 00:00:00 2001
From: Boming Zhang <bomingzh@sjtu.edu.cn>
Date: Sat, 19 Oct 2024 16:58:37 -0400
Subject: [PATCH] feat(cmd/joj3): debug log to stderr on test

---
 cmd/joj3/log.go       | 8 +++++++-
 cmd/joj3/main_test.go | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/cmd/joj3/log.go b/cmd/joj3/log.go
index bdeccc4..b4bb702 100644
--- a/cmd/joj3/log.go
+++ b/cmd/joj3/log.go
@@ -6,6 +6,8 @@ import (
 	"os"
 )
 
+var runningTest bool
+
 type multiHandler struct {
 	handlers []slog.Handler
 }
@@ -60,9 +62,13 @@ func setupSlog(logPath string) error {
 		})
 		handlers = append(handlers, debugHandler)
 	}
+	stderrLogLevel := slog.LevelInfo
+	if runningTest {
+		stderrLogLevel = slog.LevelDebug
+	}
 	// Stderr handler for info logs and above
 	stderrHandler := slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
-		Level: slog.LevelInfo,
+		Level: stderrLogLevel,
 	})
 	handlers = append(handlers, stderrHandler)
 	// Create a multi-handler
diff --git a/cmd/joj3/main_test.go b/cmd/joj3/main_test.go
index a2badc2..c0a2e7f 100644
--- a/cmd/joj3/main_test.go
+++ b/cmd/joj3/main_test.go
@@ -105,6 +105,7 @@ func TestRun(t *testing.T) {
 			os.Args = []string{"./joj3"}
 			outputFile := "joj3_result.json"
 			defer os.Remove(outputFile)
+			runningTest = true
 			_ = mainImpl()
 			stageResults := readStageResults(t, outputFile)
 			regex := true