liuxiaolong
2020-09-16 4501e38ca66f09b35aaaf43fa5a316554930fcf4
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
package testutil
 
import (
    "io"
    "log"
    "strings"
    "testing"
)
 
func TestLogger(t testing.TB) *log.Logger {
    return log.New(&testWriter{t}, "test: ", log.LstdFlags)
}
 
func TestLoggerWithName(t testing.TB, name string) *log.Logger {
    return log.New(&testWriter{t}, "test["+name+"]: ", log.LstdFlags)
}
 
func TestWriter(t testing.TB) io.Writer {
    return &testWriter{t}
}
 
type testWriter struct {
    t testing.TB
}
 
func (tw *testWriter) Write(p []byte) (n int, err error) {
    tw.t.Helper()
    tw.t.Log(strings.TrimSpace(string(p)))
    return len(p), nil
}