forked from mirror/logrus
38 lines
748 B
Go
38 lines
748 B
Go
|
package writer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDifferentLevelsGoToDifferentWriters(t *testing.T) {
|
||
|
|
var a, b bytes.Buffer
|
||
|
|
|
||
|
|
log.SetFormatter(&log.TextFormatter{
|
||
|
|
DisableTimestamp: true,
|
||
|
|
})
|
||
|
|
log.SetOutput(ioutil.Discard) // Send all logs to nowhere by default
|
||
|
|
|
||
|
|
log.AddHook(&Hook{
|
||
|
|
Writer: &a,
|
||
|
|
LogLevels: []log.Level{
|
||
|
|
log.WarnLevel,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
log.AddHook(&Hook{ // Send info and debug logs to stdout
|
||
|
|
Writer: &b,
|
||
|
|
LogLevels: []log.Level{
|
||
|
|
log.InfoLevel,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
log.Warn("send to a")
|
||
|
|
log.Info("send to b")
|
||
|
|
|
||
|
|
assert.Equal(t, a.String(), "level=warning msg=\"send to a\"\n")
|
||
|
|
assert.Equal(t, b.String(), "level=info msg=\"send to b\"\n")
|
||
|
|
}
|