You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.7 KiB
66 lines
1.7 KiB
package log
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/go-kit/kit/log/term"
|
|
)
|
|
|
|
var (
|
|
// reuse the same logger across all tests
|
|
_testingLoggerMutex = sync.Mutex{}
|
|
_testingLogger Logger
|
|
)
|
|
|
|
// TestingLogger returns a TMLogger which writes to STDOUT if testing being run
|
|
// with the verbose (-v) flag, NopLogger otherwise.
|
|
//
|
|
// Note that the call to TestingLogger() must be made
|
|
// inside a test (not in the init func) because
|
|
// verbose flag only set at the time of testing.
|
|
func TestingLogger() Logger {
|
|
return TestingLoggerWithOutput(os.Stdout)
|
|
}
|
|
|
|
// TestingLoggerWOutput returns a TMLogger which writes to (w io.Writer) if testing being run
|
|
// with the verbose (-v) flag, NopLogger otherwise.
|
|
//
|
|
// Note that the call to TestingLoggerWithOutput(w io.Writer) must be made
|
|
// inside a test (not in the init func) because
|
|
// verbose flag only set at the time of testing.
|
|
func TestingLoggerWithOutput(w io.Writer) Logger {
|
|
_testingLoggerMutex.Lock()
|
|
defer _testingLoggerMutex.Unlock()
|
|
if _testingLogger != nil {
|
|
return _testingLogger
|
|
}
|
|
|
|
if testing.Verbose() {
|
|
_testingLogger = NewTMLogger(NewSyncWriter(w))
|
|
} else {
|
|
_testingLogger = NewNopLogger()
|
|
}
|
|
|
|
return _testingLogger
|
|
}
|
|
|
|
// TestingLoggerWithColorFn allow you to provide your own color function. See
|
|
// TestingLogger for documentation.
|
|
func TestingLoggerWithColorFn(colorFn func(keyvals ...interface{}) term.FgBgColor) Logger {
|
|
_testingLoggerMutex.Lock()
|
|
defer _testingLoggerMutex.Unlock()
|
|
if _testingLogger != nil {
|
|
return _testingLogger
|
|
}
|
|
|
|
if testing.Verbose() {
|
|
_testingLogger = NewTMLoggerWithColorFn(NewSyncWriter(os.Stdout), colorFn)
|
|
} else {
|
|
_testingLogger = NewNopLogger()
|
|
}
|
|
|
|
return _testingLogger
|
|
}
|