diff --git a/pkg/scheduler/framework/runtime/framework_test.go b/pkg/scheduler/framework/runtime/framework_test.go index 755077b2b1c..74fd1d8d84a 100644 --- a/pkg/scheduler/framework/runtime/framework_test.go +++ b/pkg/scheduler/framework/runtime/framework_test.go @@ -338,6 +338,8 @@ var registry = func() Registry { r.Register(pluginNotImplementingScore, newPluginNotImplementingScore) r.Register(duplicatePluginName, newDuplicatePlugin) r.Register(testPlugin, newTestPlugin) + r.Register(queueSortPlugin, newQueueSortPlugin) + r.Register(bindPlugin, newBindPlugin) return r }() @@ -757,12 +759,19 @@ func TestNewFrameworkMultiPointExpansion(t *testing.T) { for _, tc := range tests { t.Run(tc.name, func(t *testing.T) { fw, err := NewFramework(registry, &config.KubeSchedulerProfile{Plugins: tc.plugins}) - if (err != nil && tc.wantErr == "") || (err == nil && tc.wantErr != "") || (err != nil && !strings.Contains(err.Error(), tc.wantErr)) { - t.Errorf("Unexpected error, got %v, expect: %s", err, tc.wantErr) + if err != nil { + if tc.wantErr == "" || !strings.Contains(err.Error(), tc.wantErr) { + t.Fatalf("Unexpected error, got %v, expect: %s", err, tc.wantErr) + } + } else { + if tc.wantErr != "" { + t.Fatalf("Unexpected error, got %v, expect: %s", err, tc.wantErr) + } } + if tc.wantErr == "" { if diff := cmp.Diff(tc.wantPlugins, fw.ListPlugins()); diff != "" { - t.Errorf("Unexpected eventToPlugin map (-want,+got):%s", diff) + t.Fatalf("Unexpected eventToPlugin map (-want,+got):%s", diff) } } }) @@ -1083,6 +1092,33 @@ func TestRunScorePlugins(t *testing.T) { }, err: true, }, + { + name: "single Score plugin with MultiPointExpansion", + plugins: &config.Plugins{ + MultiPoint: config.PluginSet{ + Enabled: []config.Plugin{ + {Name: scorePlugin1}, + }, + }, + Score: config.PluginSet{ + Enabled: []config.Plugin{ + {Name: scorePlugin1, Weight: 3}, + }, + }, + }, + pluginConfigs: []config.PluginConfig{ + { + Name: scorePlugin1, + Args: &runtime.Unknown{ + Raw: []byte(`{ "scoreRes": 1 }`), + }, + }, + }, + // scorePlugin1 Score returns 1, weight=3, so want=3. + want: framework.PluginToNodeScores{ + scorePlugin1: {{Name: "node1", Score: 3}, {Name: "node2", Score: 3}}, + }, + }, } for _, tt := range tests { @@ -1110,7 +1146,7 @@ func TestRunScorePlugins(t *testing.T) { t.Errorf("Expected status to be success.") } if !reflect.DeepEqual(res, tt.want) { - t.Errorf("Score map after RunScorePlugin: %+v, want: %+v.", res, tt.want) + t.Errorf("Score map after RunScorePlugin. got: %+v, want: %+v.", res, tt.want) } }) }