TestLogger: add a SetTB function that (temporarily) changes the current TB

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()