@@ -344,49 +344,42 @@ func TestDeterminePilotsToBeRemoved(t *testing.T) {
344
344
expectedOutput []* v1alpha1.Pilot
345
345
err bool
346
346
}
347
- tests := [ ]testT {
348
- {
347
+ tests := map [ string ]testT {
348
+ "should only remove highest ordinal pilot in 'test' set" : {
349
349
inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" ), pilotWithName ("es-test-1" ), pilotWithName ("es-test-2" )},
350
350
statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (3 )}),
351
351
replicaDiff : - 1 ,
352
352
expectedOutput : []* v1alpha1.Pilot {pilotWithName ("es-test-2" )},
353
353
err : false ,
354
354
},
355
- {
355
+ "should not remove pilot that is not a member of the specified statefulset" : {
356
356
inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" ), pilotWithName ("es-mixed-1" ), pilotWithName ("es-test-1" )},
357
357
statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (2 )}),
358
358
replicaDiff : - 1 ,
359
359
expectedOutput : []* v1alpha1.Pilot {pilotWithName ("es-test-1" )},
360
360
err : false ,
361
361
},
362
- {
362
+ "should not remove any pilots if replicaDiff is 0 and highest ordinal pilot is missing" : {
363
363
inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" )},
364
364
statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (2 )}),
365
365
replicaDiff : 0 ,
366
366
err : false ,
367
367
},
368
- {
368
+ "should not remove any pilots if replica diff is 0" : {
369
369
inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" )},
370
370
statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (1 )}),
371
371
replicaDiff : 0 ,
372
372
err : false ,
373
373
},
374
- {
375
- inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" ), pilotWithName ("es-test-1" )},
376
- statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (2 )}),
377
- replicaDiff : - 1 ,
378
- expectedOutput : []* v1alpha1.Pilot {pilotWithName ("es-test-1" )},
379
- err : false ,
380
- },
381
- {
374
+ "should not remove any pilots of replicaDiff is positive" : {
382
375
inputList : []* v1alpha1.Pilot {pilotWithName ("es-test-0" ), pilotWithName ("es-test-1" )},
383
376
statefulSet : generate .StatefulSet (generate.StatefulSetConfig {Name : "es-test" , Replicas : int32Ptr (2 )}),
384
377
replicaDiff : 1 ,
385
378
err : false ,
386
379
},
387
380
}
388
- for i , test := range tests {
389
- t .Run (fmt . Sprintf ( "test-%d" , i ) , func (t * testing.T ) {
381
+ for name , test := range tests {
382
+ t .Run (name , func (t * testing.T ) {
390
383
output , err := determinePilotsToBeRemoved (test .inputList , test .statefulSet , test .replicaDiff )
391
384
if err != nil && ! test .err {
392
385
t .Errorf ("Expected no error but got: %v" , err )
0 commit comments