/
log.go
46 lines (39 loc) · 1.08 KB
/
log.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package logtest
import (
"os"
"testing"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"go.uber.org/zap/zaptest"
"github.com/spacemeshos/go-spacemesh/log"
)
const testLogLevel = "TEST_LOG_LEVEL"
func Zap(tb testing.TB, override ...zapcore.Level) *zap.Logger {
return New(tb, override...).Zap()
}
// New creates log.Log instance that will use testing.TB.Log internally.
func New(tb testing.TB, override ...zapcore.Level) log.Log {
var level zapcore.Level
if len(override) > 0 {
level = override[0]
} else {
lvl := os.Getenv(testLogLevel)
if len(lvl) == 0 {
return log.NewNop()
}
if err := level.Set(lvl); err != nil {
panic(err)
}
}
return log.NewFromLog(zaptest.NewLogger(tb, zaptest.Level(level))).Named(tb.Name())
}
// SetupGlobal updates AppLog to the instance of test-specific logger.
// Some tests do not terminate all goroutines when they exit, so we can't recover original for now.
func SetupGlobal(tb testing.TB, overwrite ...zapcore.Level) {
tlog := New(tb, overwrite...)
tmp := log.GetLogger()
log.SetupGlobal(tlog)
tb.Cleanup(func() {
log.SetLogger(tmp)
})
}