1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
package log
import (
"bytes"
"fmt"
"log"
"path/filepath"
"runtime"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestStdLog(t *testing.T) {
var buf bytes.Buffer
l := New(&buf)
cases := []struct {
f func(l *log.Logger)
name string
expected string
}{
{
name: "simple",
expected: "INFO info\n",
f: func(l *log.Logger) { l.Print("info") },
},
{
name: "without level",
expected: "INFO coffee\n",
f: func(l *log.Logger) { l.Print("coffee") },
},
{
name: "error level",
expected: "ERRO coffee\n",
f: func(l *log.Logger) { l.Print("ERROR coffee") },
},
}
for _, c := range cases {
buf.Reset()
t.Run(c.name, func(t *testing.T) {
l.SetOutput(&buf)
l.SetTimeFunction(_zeroTime)
c.f(l.StandardLog())
assert.Equal(t, c.expected, buf.String())
})
}
}
func TestStdLog_forceLevel(t *testing.T) {
var buf bytes.Buffer
logger := New(&buf)
cases := []struct {
name string
expected string
level Level
}{
{
name: "debug",
expected: "",
level: DebugLevel,
},
{
name: "info",
expected: "INFO coffee\n",
level: InfoLevel,
},
{
name: "error",
expected: "ERRO coffee\n",
level: ErrorLevel,
},
}
for _, c := range cases {
buf.Reset()
t.Run(c.name, func(t *testing.T) {
l := logger.StandardLog(StandardLogOptions{ForceLevel: c.level})
l.Print("coffee")
assert.Equal(t, c.expected, buf.String())
})
}
}
func TestStdLog_writer(t *testing.T) {
var buf bytes.Buffer
logger := New(&buf)
logger.SetReportCaller(true)
_, file, line, ok := runtime.Caller(0)
require.True(t, ok)
cases := []struct {
name string
expected string
level Level
}{
{
name: "debug",
expected: "",
level: DebugLevel,
},
{
name: "info",
expected: fmt.Sprintf("INFO <log/%s:%d> coffee\n", filepath.Base(file), line+27),
level: InfoLevel,
},
{
name: "error",
expected: fmt.Sprintf("ERRO <log/%s:%d> coffee\n", filepath.Base(file), line+27),
level: ErrorLevel,
},
}
for _, c := range cases {
buf.Reset()
t.Run(c.name, func(t *testing.T) {
l := log.New(logger.StandardLog(StandardLogOptions{ForceLevel: c.level}).Writer(), "", 0)
l.Print("coffee")
assert.Equal(t, c.expected, buf.String())
})
}
}
|
