github.com/shuguocloud/go-zero@v1.3.0/zrpc/internal/rpclogger_test.go (about)

     1  package internal
     2  
     3  import (
     4  	"log"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  const content = "foo"
    12  
    13  func TestLoggerError(t *testing.T) {
    14  	var builder strings.Builder
    15  	log.SetOutput(&builder)
    16  	logger := new(Logger)
    17  	logger.Error(content)
    18  	assert.Contains(t, builder.String(), content)
    19  }
    20  
    21  func TestLoggerErrorf(t *testing.T) {
    22  	var builder strings.Builder
    23  	log.SetOutput(&builder)
    24  	logger := new(Logger)
    25  	logger.Errorf(content)
    26  	assert.Contains(t, builder.String(), content)
    27  }
    28  
    29  func TestLoggerErrorln(t *testing.T) {
    30  	var builder strings.Builder
    31  	log.SetOutput(&builder)
    32  	logger := new(Logger)
    33  	logger.Errorln(content)
    34  	assert.Contains(t, builder.String(), content)
    35  }
    36  
    37  func TestLoggerFatal(t *testing.T) {
    38  	var builder strings.Builder
    39  	log.SetOutput(&builder)
    40  	logger := new(Logger)
    41  	logger.Fatal(content)
    42  	assert.Contains(t, builder.String(), content)
    43  }
    44  
    45  func TestLoggerFatalf(t *testing.T) {
    46  	var builder strings.Builder
    47  	log.SetOutput(&builder)
    48  	logger := new(Logger)
    49  	logger.Fatalf(content)
    50  	assert.Contains(t, builder.String(), content)
    51  }
    52  
    53  func TestLoggerFatalln(t *testing.T) {
    54  	var builder strings.Builder
    55  	log.SetOutput(&builder)
    56  	logger := new(Logger)
    57  	logger.Fatalln(content)
    58  	assert.Contains(t, builder.String(), content)
    59  }
    60  
    61  func TestLoggerInfo(t *testing.T) {
    62  	var builder strings.Builder
    63  	log.SetOutput(&builder)
    64  	logger := new(Logger)
    65  	logger.Info(content)
    66  	assert.Empty(t, builder.String())
    67  }
    68  
    69  func TestLoggerInfof(t *testing.T) {
    70  	var builder strings.Builder
    71  	log.SetOutput(&builder)
    72  	logger := new(Logger)
    73  	logger.Infof(content)
    74  	assert.Empty(t, builder.String())
    75  }
    76  
    77  func TestLoggerWarning(t *testing.T) {
    78  	var builder strings.Builder
    79  	log.SetOutput(&builder)
    80  	logger := new(Logger)
    81  	logger.Warning(content)
    82  	assert.Empty(t, builder.String())
    83  }
    84  
    85  func TestLoggerInfoln(t *testing.T) {
    86  	var builder strings.Builder
    87  	log.SetOutput(&builder)
    88  	logger := new(Logger)
    89  	logger.Infoln(content)
    90  	assert.Empty(t, builder.String())
    91  }
    92  
    93  func TestLoggerWarningf(t *testing.T) {
    94  	var builder strings.Builder
    95  	log.SetOutput(&builder)
    96  	logger := new(Logger)
    97  	logger.Warningf(content)
    98  	assert.Empty(t, builder.String())
    99  }
   100  
   101  func TestLoggerWarningln(t *testing.T) {
   102  	var builder strings.Builder
   103  	log.SetOutput(&builder)
   104  	logger := new(Logger)
   105  	logger.Warningln(content)
   106  	assert.Empty(t, builder.String())
   107  }
   108  
   109  func TestLogger_V(t *testing.T) {
   110  	logger := new(Logger)
   111  	// grpclog.fatalLog
   112  	assert.True(t, logger.V(3))
   113  	// grpclog.infoLog
   114  	assert.False(t, logger.V(0))
   115  }