# HG changeset patch
# User Christophe de Vienne <christophe@cdevienne.info>
# Date 1604525989 -3600
#      Wed Nov 04 22:39:49 2020 +0100
# Node ID 99ba412ea469e5e10d18e4fbee42ccfe303e8572
# Parent  e09c0f5363721c4c0b7c7217b4d3391b480ddbdc
TestLogger: add a SetTB function that (temporarily) changes the current TB

diff --git a/testutils/logger.go b/testutils/logger.go
--- 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()