@@ -64,15 +64,17 @@ func TestNewLogSinkConfiguration(t *testing.T) {
64
64
65
65
func TestLogSink (t * testing.T ) {
66
66
for _ , tt := range []struct {
67
- name string
68
- f func (* logr.Logger )
69
- expectedRecords []expectedRecord
67
+ name string
68
+ f func (* logr.Logger )
69
+ expectedLoggerCount int
70
+ expectedRecords []expectedRecord
70
71
}{
71
72
{
72
73
name : "info" ,
73
74
f : func (l * logr.Logger ) {
74
75
l .Info ("info message" )
75
76
},
77
+ expectedLoggerCount : 1 ,
76
78
expectedRecords : []expectedRecord {
77
79
{
78
80
Body : log .StringValue ("info message" ),
@@ -94,6 +96,7 @@ func TestLogSink(t *testing.T) {
94
96
"uint64" , uint64 (3 ),
95
97
)
96
98
},
99
+ expectedLoggerCount : 1 ,
97
100
expectedRecords : []expectedRecord {
98
101
{
99
102
Body : log .StringValue ("msg" ),
@@ -116,6 +119,7 @@ func TestLogSink(t *testing.T) {
116
119
f : func (l * logr.Logger ) {
117
120
l .Error (errors .New ("test error" ), "error message" )
118
121
},
122
+ expectedLoggerCount : 1 ,
119
123
expectedRecords : []expectedRecord {
120
124
{
121
125
Body : log .StringValue ("error message" ),
@@ -140,6 +144,7 @@ func TestLogSink(t *testing.T) {
140
144
"uint64" , uint64 (3 ),
141
145
)
142
146
},
147
+ expectedLoggerCount : 1 ,
143
148
expectedRecords : []expectedRecord {
144
149
{
145
150
Body : log .StringValue ("msg" ),
@@ -163,13 +168,11 @@ func TestLogSink(t *testing.T) {
163
168
f : func (l * logr.Logger ) {
164
169
l .WithName ("test" ).Info ("info message with name" )
165
170
},
171
+ expectedLoggerCount : 2 ,
166
172
expectedRecords : []expectedRecord {
167
173
{
168
174
Body : log .StringValue ("info message with name" ),
169
175
Severity : log .SeverityInfo ,
170
- // Attributes: []log.KeyValue{
171
- // log.String(nameKey, "test"),
172
- // },
173
176
},
174
177
},
175
178
},
@@ -178,13 +181,11 @@ func TestLogSink(t *testing.T) {
178
181
f : func (l * logr.Logger ) {
179
182
l .WithName ("test" ).WithName ("test" ).Info ("info message with name" )
180
183
},
184
+ expectedLoggerCount : 3 ,
181
185
expectedRecords : []expectedRecord {
182
186
{
183
187
Body : log .StringValue ("info message with name" ),
184
188
Severity : log .SeverityInfo ,
185
- // Attributes: []log.KeyValue{
186
- // log.String(nameKey, "test/test"),
187
- // },
188
189
},
189
190
},
190
191
},
@@ -193,6 +194,7 @@ func TestLogSink(t *testing.T) {
193
194
f : func (l * logr.Logger ) {
194
195
l .WithValues ("key" , "value" ).Info ("info message with attrs" )
195
196
},
197
+ expectedLoggerCount : 1 ,
196
198
expectedRecords : []expectedRecord {
197
199
{
198
200
Body : log .StringValue ("info message with attrs" ),
@@ -208,6 +210,7 @@ func TestLogSink(t *testing.T) {
208
210
f : func (l * logr.Logger ) {
209
211
l .WithValues ("key1" , "value1" ).Info ("info message with attrs" , "key2" , "value2" )
210
212
},
213
+ expectedLoggerCount : 1 ,
211
214
expectedRecords : []expectedRecord {
212
215
{
213
216
Body : log .StringValue ("info message with attrs" ),
@@ -226,9 +229,10 @@ func TestLogSink(t *testing.T) {
226
229
l := logr .New (ls )
227
230
tt .f (& l )
228
231
229
- require .Len (t , rec .Result (), 2 )
230
- assert .Len (t , rec .Result ()[1 ].Records , len (tt .expectedRecords ))
231
- for i , record := range rec .Result ()[1 ].Records {
232
+ require .Len (t , rec .Result (), tt .expectedLoggerCount + 1 )
233
+
234
+ assert .Len (t , rec .Result ()[tt .expectedLoggerCount ].Records , len (tt .expectedRecords ))
235
+ for i , record := range rec .Result ()[tt .expectedLoggerCount ].Records {
232
236
assert .Equal (t , tt .expectedRecords [i ].Body , record .Body ())
233
237
assert .Equal (t , tt .expectedRecords [i ].Severity , record .Severity ())
234
238
0 commit comments