diff --git a/README.md b/README.md index b3c75cf..46dda85 100644 --- a/README.md +++ b/README.md @@ -440,21 +440,23 @@ Logrus has a built in facility for asserting the presence of log messages. This * a test logger (`test.NewNullLogger`) that just records log messages (and does not output any): ```go -import ( - "testing" +import( + "github.com/sirupsen/logrus" "github.com/sirupsen/logrus/hooks/null" + "github.com/stretchr/testify/assert" + "testing" ) -func TestSomething(t *testing.T) { +func TestSomething(t*testing.T){ logger, hook := null.NewNullLogger() - logger.Error("Hello error") + logger.Error("Helloerror") - assert.Equal(1, len(hook.Entries)) - assert.Equal(logrus.ErrorLevel, hook.LastEntry().Level) - assert.Equal("Hello error", hook.LastEntry().Message) + assert.Equal(t, 1, len(hook.Entries)) + assert.Equal(t, logrus.ErrorLevel, hook.LastEntry().Level) + assert.Equal(t, "Helloerror", hook.LastEntry().Message) hook.Reset() - assert.Nil(hook.LastEntry()) + assert.Nil(t, hook.LastEntry()) } ``` diff --git a/hooks/null/null_test.go b/hooks/null/null_test.go new file mode 100644 index 0000000..10e8413 --- /dev/null +++ b/hooks/null/null_test.go @@ -0,0 +1,19 @@ +package null + +import ( + "github.com/sirupsen/logrus" + "github.com/stretchr/testify/assert" + "testing" +) + +func TestNullLogger(t *testing.T) { + logger, hook := NewNullLogger() + logger.Error("Helloerror") + + assert.Equal(t, 1, len(hook.Entries)) + assert.Equal(t, logrus.ErrorLevel, hook.LastEntry().Level) + assert.Equal(t, "Helloerror", hook.LastEntry().Message) + + hook.Reset() + assert.Nil(t, hook.LastEntry()) +}