logger/test_output.go
package logger
import (
"testing"
"github.com/ethereum/go-ethereum/log"
)
// SetTestMode sets test mode.
func SetTestMode(t testing.TB) {
log.Root().SetHandler(
log.CallerStackHandler("%v", TestHandler(t, log.LogfmtFormat())))
}
// TestHandler writes into test log.
func TestHandler(t testing.TB, fmtr log.Format) log.Handler {
return log.FuncHandler(func(r *log.Record) error {
t.Log(string(fmtr.Format(r)))
return nil
})
}