github.com/alibabacloud-go/tea@v1.3.10/utils/logger_test.go (about)

     1  package utils
     2  
     3  import (
     4  	"bytes"
     5  	"errors"
     6  	"testing"
     7  )
     8  
     9  func Test_Param(t *testing.T) {
    10  	log := new(Logger)
    11  	log.SetFormatTemplate("template")
    12  	temp := log.GetFormatTemplate()
    13  	AssertEqual(t, "template", temp)
    14  
    15  	log.SetIsopen(true)
    16  	isOpen := log.GetIsopen()
    17  	AssertEqual(t, true, isOpen)
    18  
    19  	log.SetLastLogMsg("logMsg")
    20  	msg := log.GetLastLogMsg()
    21  	AssertEqual(t, "logMsg", msg)
    22  
    23  	log.CloseLogger()
    24  	AssertEqual(t, false, log.isOpen)
    25  
    26  	log.OpenLogger()
    27  	AssertEqual(t, true, log.isOpen)
    28  }
    29  
    30  func Test_PrintLog(t *testing.T) {
    31  	fieldMap := make(map[string]string)
    32  	InitLogMsg(fieldMap)
    33  
    34  	originlogChannel := logChannel
    35  	SetLogChannel("Info")
    36  	defer func() {
    37  		logChannel = originlogChannel
    38  	}()
    39  
    40  	byt := new(bytes.Buffer)
    41  	logger := NewLogger("", "tea", byt, "")
    42  	logger.formatTemplate = "{channel} {error}"
    43  	logger.SetOutput(byt)
    44  	logger.PrintLog(fieldMap, errors.New("tea error"))
    45  	AssertEqual(t, byt.String(), "[INFO]logger_test.go:44: tea tea error\n")
    46  }