Improve tests for logger.*Fn functions

This commit is contained in:
David Bariod 2020-05-19 17:02:33 +02:00
parent d7edea4451
commit ba4da53cff
1 changed files with 11 additions and 6 deletions

View File

@ -1,26 +1,31 @@
package logrus_test
import (
"fmt"
log "github.com/sirupsen/logrus"
"testing"
log "github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
)
func TestLogger_LogFn(t *testing.T) {
log.SetFormatter(&log.JSONFormatter{})
log.SetLevel(log.WarnLevel)
notCalled := 0
log.InfoFn(func() []interface{} {
fmt.Println("This is never run")
notCalled++
return []interface{}{
"Hello",
}
})
assert.Equal(t, 0, notCalled)
called := 0
log.ErrorFn(func() []interface{} {
fmt.Println("This runs")
called++
return []interface{}{
"Oopsi",
}
})
assert.Equal(t, 1, called)
}