mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 02:46:30 +02:00
23 lines
615 B
Go
23 lines
615 B
Go
package testutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// SetLogger sets the given logger as the global logger for the remainder of
|
|
// the current test. Because the logger is global, this must not be called from
|
|
// parallel tests.
|
|
func SetLogger(t *testing.T, logger zerolog.Logger) {
|
|
t.Helper()
|
|
|
|
originalLogger := log.Logger
|
|
t.Cleanup(func() { log.Logger = originalLogger })
|
|
log.Logger = logger
|
|
|
|
originalContextLogger := zerolog.DefaultContextLogger
|
|
t.Cleanup(func() { zerolog.DefaultContextLogger = originalContextLogger })
|
|
zerolog.DefaultContextLogger = &logger
|
|
}
|