Fix data race in hooks.test package

This commit is contained in:
Francois 2022-12-22 10:54:08 +01:00
parent f8bf7650dc
commit ff07b25fdf
2 changed files with 20 additions and 1 deletions

View File

@ -32,7 +32,7 @@ func NewGlobal() *Hook {
func NewLocal(logger *logrus.Logger) *Hook {
hook := new(Hook)
logger.Hooks.Add(hook)
logger.AddHook(hook)
return hook

View File

@ -83,3 +83,22 @@ func TestFatalWithAlternateExit(t *testing.T) {
assert.Equal("something went very wrong", hook.LastEntry().Message)
assert.Equal(1, len(hook.Entries))
}
func TestNewLocal(t *testing.T) {
assert := assert.New(t)
logger := logrus.New()
var wg sync.WaitGroup
defer wg.Wait()
wg.Add(10)
for i := 0; i < 10; i++ {
go func(i int) {
logger.Info("info")
wg.Done()
}(i)
}
hook := NewLocal(logger)
assert.NotNil(hook)
}