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  }