mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-23 09:00:20 +08:00
ac321fc146
* backup * simplify * chore: remove unused pool * chore: fix lint errors * chore: use strings.Builder instead of bytes.Buffer * test: add more tests * chore: fix reviewdog * test: fix data race * feat: make logger customizable * chore: fix reviewdog * test: fix fails * chore: fix set writer twice * chore: use context instead of golang.org context * chore: specify uint32 for level types
35 lines
578 B
Go
35 lines
578 B
Go
package logx
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLessLogger_Error(t *testing.T) {
|
|
w := new(mockWriter)
|
|
old := writer.Swap(w)
|
|
defer writer.Store(old)
|
|
|
|
l := NewLessLogger(500)
|
|
for i := 0; i < 100; i++ {
|
|
l.Error("hello")
|
|
}
|
|
|
|
assert.Equal(t, 1, strings.Count(w.String(), "\n"))
|
|
}
|
|
|
|
func TestLessLogger_Errorf(t *testing.T) {
|
|
w := new(mockWriter)
|
|
old := writer.Swap(w)
|
|
defer writer.Store(old)
|
|
|
|
l := NewLessLogger(500)
|
|
for i := 0; i < 100; i++ {
|
|
l.Errorf("hello")
|
|
}
|
|
|
|
assert.Equal(t, 1, strings.Count(w.String(), "\n"))
|
|
}
|