A mapping from predicates/priorities to pluing configuration.

This PR only implements the mapping, but does not use it. A followup PR will use this mapping to produce a framework configuration that redirects mapped predicates/priorites to be exected as plugins.

pkg/scheduler/framework/plugins/default_registry_test.go

@@ -0,0 +1,135 @@
+/*
+Copyright 2019 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package plugins
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+
+	"k8s.io/kubernetes/pkg/scheduler/apis/config"
+)
+
+func appendToPluginSet(pluginSet *config.PluginSet, name string, weight *int32) *config.PluginSet {
+	if pluginSet == nil {
+		pluginSet = &config.PluginSet{}
+	}
+	config := config.Plugin{Name: name}
+	if weight != nil {
+		config.Weight = *weight
+	}
+	pluginSet.Enabled = append(pluginSet.Enabled, config)
+	return pluginSet
+}
+
+func produceConfig(keys []string, producersMap map[string]ConfigProducer, args ConfigProducerArgs) (*config.Plugins, []config.PluginConfig, error) {
+	var plugins config.Plugins
+	var pluginConfig []config.PluginConfig
+	for _, k := range keys {
+		producer, exist := producersMap[k]
+		if !exist {
+			return nil, nil, fmt.Errorf("finding key %q", k)
+		}
+		p, pc := producer(args)
+		plugins.Append(&p)
+		pluginConfig = append(pluginConfig, pc...)
+	}
+	return &plugins, pluginConfig, nil
+}
+
+func TestRegisterConfigProducers(t *testing.T) {
+	registry := NewConfigProducerRegistry()
+	testPredicateName1 := "testPredicate1"
+	testFilterName1 := "testFilter1"
+	registry.RegisterPredicate(testPredicateName1,
+		func(_ ConfigProducerArgs) (plugins config.Plugins, pluginConfig []config.PluginConfig) {
+			plugins.Filter = appendToPluginSet(plugins.Filter, testFilterName1, nil)
+			return
+		})
+
+	testPredicateName2 := "testPredicate2"
+	testFilterName2 := "testFilter2"
+	registry.RegisterPredicate(testPredicateName2,
+		func(_ ConfigProducerArgs) (plugins config.Plugins, pluginConfig []config.PluginConfig) {
+			plugins.Filter = appendToPluginSet(plugins.Filter, testFilterName2, nil)
+			return
+		})
+
+	testPriorityName1 := "testPriority1"
+	testScoreName1 := "testScore1"
+	registry.RegisterPriority(testPriorityName1,
+		func(args ConfigProducerArgs) (plugins config.Plugins, pluginConfig []config.PluginConfig) {
+			plugins.Score = appendToPluginSet(plugins.Score, testScoreName1, &args.Weight)
+			return
+		})
+
+	testPriorityName2 := "testPriority2"
+	testScoreName2 := "testScore2"
+	registry.RegisterPriority(testPriorityName2,
+		func(args ConfigProducerArgs) (plugins config.Plugins, pluginConfig []config.PluginConfig) {
+			plugins.Score = appendToPluginSet(plugins.Score, testScoreName2, &args.Weight)
+			return
+		})
+
+	args := ConfigProducerArgs{Weight: 1}
+	predicatePlugins, _, err := produceConfig(
+		[]string{testPredicateName1, testPredicateName2}, registry.PredicateToConfigProducer, args)
+	if err != nil {
+		t.Fatalf("producing predicate framework configs: %v.", err)
+	}
+
+	priorityPlugins, _, err := produceConfig(
+		[]string{testPriorityName1, testPriorityName2}, registry.PriorityToConfigProducer, args)
+	if err != nil {
+		t.Fatalf("producing predicate framework configs: %v.", err)
+	}
+
+	// Verify that predicates and priorities are in the map and produce the expected score configurations.
+	var gotPlugins config.Plugins
+	gotPlugins.Append(predicatePlugins)
+	gotPlugins.Append(priorityPlugins)
+
+	// Verify the aggregated configuration.
+	wantPlugins := config.Plugins{
+		QueueSort: &config.PluginSet{},
+		PreFilter: &config.PluginSet{},
+		Filter: &config.PluginSet{
+			Enabled: []config.Plugin{
+				{Name: testFilterName1},
+				{Name: testFilterName2},
+			},
+		},
+		PostFilter: &config.PluginSet{},
+		Score: &config.PluginSet{
+			Enabled: []config.Plugin{
+				{Name: testScoreName1, Weight: 1},
+				{Name: testScoreName2, Weight: 1},
+			},
+		},
+		Reserve:   &config.PluginSet{},
+		Permit:    &config.PluginSet{},
+		PreBind:   &config.PluginSet{},
+		Bind:      &config.PluginSet{},
+		PostBind:  &config.PluginSet{},
+		Unreserve: &config.PluginSet{},
+	}
+
+	if diff := cmp.Diff(wantPlugins, gotPlugins); diff != "" {
+		t.Errorf("unexpected plugin configuration (-want, +got): %s", diff)
+	}
+}