go-zero/zrpc/internal/rpclogger_test.go

package internal

import (
	"log"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

const content = "foo"

func TestLoggerError(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Error(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerErrorf(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Errorf(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerErrorln(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Errorln(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerFatal(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Fatal(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerFatalf(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Fatalf(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerFatalln(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Fatalln(content)
	assert.Contains(t, builder.String(), content)
}

func TestLoggerInfo(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Info(content)
	assert.Empty(t, builder.String())
}

func TestLoggerInfof(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Infof(content)
	assert.Empty(t, builder.String())
}

func TestLoggerWarning(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Warning(content)
	assert.Empty(t, builder.String())
}

func TestLoggerInfoln(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Infoln(content)
	assert.Empty(t, builder.String())
}

func TestLoggerWarningf(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Warningf(content)
	assert.Empty(t, builder.String())
}

func TestLoggerWarningln(t *testing.T) {
	var builder strings.Builder
	log.SetOutput(&builder)
	logger := new(Logger)
	logger.Warningln(content)
	assert.Empty(t, builder.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))
}