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
|
package jl
import (
"bytes"
"encoding/json"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestCompactPrinter_PrintNonJSON(t *testing.T) {
raw := "hello world"
buf := &bytes.Buffer{}
printer := NewCompactPrinter(buf)
entry := &Entry{
Raw: []byte(raw),
}
printer.Print(entry)
assert.Equal(t, raw+"\n", buf.String())
}
func TestCompactPrinter_Print(t *testing.T) {
tests := []struct {
name string
json string
formatted string
}{{
name: "basic",
json: `{"timestamp":"2019-01-01 15:23:45","level":"INFO","thread":"truck-manager","traceId":"123125f435d657a1","logger":"TruckRepairServiceOverlordManager","message":"There are 7 more trucks in the garage to fix. Get to work."}`,
formatted: "INFO 2019-01-01 15:23:45 [truck-manager] TruckRepa…ordManager| 123125f435d657a1| There are 7 more trucks in the garage to fix. Get to work.\n",
}, {
name: "exception",
json: `{"timestamp":"2019-01-01 15:34:45","level":"ERROR","thread":"repair-worker-2","traceId":"123125f435d657a1","logger":"TruckRepairMinion","message":"Truck 5 has is really broken! I'm need parts, waiting till they come."}`,
formatted: `ERRO 2019-01-01 15:34:45 [repair-worker-2] TruckRepairMinion| 123125f435d657a1| Truck 5 has is really broken! I'm need parts, waiting till they come.
`}, {
name: "logrus_pgk_error",
json: `{"timestamp":"2019-01-01 15:23:45","level":"error","message":"an error occurred","traceId":"123125f435d657a1","error":"BOOM!","stack":"github.com/pkg/errors_test.fn\n\t/home/dfc/src/github.com/pkg/errors/example_test.go:47\ngithub.com/pkg/errors_test.Example_stackTrace\n\t/home/dfc/src/github.com/pkg/errors/example_test.go:127\n"}`,
formatted: `ERRO 2019-01-01 15:23:45 123125f435d657a1| an error occurred
BOOM!
github.com/pkg/errors_test.fn
/home/dfc/src/github.com/pkg/errors/example_test.go:47
github.com/pkg/errors_test.Example_stackTrace
/home/dfc/src/github.com/pkg/errors/example_test.go:127
`,
}}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
buf := &bytes.Buffer{}
printer := NewCompactPrinter(buf)
printer.DisableColor = true
entry := &Entry{
Raw: []byte(test.json),
}
require.NoError(t, json.Unmarshal([]byte(test.json), &entry.Partials))
printer.Print(entry)
assert.Equal(t, test.formatted, buf.String())
})
}
}
func TestCompactPrinter_PrintWithColors(t *testing.T) {
var logs = []string{
`{"timestamp":"2019-01-01 15:24:45","level":"info","thread":"repair-worker-1","logger":"truckrepairminion","message":"fixing truck 1, it's got a broken axle"}`,
`{"timestamp":"2019-01-01 15:25:45","level":"info","thread":"repair-worker-2","logger":"truckrepairminion","message":"fixing truck 2, it's got a broken axle"}`,
}
var formatted = []string{
"\x1b[32mINFO\x1b[0m 2019-01-01 15:24:45 \x1b[32m[repair-worker-1] \x1b[0m \x1b[32m truckrepairminion|\x1b[0m fixing truck 1, it's got a broken axle\n",
"\x1b[32mINFO\x1b[0m 2019-01-01 15:25:45 \x1b[33m[repair-worker-2] \x1b[0m \x1b[32m truckrepairminion|\x1b[0m fixing truck 2, it's got a broken axle\n",
}
printer := NewCompactPrinter(nil)
for i, log := range logs {
buf := &bytes.Buffer{}
printer.Out = buf
entry := &Entry{
Raw: []byte(log),
}
require.NoError(t, json.Unmarshal([]byte(log), &entry.Partials))
printer.Print(entry)
assert.Equal(t, formatted[i], buf.String())
}
}
|
