mirror of https://github.com/hashicorp/consul
63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
package testutil
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
)
|
|
|
|
func Logger(t testing.TB) hclog.InterceptLogger {
|
|
return LoggerWithOutput(t, NewLogBuffer(t))
|
|
}
|
|
|
|
func LoggerWithOutput(t testing.TB, output io.Writer) hclog.InterceptLogger {
|
|
return hclog.NewInterceptLogger(&hclog.LoggerOptions{
|
|
Name: t.Name(),
|
|
Level: hclog.Trace,
|
|
Output: output,
|
|
})
|
|
}
|
|
|
|
var sendTestLogsToStdout = os.Getenv("NOLOGBUFFER") == "1"
|
|
|
|
// NewLogBuffer returns an io.Writer which buffers all writes. When the test
|
|
// ends, t.Failed is checked. If the test has failed all log output is printed
|
|
// to stdout.
|
|
//
|
|
// Set the env var NOLOGBUFFER=1 to disable buffering, resulting in all log
|
|
// output being written immediately to stdout.
|
|
func NewLogBuffer(t CleanupT) io.Writer {
|
|
if sendTestLogsToStdout {
|
|
return os.Stdout
|
|
}
|
|
buf := &logBuffer{buf: new(bytes.Buffer)}
|
|
t.Cleanup(func() {
|
|
if t.Failed() {
|
|
buf.Lock()
|
|
defer buf.Unlock()
|
|
buf.buf.WriteTo(os.Stdout)
|
|
}
|
|
})
|
|
return buf
|
|
}
|
|
|
|
type CleanupT interface {
|
|
Cleanup(f func())
|
|
Failed() bool
|
|
}
|
|
|
|
type logBuffer struct {
|
|
buf *bytes.Buffer
|
|
sync.Mutex
|
|
}
|
|
|
|
func (lb *logBuffer) Write(p []byte) (n int, err error) {
|
|
lb.Lock()
|
|
defer lb.Unlock()
|
|
return lb.buf.Write(p)
|
|
}
|