You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
140 lines
2.5 KiB
Go
140 lines
2.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
const content = "foo"
|
|
|
|
func TestLoggerError(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Error(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerErrorf(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Errorf(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerErrorln(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Errorln(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerFatal(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Fatal(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerFatalf(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Fatalf(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerFatalln(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Fatalln(content)
|
|
assert.Contains(t, w.String(), content)
|
|
}
|
|
|
|
func TestLoggerInfo(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Info(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLoggerInfof(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Infof(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLoggerWarning(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Warning(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLoggerInfoln(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Infoln(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLoggerWarningf(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Warningf(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLoggerWarningln(t *testing.T) {
|
|
w, restore := injectLog()
|
|
defer restore()
|
|
|
|
logger := new(Logger)
|
|
logger.Warningln(content)
|
|
assert.Empty(t, w.String())
|
|
}
|
|
|
|
func TestLogger_V(t *testing.T) {
|
|
logger := new(Logger)
|
|
// grpclog.fatalLog
|
|
assert.True(t, logger.V(3))
|
|
// grpclog.infoLog
|
|
assert.False(t, logger.V(0))
|
|
}
|
|
|
|
func injectLog() (r *strings.Builder, restore func()) {
|
|
var buf strings.Builder
|
|
w := logx.NewWriter(&buf)
|
|
o := logx.Reset()
|
|
logx.SetWriter(w)
|
|
|
|
return &buf, func() {
|
|
logx.Reset()
|
|
logx.SetWriter(o)
|
|
}
|
|
}
|