github.com/secure-build/gitlab-runner@v12.5.0+incompatible/log/test/hook_test.go (about) 1 package test 2 3 import ( 4 "testing" 5 6 "github.com/sirupsen/logrus" 7 "github.com/stretchr/testify/assert" 8 ) 9 10 func TestNewHook(t *testing.T) { 11 beforeCount := countHooks() 12 13 _, cleanup := NewHook() 14 afterCount := countHooks() 15 16 cleanup() 17 18 assert.True(t, afterCount > beforeCount) 19 assert.Equal(t, beforeCount, countHooks()) 20 } 21 22 func countHooks() int { 23 count := 0 24 for _, levels := range logrus.StandardLogger().Hooks { 25 for range levels { 26 count++ 27 } 28 } 29 30 return count 31 }