From e95c8b868413664a4fcfedf7c714bc0c75a4e843 Mon Sep 17 00:00:00 2001 From: Wei Huang Date: Sat, 5 Feb 2022 23:57:26 -0800 Subject: [PATCH] missing capacitySchedulingConfig config test --- cmd/scheduler/main_test.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/cmd/scheduler/main_test.go b/cmd/scheduler/main_test.go index 0e7e79423..b45acdab9 100644 --- a/cmd/scheduler/main_test.go +++ b/cmd/scheduler/main_test.go @@ -35,6 +35,7 @@ import ( "k8s.io/kubernetes/pkg/scheduler/apis/config" "k8s.io/kubernetes/pkg/scheduler/apis/config/testing/defaults" + "sigs.k8s.io/scheduler-plugins/pkg/capacityscheduling" "sigs.k8s.io/scheduler-plugins/pkg/coscheduling" "sigs.k8s.io/scheduler-plugins/pkg/noderesources" "sigs.k8s.io/scheduler-plugins/pkg/noderesourcetopology" @@ -526,6 +527,28 @@ profiles: }, }, }, + { + name: "single profile config - Capacityscheduling", + flags: []string{"--config", capacitySchedulingConfigWithArgsFile}, + registryOptions: []app.Option{app.WithPlugin(capacityscheduling.Name, capacityscheduling.New)}, + wantPlugins: map[string]*config.Plugins{ + "default-scheduler": { + QueueSort: defaults.PluginsV1beta2.QueueSort, + Bind: defaults.PluginsV1beta2.Bind, + PreFilter: config.PluginSet{ + Enabled: append(defaults.PluginsV1beta2.PreFilter.Enabled, config.Plugin{Name: capacityscheduling.Name}), + }, + Filter: defaults.PluginsV1beta2.Filter, + PostFilter: config.PluginSet{Enabled: []config.Plugin{{Name: capacityscheduling.Name}}}, + PreScore: defaults.PluginsV1beta2.PreScore, + Score: defaults.PluginsV1beta2.Score, + Reserve: config.PluginSet{ + Enabled: append(defaults.PluginsV1beta2.Reserve.Enabled, config.Plugin{Name: capacityscheduling.Name}), + }, + PreBind: defaults.PluginsV1beta2.PreBind, + }, + }, + }, { name: "single profile config - TargetLoadPacking with args", flags: []string{"--config", targetLoadPackingConfigWithArgsFile},