-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpretty_target_test.go
122 lines (89 loc) · 2.93 KB
/
pretty_target_test.go
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
120
121
122
package blackbox_test
import (
"bytes"
"testing"
"github.com/RobertWHurst/blackbox"
"github.com/stretchr/testify/assert"
)
func TestPrettyTarget(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
values := make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"key": "value"})
assert.Regexp(
t,
`^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:-\d{2}:\d{2})|Z \x1b\[\d{2}mtrace\x1b\[`+
`0m Hello Test \x1b\[\d{2}mkey\x1b\[0m=value\n$`,
outBuf.String(),
)
}
func TestPrettyTargetSetLevel(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
prettyTarget.SetLevel(blackbox.Info)
values := make([]interface{}, 1)
values[0] = "Filtered Message"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"x": "y"})
values = make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Info, values, blackbox.Ctx{"key": "value"})
assert.NotRegexp(t, `Filtered Message`, outBuf.String())
assert.Regexp(t, `Hello Test`, outBuf.String())
}
func TestPrettyTargetShowTimestamp(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
prettyTarget.ShowTimestamp(false)
values := make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"key": "value"})
assert.Regexp(
t,
`^\x1b\[\d{2}mtrace\x1b\[0m Hello Test \x1b\[\d{2}mkey\x1b\[0m=value\n$`,
outBuf.String(),
)
}
func TestPrettyTargetSelectContext(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
prettyTarget.SelectContext("x")
values := make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"key": "value", "x": "y"})
assert.Regexp(t, "x[^ ]*=y", outBuf.String())
}
func TestPrettyTargetShowContext(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
prettyTarget.ShowContext(false)
values := make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"key": "value"})
assert.Regexp(
t,
`^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:-\d{2}:\d{2})|Z \x1b\[\d{2}mtrace\x1b\[`+
`0m Hello Test\n$`,
outBuf.String(),
)
}
func TestPrettyTargetUseColor(t *testing.T) {
outBuf := new(bytes.Buffer)
errBuf := new(bytes.Buffer)
prettyTarget := blackbox.NewPrettyTarget(outBuf, errBuf)
prettyTarget.UseColor(false)
values := make([]interface{}, 1)
values[0] = "Hello Test"
prettyTarget.Log(blackbox.Trace, values, blackbox.Ctx{"key": "value"})
assert.Regexp(
t,
`^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:-\d{2}:\d{2})|Z trace Hello Test key=`+
`value\n$`,
outBuf.String(),
)
}