1 package grpc_logrus
2
3 import (
4 "testing"
5
6 "github.com/sirupsen/logrus"
7 )
8
9 func Test_logrusGrpcLoggerV2_V(t *testing.T) {
10 tests := []struct {
11 name string
12 setupLevel logrus.Level
13 inLevel logrus.Level
14 want bool
15 }{
16 {
17 name: "WarnLevel setup when we have WarnLevel msg should return TRUE",
18 setupLevel: logrus.WarnLevel,
19 inLevel: logrus.WarnLevel,
20 want: true,
21 },
22 {
23 name: "WarnLevel setup when we have ErrorLevel msg should return TRUE",
24 setupLevel: logrus.WarnLevel,
25 inLevel: logrus.ErrorLevel,
26 want: true,
27 },
28 {
29 name: "WarnLevel setup when we have InfoLevel msg should return FALSE",
30 setupLevel: logrus.WarnLevel,
31 inLevel: logrus.InfoLevel,
32 want: false,
33 },
34 {
35 name: "WarnLevel setup when we have DebugLevel msg should return FALSE",
36 setupLevel: logrus.WarnLevel,
37 inLevel: logrus.DebugLevel,
38 want: false,
39 },
40 {
41 name: "WarnLevel setup when we have TraceLevel msg should return FALSE",
42 setupLevel: logrus.WarnLevel,
43 inLevel: logrus.TraceLevel,
44 want: false,
45 },
46 {
47 name: "TraceLevel setup when we have WarnLevel msg should return TRUE",
48 setupLevel: logrus.TraceLevel,
49 inLevel: logrus.WarnLevel,
50 want: true,
51 },
52 {
53 name: "TraceLevel setup when we have TraceLevel msg should return TRUE",
54 setupLevel: logrus.TraceLevel,
55 inLevel: logrus.TraceLevel,
56 want: true,
57 },
58 }
59 for _, tt := range tests {
60 t.Run(tt.name, func(t *testing.T) {
61 lr := logrus.New()
62 lr.SetLevel(tt.setupLevel)
63
64 l := &logrusGrpcLoggerV2{logrus.NewEntry(lr)}
65
66 if got := l.V(int(tt.inLevel)); got != tt.want {
67 t.Errorf("V() = %v, want %v", got, tt.want)
68 }
69 })
70 }
71 }
72
View as plain text