diff options
author | Jacob Vosmaer <jacob@gitlab.com> | 2019-10-22 14:00:48 +0300 |
---|---|---|
committer | Jacob Vosmaer <jacob@gitlab.com> | 2019-10-22 14:00:48 +0300 |
commit | 50c9942e60aa02dbc708be09588b56b59cc7c0d3 (patch) | |
tree | 1cd69a9fe91e9d2197d734f2cb763081fe346027 | |
parent | b69f0b8837eca8ebe716ef03338292a185b0ce65 (diff) | |
parent | d2504f0229c064489b7ca2b321796c8873295ef5 (diff) |
Merge branch 'jc-remove-test-hooks' into 'master'
Remove test hooks
Closes #2096
See merge request gitlab-org/gitaly!1567
-rw-r--r-- | internal/testhelper/test_hook.go | 40 |
1 files changed, 1 insertions, 39 deletions
diff --git a/internal/testhelper/test_hook.go b/internal/testhelper/test_hook.go index 93d229a2a..448dd2c5b 100644 --- a/internal/testhelper/test_hook.go +++ b/internal/testhelper/test_hook.go @@ -7,48 +7,10 @@ import ( log "github.com/sirupsen/logrus" ) -type testHook struct { - t testing.TB - formatter log.Formatter -} - -func (s testHook) Levels() []log.Level { - return []log.Level{ - log.DebugLevel, - log.InfoLevel, - log.WarnLevel, - log.ErrorLevel, - log.FatalLevel, - log.PanicLevel, - } -} - -func (s testHook) Fire(entry *log.Entry) error { - formatted, err := s.formatter.Format(entry) - if err != nil { - return err - } - - formattedString := string(formatted) - - switch entry.Level { - case log.FatalLevel, log.PanicLevel: - s.t.Fatal(formattedString) - - default: - s.t.Log(formattedString) - } - - return nil -} - -// NewTestLogger created a logrus hook which can be used with testing logs +// NewTestLogger created a logrus hook func NewTestLogger(tb testing.TB) *log.Logger { logger := log.New() logger.Out = ioutil.Discard - formatter := &log.TextFormatter{} - - logger.Hooks.Add(testHook{tb, formatter}) return logger } |