35 lines
739 B
Go
35 lines
739 B
Go
|
package testutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
)
|
||
|
|
|
||
|
|
// testingWriter is a WriteSyncer that writes to the given testing.TB.
|
||
|
|
type testingWriter struct {
|
||
|
|
t *testing.T
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTestingWriter(t *testing.T) *testingWriter {
|
||
|
|
return &testingWriter{t: t}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *testingWriter) Write(p []byte) (n int, err error) {
|
||
|
|
n = len(p)
|
||
|
|
|
||
|
|
// Strip trailing newline because t.Log always adds one.
|
||
|
|
p = bytes.TrimRight(p, "\n")
|
||
|
|
|
||
|
|
// Note: t.Log is safe for concurrent use.
|
||
|
|
w.t.Logf("%s", p)
|
||
|
|
|
||
|
|
return n, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLogger(t *testing.T) zerolog.Logger {
|
||
|
|
return zerolog.New(zerolog.ConsoleWriter{Out: NewTestingWriter(t), TimeFormat: time.RFC3339Nano}).With().Timestamp().Caller().Logger().Level(zerolog.InfoLevel)
|
||
|
|
}
|