/
grpc_logger_test.go
75 lines (67 loc) · 1.4 KB
/
grpc_logger_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
package sklog_test
import (
"bytes"
"testing"
"github.com/go-kit/kit/log"
"github.com/piotrkowalczuk/sklog"
"github.com/stretchr/testify/assert"
)
func TestGRPCLogger_Print(t *testing.T) {
b := bytes.NewBuffer(nil)
l := log.NewJSONLogger(b)
g := sklog.NewGRPCLogger(l)
success := map[string][]interface{}{
"message": {
"message",
},
"message1, message2": {
"message1",
"message2",
},
"1, [109 101 115 115 97 103 101 50]": {
1,
[]byte("message2"),
},
"true, [message1 message2]": {
true,
[]string{"message1", "message2"},
},
}
for expected, args := range success {
g.Print(args...)
assert.Contains(t, b.String(), sklog.LevelDebug)
assert.Contains(t, b.String(), sklog.KeyLevel)
assert.Contains(t, b.String(), expected)
b.Reset()
}
}
func TestGRPCLogger_Printf(t *testing.T) {
b := bytes.NewBuffer(nil)
l := log.NewJSONLogger(b)
g := sklog.NewGRPCLogger(l)
success := []struct {
expected string
format string
args []interface{}
}{
{
expected: "message",
format: "message",
},
{
expected: "message1, message2",
format: "%s, %s",
args: []interface{}{
"message1",
"message2",
},
},
}
for _, data := range success {
g.Printf(data.format, data.args...)
assert.Contains(t, b.String(), sklog.LevelDebug)
assert.Contains(t, b.String(), sklog.KeyLevel)
assert.Contains(t, b.String(), data.expected)
b.Reset()
}
}