@@ -272,39 +272,39 @@ func TestAddingExtResources(t *testing.T) {
272
272
fakeMaster := newFakeMaster ()
273
273
Convey ("When there are no matching labels" , func () {
274
274
testNode := newTestNode ()
275
- resourceLabels := ExtendedResources {}
276
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
275
+ extendedResources := ExtendedResources {}
276
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
277
277
So (len (patches ), ShouldEqual , 0 )
278
278
})
279
279
280
280
Convey ("When there are matching labels" , func () {
281
281
testNode := newTestNode ()
282
- resourceLabels := ExtendedResources {"feature-1" : "1" , "feature-2" : "2" }
282
+ extendedResources := ExtendedResources {"feature-1" : "1" , "feature-2" : "2" }
283
283
expectedPatches := []utils.JsonPatch {
284
284
utils .NewJsonPatch ("add" , "/status/capacity" , "feature-1" , "1" ),
285
285
utils .NewJsonPatch ("add" , "/status/capacity" , "feature-2" , "2" ),
286
286
}
287
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
287
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
288
288
So (sortJsonPatches (patches ), ShouldResemble , sortJsonPatches (expectedPatches ))
289
289
})
290
290
291
291
Convey ("When the resource already exists" , func () {
292
292
testNode := newTestNode ()
293
293
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-1" )] = * resource .NewQuantity (1 , resource .BinarySI )
294
- resourceLabels := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-1" : "1" }
295
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
294
+ extendedResources := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-1" : "1" }
295
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
296
296
So (len (patches ), ShouldEqual , 0 )
297
297
})
298
298
299
299
Convey ("When the resource already exists but its capacity has changed" , func () {
300
300
testNode := newTestNode ()
301
301
testNode .Status .Capacity [corev1 .ResourceName ("feature-1" )] = * resource .NewQuantity (2 , resource .BinarySI )
302
- resourceLabels := ExtendedResources {"feature-1" : "1" }
302
+ extendedResources := ExtendedResources {"feature-1" : "1" }
303
303
expectedPatches := []utils.JsonPatch {
304
304
utils .NewJsonPatch ("replace" , "/status/capacity" , "feature-1" , "1" ),
305
305
utils .NewJsonPatch ("replace" , "/status/allocatable" , "feature-1" , "1" ),
306
306
}
307
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
307
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
308
308
So (sortJsonPatches (patches ), ShouldResemble , sortJsonPatches (expectedPatches ))
309
309
})
310
310
})
@@ -315,29 +315,29 @@ func TestRemovingExtResources(t *testing.T) {
315
315
fakeMaster := newFakeMaster ()
316
316
Convey ("When none are removed" , func () {
317
317
testNode := newTestNode ()
318
- resourceLabels := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-1" : "1" , nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
318
+ extendedResources := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-1" : "1" , nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
319
319
testNode .Annotations [nfdv1alpha1 .AnnotationNs + "/extended-resources" ] = "feature-1,feature-2"
320
320
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-1" )] = * resource .NewQuantity (1 , resource .BinarySI )
321
321
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-2" )] = * resource .NewQuantity (2 , resource .BinarySI )
322
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
322
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
323
323
So (len (patches ), ShouldEqual , 0 )
324
324
})
325
325
Convey ("When the related label is gone" , func () {
326
326
testNode := newTestNode ()
327
- resourceLabels := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-4" : "" , nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
327
+ extendedResources := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-4" : "" , nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
328
328
testNode .Annotations [nfdv1alpha1 .AnnotationNs + "/extended-resources" ] = "feature-4,feature-2"
329
329
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-4" )] = * resource .NewQuantity (4 , resource .BinarySI )
330
330
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-2" )] = * resource .NewQuantity (2 , resource .BinarySI )
331
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
331
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
332
332
So (len (patches ), ShouldBeGreaterThan , 0 )
333
333
})
334
334
Convey ("When the extended resource is no longer wanted" , func () {
335
335
testNode := newTestNode ()
336
336
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-1" )] = * resource .NewQuantity (1 , resource .BinarySI )
337
337
testNode .Status .Capacity [corev1 .ResourceName (nfdv1alpha1 .FeatureLabelNs + "/feature-2" )] = * resource .NewQuantity (2 , resource .BinarySI )
338
- resourceLabels := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
338
+ extendedResources := ExtendedResources {nfdv1alpha1 .FeatureLabelNs + "/feature-2" : "2" }
339
339
testNode .Annotations [nfdv1alpha1 .AnnotationNs + "/extended-resources" ] = "feature-1,feature-2"
340
- patches := fakeMaster .createExtendedResourcePatches (testNode , resourceLabels )
340
+ patches := fakeMaster .createExtendedResourcePatches (testNode , extendedResources )
341
341
So (len (patches ), ShouldBeGreaterThan , 0 )
342
342
})
343
343
})
@@ -528,7 +528,7 @@ func TestRemoveLabelsWithPrefix(t *testing.T) {
528
528
func TestConfigParse (t * testing.T ) {
529
529
Convey ("When parsing configuration" , t , func () {
530
530
master := newFakeMaster ()
531
- overrides := `{"noPublish": true, "enableTaints": true, "extraLabelNs": ["added.ns.io","added.kubernetes.io"], "denyLabelNs": ["denied.ns.io","denied.kubernetes.io"], "resourceLabels": ["vendor-1.com/feature-1","vendor-2.io/feature-2"], " labelWhiteList": "foo"}`
531
+ overrides := `{"noPublish": true, "enableTaints": true, "extraLabelNs": ["added.ns.io","added.kubernetes.io"], "denyLabelNs": ["denied.ns.io","denied.kubernetes.io"], "labelWhiteList": "foo"}`
532
532
533
533
Convey ("and no core cmdline flags have been specified" , func () {
534
534
So (master .configure ("non-existing-file" , overrides ), ShouldBeNil )
@@ -537,7 +537,6 @@ func TestConfigParse(t *testing.T) {
537
537
So (master .config .EnableTaints , ShouldResemble , true )
538
538
So (master .config .ExtraLabelNs , ShouldResemble , utils.StringSetVal {"added.ns.io" : struct {}{}, "added.kubernetes.io" : struct {}{}})
539
539
So (master .config .DenyLabelNs , ShouldResemble , utils.StringSetVal {"denied.ns.io" : struct {}{}, "denied.kubernetes.io" : struct {}{}})
540
- So (master .config .ResourceLabels , ShouldResemble , utils.StringSetVal {"vendor-1.com/feature-1" : struct {}{}, "vendor-2.io/feature-2" : struct {}{}})
541
540
So (master .config .LabelWhiteList .String (), ShouldEqual , "foo" )
542
541
})
543
542
})
@@ -563,7 +562,6 @@ func TestConfigParse(t *testing.T) {
563
562
_ , err = f .WriteString (`
564
563
noPublish: true
565
564
denyLabelNs: ["denied.ns.io","denied.kubernetes.io"]
566
- resourceLabels: ["vendor-1.com/feature-1","vendor-2.io/feature-2"]
567
565
enableTaints: false
568
566
labelWhiteList: "foo"
569
567
leaderElection:
@@ -582,7 +580,6 @@ leaderElection:
582
580
So (master .config .NoPublish , ShouldBeTrue )
583
581
So (master .config .EnableTaints , ShouldBeFalse )
584
582
So (master .config .ExtraLabelNs , ShouldResemble , utils.StringSetVal {"override.added.ns.io" : struct {}{}})
585
- So (master .config .ResourceLabels , ShouldResemble , utils.StringSetVal {"vendor-1.com/feature-1" : struct {}{}, "vendor-2.io/feature-2" : struct {}{}}) // from cmdline
586
583
So (master .config .DenyLabelNs , ShouldResemble , utils.StringSetVal {"denied.ns.io" : struct {}{}, "denied.kubernetes.io" : struct {}{}})
587
584
So (master .config .LabelWhiteList .String (), ShouldEqual , "foo" )
588
585
So (master .config .LeaderElection .LeaseDuration .Seconds (), ShouldEqual , float64 (20 ))
0 commit comments