Enhance scheduler for TaintNodeByCondition.

plugin/pkg/scheduler/algorithmprovider/plugins_test.go

@@ -19,6 +19,7 @@ package algorithmprovider
 import (
 	"testing"
 
+	utilfeature "k8s.io/apiserver/pkg/util/feature"
 	"k8s.io/kubernetes/plugin/pkg/scheduler/factory"
 )
 
@@ -63,3 +64,46 @@ func TestAlgorithmProviders(t *testing.T) {
 		}
 	}
 }
+
+func TestApplyFeatureGates(t *testing.T) {
+	for _, pn := range algorithmProviderNames {
+		p, err := factory.GetAlgorithmProvider(pn)
+		if err != nil {
+			t.Errorf("Error retrieving '%s' provider: %v", pn, err)
+			break
+		}
+
+		if !p.FitPredicateKeys.Has("CheckNodeCondition") {
+			t.Errorf("Failed to find predicate: 'CheckNodeCondition'")
+			break
+		}
+
+		if !p.FitPredicateKeys.Has("PodToleratesNodeTaints") {
+			t.Errorf("Failed to find predicate: 'PodToleratesNodeTaints'")
+			break
+		}
+	}
+
+	// Apply features for algorithm providers.
+	utilfeature.DefaultFeatureGate.Set("TaintNodesByCondition=True")
+
+	ApplyFeatureGates()
+
+	for _, pn := range algorithmProviderNames {
+		p, err := factory.GetAlgorithmProvider(pn)
+		if err != nil {
+			t.Errorf("Error retrieving '%s' provider: %v", pn, err)
+			break
+		}
+
+		if !p.FitPredicateKeys.Has("PodToleratesNodeTaints") {
+			t.Errorf("Failed to find predicate: 'PodToleratesNodeTaints'")
+			break
+		}
+
+		if p.FitPredicateKeys.Has("CheckNodeCondition") {
+			t.Errorf("Unexpected predicate: 'CheckNodeCondition'")
+			break
+		}
+	}
+}