diff options
Diffstat (limited to 'internal/testhelper')
| -rw-r--r-- | internal/testhelper/testhelper.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/internal/testhelper/testhelper.go b/internal/testhelper/testhelper.go index a925c79..da781ce 100644 --- a/internal/testhelper/testhelper.go +++ b/internal/testhelper/testhelper.go @@ -8,6 +8,8 @@ import ( "runtime" "github.com/otiai10/copy" + "github.com/sirupsen/logrus" + "github.com/sirupsen/logrus/hooks/test" ) var ( @@ -91,3 +93,10 @@ func Setenv(key, value string) (func(), error) { err := os.Setenv(key, value) return func() { os.Setenv(key, oldValue) }, err } + +func SetupLogger() *test.Hook { + logger, hook := test.NewNullLogger() + logrus.SetOutput(logger.Writer()) + + return hook +} |