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 }
47
View as plain text