github.com/shuguocloud/go-zero@v1.3.0/core/logx/lesslogger_test.go (about)

     1  package logx
     2  
     3  import (
     4  	"log"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestLessLogger_Error(t *testing.T) {
    12  	var builder strings.Builder
    13  	log.SetOutput(&builder)
    14  	l := NewLessLogger(500)
    15  	for i := 0; i < 100; i++ {
    16  		l.Error("hello")
    17  	}
    18  
    19  	assert.Equal(t, 1, strings.Count(builder.String(), "\n"))
    20  }
    21  
    22  func TestLessLogger_Errorf(t *testing.T) {
    23  	var builder strings.Builder
    24  	log.SetOutput(&builder)
    25  	l := NewLessLogger(500)
    26  	for i := 0; i < 100; i++ {
    27  		l.Errorf("hello")
    28  	}
    29  
    30  	assert.Equal(t, 1, strings.Count(builder.String(), "\n"))
    31  }