forked from mirror/logrus
111 lines
2.1 KiB
Go
111 lines
2.1 KiB
Go
package logrus
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func LogAndAssertJSON(t *testing.T, log func(*Logger), assertions func(fields Fields)) {
|
|
var buffer bytes.Buffer
|
|
var fields Fields
|
|
|
|
logger := New()
|
|
logger.Out = &buffer
|
|
logger.Formatter = new(JSONFormatter)
|
|
|
|
log(logger)
|
|
|
|
err := json.Unmarshal(buffer.Bytes(), &fields)
|
|
assert.Nil(t, err)
|
|
|
|
assertions(fields)
|
|
}
|
|
|
|
func TestPrint(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Print("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "info")
|
|
})
|
|
}
|
|
|
|
func TestInfo(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Info("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "info")
|
|
})
|
|
}
|
|
|
|
func TestWarn(t *testing.T) {
|
|
LogAndAssertJSON(t, func(log *Logger) {
|
|
log.Warn("test")
|
|
}, func(fields Fields) {
|
|
assert.Equal(t, fields["msg"], "test")
|
|
assert.Equal(t, fields["level"], "warning")
|
|
})
|
|
}
|
|
|
|
type SlowString string
|
|
|
|
func (s SlowString) String() string {
|
|
time.Sleep(time.Millisecond)
|
|
return string(s)
|
|
}
|
|
|
|
func getLogAtLevel(l Level) *Logger {
|
|
log := New()
|
|
log.Level = l
|
|
log.Out = ioutil.Discard
|
|
return log
|
|
}
|
|
|
|
func BenchmarkLevelDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Info(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debug(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelfDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Infof("%s", SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevelfHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debugf("%s", SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevellnDisplayed(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Infoln(SlowString("foo"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkLevellnHidden(b *testing.B) {
|
|
log := getLogAtLevel(Info)
|
|
for i := 0; i < b.N; i++ {
|
|
log.Debugln(SlowString("foo"))
|
|
}
|
|
}
|