b418daed4d
Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Julien Robert <julien@rbrt.fr>
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package log_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
"gotest.tools/v3/assert"
|
|
|
|
"cosmossdk.io/log"
|
|
)
|
|
|
|
func TestLoggerOptionStackTrace(t *testing.T) {
|
|
buf := new(bytes.Buffer)
|
|
logger := log.NewLogger(buf, log.TraceOption(true), log.ColorOption(false))
|
|
logger.Error("this log should be displayed", "error", inner())
|
|
if strings.Count(buf.String(), "logger_test.go") != 1 {
|
|
t.Fatalf("stack trace not found, got: %s", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
logger = log.NewLogger(buf, log.TraceOption(false), log.ColorOption(false))
|
|
logger.Error("this log should be displayed", "error", inner())
|
|
if strings.Count(buf.String(), "logger_test.go") > 0 {
|
|
t.Fatalf("stack trace found, got: %s", buf.String())
|
|
}
|
|
}
|
|
|
|
func inner() error {
|
|
return errors.New("seems we have an error here")
|
|
}
|
|
|
|
type _MockHook string
|
|
|
|
func (h _MockHook) Run(e *zerolog.Event, l zerolog.Level, msg string) {
|
|
e.Bool(string(h), true)
|
|
}
|
|
|
|
func TestLoggerOptionHooks(t *testing.T) {
|
|
buf := new(bytes.Buffer)
|
|
var (
|
|
mockHook1 _MockHook = "mock_message1"
|
|
mockHook2 _MockHook = "mock_message2"
|
|
)
|
|
logger := log.NewLogger(buf, log.HooksOption(mockHook1, mockHook2), log.ColorOption(false))
|
|
logger.Info("hello world")
|
|
assert.Assert(t, strings.Contains(buf.String(), "mock_message1=true"))
|
|
assert.Assert(t, strings.Contains(buf.String(), "mock_message2=true"))
|
|
|
|
buf.Reset()
|
|
logger = log.NewLogger(buf, log.HooksOption(), log.ColorOption(false))
|
|
logger.Info("hello world")
|
|
assert.Assert(t, strings.Contains(buf.String(), "hello world"))
|
|
}
|