test: Reset logging to global defaults on cleanup.

This commit is contained in:
Joachim Bauch 2024-05-14 16:52:46 +02:00
parent 4603b2b290
commit 94a8f0f02b
No known key found for this signature in database
GPG key ID: 77C1D22D53E15F02

View file

@ -22,10 +22,21 @@
package signaling package signaling
import ( import (
"io"
"log" "log"
"testing" "testing"
) )
var (
prevWriter io.Writer
prevFlags int
)
func init() {
prevWriter = log.Writer()
prevFlags = log.Flags()
}
type testLogWriter struct { type testLogWriter struct {
t testing.TB t testing.TB
} }
@ -37,8 +48,6 @@ func (w *testLogWriter) Write(b []byte) (int, error) {
} }
func CatchLogForTest(t testing.TB) { func CatchLogForTest(t testing.TB) {
prevWriter := log.Writer()
prevFlags := log.Flags()
t.Cleanup(func() { t.Cleanup(func() {
log.SetOutput(prevWriter) log.SetOutput(prevWriter)
log.SetFlags(prevFlags) log.SetFlags(prevFlags)