30 lines
452 B
Go
30 lines
452 B
Go
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"golang.org/x/exp/slog"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ io.Writer = (*testWriter)(nil)
|
||
|
|
|
||
|
|
func NewWriter(tb testing.TB) io.Writer {
|
||
|
|
tb.Helper()
|
||
|
|
return testWriter{tb: tb}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLogger(tb testing.TB) *slog.Logger {
|
||
|
|
tb.Helper()
|
||
|
|
return slog.New(slog.NewTextHandler(NewWriter(tb)))
|
||
|
|
}
|
||
|
|
|
||
|
|
type testWriter struct {
|
||
|
|
tb testing.TB
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t testWriter) Write(p []byte) (n int, err error) {
|
||
|
|
t.tb.Log(string(p))
|
||
|
|
return len(p), nil
|
||
|
|
}
|