You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.9 KiB
122 lines
2.9 KiB
package log_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tendermint/tmlibs/log"
|
|
)
|
|
|
|
func TestVariousLevels(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
allowed log.Option
|
|
want string
|
|
}{
|
|
{
|
|
"AllowAll",
|
|
log.AllowAll(),
|
|
strings.Join([]string{
|
|
`{"_msg":"here","level":"debug","this is":"debug log"}`,
|
|
`{"_msg":"here","level":"info","this is":"info log"}`,
|
|
`{"_msg":"here","level":"error","this is":"error log"}`,
|
|
}, "\n"),
|
|
},
|
|
{
|
|
"AllowDebug",
|
|
log.AllowDebug(),
|
|
strings.Join([]string{
|
|
`{"_msg":"here","level":"debug","this is":"debug log"}`,
|
|
`{"_msg":"here","level":"info","this is":"info log"}`,
|
|
`{"_msg":"here","level":"error","this is":"error log"}`,
|
|
}, "\n"),
|
|
},
|
|
{
|
|
"AllowInfo",
|
|
log.AllowInfo(),
|
|
strings.Join([]string{
|
|
`{"_msg":"here","level":"info","this is":"info log"}`,
|
|
`{"_msg":"here","level":"error","this is":"error log"}`,
|
|
}, "\n"),
|
|
},
|
|
{
|
|
"AllowError",
|
|
log.AllowError(),
|
|
strings.Join([]string{
|
|
`{"_msg":"here","level":"error","this is":"error log"}`,
|
|
}, "\n"),
|
|
},
|
|
{
|
|
"AllowNone",
|
|
log.AllowNone(),
|
|
``,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
logger := log.NewFilter(log.NewTMJSONLogger(&buf), tc.allowed)
|
|
|
|
logger.Debug("here", "this is", "debug log")
|
|
logger.Info("here", "this is", "info log")
|
|
logger.Error("here", "this is", "error log")
|
|
|
|
if want, have := tc.want, strings.TrimSpace(buf.String()); want != have {
|
|
t.Errorf("\nwant:\n%s\nhave:\n%s", want, have)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestErrNotAllowed(t *testing.T) {
|
|
myError := errors.New("squelched!")
|
|
opts := []log.Option{
|
|
log.AllowError(),
|
|
log.ErrNotAllowed(myError),
|
|
}
|
|
logger := log.NewFilter(log.NewNopLogger(), opts...)
|
|
|
|
if want, have := myError, logger.Info("foo", "bar", "baz"); want != have {
|
|
t.Errorf("want %#+v, have %#+v", want, have)
|
|
}
|
|
|
|
if want, have := error(nil), logger.Error("foo", "bar", "baz"); want != have {
|
|
t.Errorf("want %#+v, have %#+v", want, have)
|
|
}
|
|
}
|
|
|
|
func TestLevelContext(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
|
|
var logger log.Logger
|
|
logger = log.NewTMJSONLogger(&buf)
|
|
logger = log.NewFilter(logger, log.AllowError())
|
|
logger = logger.With("context", "value")
|
|
|
|
logger.Error("foo", "bar", "baz")
|
|
if want, have := `{"_msg":"foo","bar":"baz","context":"value","level":"error"}`, strings.TrimSpace(buf.String()); want != have {
|
|
t.Errorf("\nwant '%s'\nhave '%s'", want, have)
|
|
}
|
|
|
|
buf.Reset()
|
|
logger.Info("foo", "bar", "baz")
|
|
if want, have := ``, strings.TrimSpace(buf.String()); want != have {
|
|
t.Errorf("\nwant '%s'\nhave '%s'", want, have)
|
|
}
|
|
}
|
|
|
|
func TestNewFilterByLevel(t *testing.T) {
|
|
assert := assert.New(t)
|
|
var logger log.Logger
|
|
logger = log.NewNopLogger()
|
|
assert.NotPanics(func() {
|
|
logger = log.NewFilterByLevel(logger, "info")
|
|
})
|
|
assert.Panics(func() {
|
|
logger = log.NewFilterByLevel(logger, "smth")
|
|
})
|
|
}
|