@@ -314,7 +314,8 @@ func TestDistributorQuerier_Retry(t *testing.T) {
314
314
t .Run (name , func (t * testing.T ) {
315
315
d := & MockDistributor {}
316
316
317
- if tc .api == "Select" {
317
+ switch tc .api {
318
+ case "Select" :
318
319
promChunk := util .GenerateChunk (t , time .Second , model .TimeFromUnix (time .Now ().Unix ()), 10 , promchunk .PrometheusXorChunk )
319
320
clientChunks , err := chunkcompat .ToChunks ([]chunk.Chunk {promChunk })
320
321
require .NoError (t , err )
@@ -331,13 +332,13 @@ func TestDistributorQuerier_Retry(t *testing.T) {
331
332
},
332
333
}, err ).Once ()
333
334
}
334
- } else if tc . api == "LabelNames" {
335
+ case "LabelNames" :
335
336
res := []string {"foo" }
336
337
for _ , err := range tc .errors {
337
338
d .On ("LabelNamesStream" , mock .Anything , mock .Anything , mock .Anything , mock .Anything , mock .Anything ).Return (res , err ).Once ()
338
339
d .On ("LabelNames" , mock .Anything , mock .Anything , mock .Anything , mock .Anything , mock .Anything ).Return (res , err ).Once ()
339
340
}
340
- } else if tc . api == "LabelValues" {
341
+ case "LabelValues" :
341
342
res := []string {"foo" }
342
343
for _ , err := range tc .errors {
343
344
d .On ("LabelValuesForLabelNameStream" , mock .Anything , mock .Anything , mock .Anything , mock .Anything , mock .Anything , mock .Anything ).Return (res , err ).Once ()
@@ -366,9 +367,10 @@ func TestDistributorQuerier_Retry(t *testing.T) {
366
367
} else {
367
368
var annots annotations.Annotations
368
369
var err error
369
- if tc .api == "LabelNames" {
370
+ switch tc .api {
371
+ case "LabelNames" :
370
372
_ , annots , err = querier .LabelNames (ctx , nil , labels .MustNewMatcher (labels .MatchEqual , "foo" , "bar" ))
371
- } else if tc . api == "LabelValues" {
373
+ case "LabelValues" :
372
374
_ , annots , err = querier .LabelValues (ctx , "foo" , nil , labels .MustNewMatcher (labels .MatchEqual , "foo" , "bar" ))
373
375
}
374
376
0 commit comments