@@ -379,12 +379,13 @@ func TestFilterLabels(t *testing.T) {
379379 }
380380
381381 type TC struct {
382- description string
383- labelName string
384- labelValue string
385- features nfdv1alpha1.Features
386- expectErr bool
387- expectedValue string
382+ description string
383+ labelName string
384+ labelValue string
385+ features nfdv1alpha1.Features
386+ expectErr bool
387+ expectedValue string
388+ expectedExtResources ExtendedResources
388389 }
389390
390391 tcs := []TC {
@@ -459,6 +460,24 @@ func TestFilterLabels(t *testing.T) {
459460 }
460461 })
461462 }
463+
464+ tcs = []TC {
465+ {
466+ description : "Unprefixed extended resources should not be allowed" ,
467+ expectedExtResources : ExtendedResources {},
468+ },
469+ }
470+
471+ extendedResources := ExtendedResources {"micromicrowaves" : "10" , "tooster" : "5" }
472+ for _ , tc := range tcs {
473+ t .Run (tc .description , func (t * testing.T ) {
474+ outExtendedResources := fakeMaster .filterExtendedResources (& tc .features , extendedResources )
475+ Convey ("Unprefixed extended resources should npotbe allowed" , t , func () {
476+ So (outExtendedResources , ShouldEqual , tc .expectedExtResources )
477+ })
478+ })
479+ }
480+
462481 // Create a new fake master with the feature gate enabled
463482 fakeMaster = newFakeMasterWithFeatureGate ()
464483 tcs = []TC {
@@ -481,6 +500,26 @@ func TestFilterLabels(t *testing.T) {
481500 })
482501 })
483502 }
503+
504+ tcs = []TC {
505+ {
506+ description : "Unprefixed extended resources should be allowed" ,
507+ expectedExtResources : ExtendedResources {
508+ "micromicrowaves" : "10" ,
509+ "tooster" : "5" ,
510+ },
511+ },
512+ }
513+
514+ extendedResources = ExtendedResources {"micromicrowaves" : "10" , "tooster" : "5" }
515+ for _ , tc := range tcs {
516+ t .Run (tc .description , func (t * testing.T ) {
517+ outExtendedResources := fakeMaster .filterExtendedResources (& tc .features , extendedResources )
518+ Convey ("Unprefixed extended resources should be allowed" , t , func () {
519+ So (outExtendedResources , ShouldEqual , tc .expectedExtResources )
520+ })
521+ })
522+ }
484523}
485524
486525func TestCreatePatches (t * testing.T ) {
0 commit comments