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.
110 lines
2.5 KiB
110 lines
2.5 KiB
package log_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/ioutil"
|
|
"math"
|
|
"regexp"
|
|
"testing"
|
|
|
|
kitlog "github.com/go-kit/kit/log"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tendermint/tmlibs/log"
|
|
)
|
|
|
|
func TestTMFmtLogger(t *testing.T) {
|
|
t.Parallel()
|
|
buf := &bytes.Buffer{}
|
|
logger := log.NewTMFmtLogger(buf)
|
|
|
|
if err := logger.Log("hello", "world"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Regexp(t, regexp.MustCompile(`N\[.+\] unknown \s+ hello=world\n$`), buf.String())
|
|
|
|
buf.Reset()
|
|
if err := logger.Log("a", 1, "err", errors.New("error")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Regexp(t, regexp.MustCompile(`N\[.+\] unknown \s+ a=1 err=error\n$`), buf.String())
|
|
|
|
buf.Reset()
|
|
err := logger.Log("std_map", map[int]int{1: 2}, "my_map", mymap{0: 0})
|
|
assert.NotNil(t, err)
|
|
|
|
buf.Reset()
|
|
if err := logger.Log("level", "error"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Regexp(t, regexp.MustCompile(`E\[.+\] unknown \s+\n$`), buf.String())
|
|
|
|
buf.Reset()
|
|
if err := logger.Log("_msg", "Hello"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Regexp(t, regexp.MustCompile(`N\[.+\] Hello \s+\n$`), buf.String())
|
|
}
|
|
|
|
func BenchmarkTMFmtLoggerSimple(b *testing.B) {
|
|
benchmarkRunnerKitlog(b, log.NewTMFmtLogger(ioutil.Discard), baseMessage)
|
|
}
|
|
|
|
func BenchmarkTMFmtLoggerContextual(b *testing.B) {
|
|
benchmarkRunnerKitlog(b, log.NewTMFmtLogger(ioutil.Discard), withMessage)
|
|
}
|
|
|
|
func TestTMFmtLoggerConcurrency(t *testing.T) {
|
|
t.Parallel()
|
|
testConcurrency(t, log.NewTMFmtLogger(ioutil.Discard), 10000)
|
|
}
|
|
|
|
func benchmarkRunnerKitlog(b *testing.B, logger kitlog.Logger, f func(kitlog.Logger)) {
|
|
lc := kitlog.With(logger, "common_key", "common_value")
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
f(lc)
|
|
}
|
|
}
|
|
|
|
var (
|
|
baseMessage = func(logger kitlog.Logger) { logger.Log("foo_key", "foo_value") }
|
|
withMessage = func(logger kitlog.Logger) { kitlog.With(logger, "a", "b").Log("d", "f") }
|
|
)
|
|
|
|
// These test are designed to be run with the race detector.
|
|
|
|
func testConcurrency(t *testing.T, logger kitlog.Logger, total int) {
|
|
n := int(math.Sqrt(float64(total)))
|
|
share := total / n
|
|
|
|
errC := make(chan error, n)
|
|
|
|
for i := 0; i < n; i++ {
|
|
go func() {
|
|
errC <- spam(logger, share)
|
|
}()
|
|
}
|
|
|
|
for i := 0; i < n; i++ {
|
|
err := <-errC
|
|
if err != nil {
|
|
t.Fatalf("concurrent logging error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func spam(logger kitlog.Logger, count int) error {
|
|
for i := 0; i < count; i++ {
|
|
err := logger.Log("key", i)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type mymap map[int]int
|
|
|
|
func (m mymap) String() string { return "special_behavior" }
|