diff --git a/testutils/logger.go b/testutils/logger.go index e09c0f5363721c4c0b7c7217b4d3391b480ddbdc_dGVzdHV0aWxzL2xvZ2dlci5nbw==..99ba412ea469e5e10d18e4fbee42ccfe303e8572_dGVzdHV0aWxzL2xvZ2dlci5nbw== 100644 --- a/testutils/logger.go +++ b/testutils/logger.go @@ -34,6 +34,16 @@ return len(m), nil } +// SetTB changes the current TB, and returns a function to get back to the +// previous one +func (tl *TestLogger) SetTB(tb testing.TB) func() { + otb := tl.tb + tl.tb = tb + return func() { + tl.tb = otb + } +} + // GetLogger returns a test Logger func GetLogger(tb testing.TB) zerolog.Logger { return NewTestLogger(tb).Logger()