diff --git a/processor/test_helper.go b/processor/test_helper.go index e708343..ec09c11 100644 --- a/processor/test_helper.go +++ b/processor/test_helper.go @@ -1,22 +1,26 @@ package processor import ( - "io/ioutil" + "io" "modify/logger" "os" ) func init() { - // Initialize logger with ERROR level for tests - // to minimize noise in test output - logger.Init(logger.LevelError) + // Only modify logger in test mode + // This checks if we're running under 'go test' + if os.Getenv("GO_TESTING") == "1" || os.Getenv("TESTING") == "1" { + // Initialize logger with ERROR level for tests + // to minimize noise in test output + logger.Init(logger.LevelError) - // Optionally redirect logger output to discard - // This prevents logger output from interfering with test output - disableTestLogs := os.Getenv("ENABLE_TEST_LOGS") != "1" - if disableTestLogs { - // Create a new logger that writes to nowhere - silentLogger := logger.New(ioutil.Discard, "", 0) - logger.DefaultLogger = silentLogger + // Optionally redirect logger output to discard + // This prevents logger output from interfering with test output + disableTestLogs := os.Getenv("ENABLE_TEST_LOGS") != "1" + if disableTestLogs { + // Create a new logger that writes to nowhere + silentLogger := logger.New(io.Discard, "", 0) + logger.DefaultLogger = silentLogger + } } }