forked from mirror/logrus
42 lines
847 B
Go
42 lines
847 B
Go
|
package logrus_caller
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func LogAndAssertJSON(t *testing.T, log func(*logrus.Logger), assertions func(fields logrus.Fields)) {
|
||
|
|
var buffer bytes.Buffer
|
||
|
|
var fields logrus.Fields
|
||
|
|
|
||
|
|
logger := logrus.New()
|
||
|
|
logger.Hooks.Add(&CallerHook{})
|
||
|
|
logger.Out = &buffer
|
||
|
|
logger.Formatter = new(logrus.JSONFormatter)
|
||
|
|
|
||
|
|
log(logger)
|
||
|
|
|
||
|
|
err := json.Unmarshal(buffer.Bytes(), &fields)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("Error unmarshaling log entry")
|
||
|
|
}
|
||
|
|
|
||
|
|
assertions(fields)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCaller(t *testing.T) {
|
||
|
|
LogAndAssertJSON(t, func(logger *logrus.Logger) {
|
||
|
|
logger.Info("Hello World")
|
||
|
|
}, func(fields logrus.Fields) {
|
||
|
|
expected := "caller_test.go:33"
|
||
|
|
|
||
|
|
if fields["caller"] != expected {
|
||
|
|
t.Error(fmt.Sprintf("Caller was %s, expected %s", fields["caller"], expected))
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|