func TestDefaultFormatter(t *testing.T) {
var (
out, errOut = bytes.NewBuffer([]byte{}), bytes.NewBuffer([]byte{})
logger = logging.New(out, errOut, nil)
now time.Time
actual []byte
expected string
err error
)
// Test logger.Info
now = time.Now()
logger.Info("Test logger.Info")
actual, err = ioutil.ReadAll(out)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"INFO: %s default_formatter_test.go:27 Test logger.Info\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Infof
now = time.Now()
logger.Infof("Test %s.%s", "logger", "Infof")
actual, err = ioutil.ReadAll(out)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"INFO: %s default_formatter_test.go:40 Test logger.Infof\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Warning
now = time.Now()
logger.Warning("Test logger.Warning")
actual, err = ioutil.ReadAll(out)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"WARNING: %s default_formatter_test.go:53 Test logger.Warning\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Warningf
now = time.Now()
logger.Warningf("Test %s.%s", "logger", "Warningf")
actual, err = ioutil.ReadAll(out)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"WARNING: %s default_formatter_test.go:66 Test logger.Warningf\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Error
now = time.Now()
logger.Error("Test logger.Error")
actual, err = ioutil.ReadAll(errOut)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"ERROR: %s default_formatter_test.go:79 Test logger.Error\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Errorf
now = time.Now()
logger.Errorf("Test %s.%s", "logger", "Errorf")
actual, err = ioutil.ReadAll(errOut)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"ERROR: %s default_formatter_test.go:92 Test logger.Errorf\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Fatal
now = time.Now()
logger.Fatal("Test logger.Fatal")
actual, err = ioutil.ReadAll(errOut)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"FATAL: %s default_formatter_test.go:105 Test logger.Fatal\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
// Test logger.Fatalf
now = time.Now()
logger.Fatalf("Test %s.%s", "logger", "Fatalf")
actual, err = ioutil.ReadAll(errOut)
if err != nil {
log.Fatal(err)
}
expected = fmt.Sprintf(
"FATAL: %s default_formatter_test.go:118 Test logger.Fatalf\n",
now.Format("2006/01/02 15:04:05"),
)
assert.Equal(t, expected, string(actual))
}
func init() {
logger = logging.New(nil, nil, new(logging.ColouredFormatter))
}
func init() {
plainLogger = logging.New(nil, nil, nil)
colouredLogger = logging.New(nil, nil, new(logging.ColouredFormatter))
}
