add UT for shuffle action and mock a plugin with golang mock

Signed-off-by: Thor-wl <13164644535@163.com>

go.mod

@@ -5,6 +5,7 @@ go 1.17
 require (
 	github.com/agiledragon/gomonkey/v2 v2.1.0
 	github.com/fsnotify/fsnotify v1.4.9
+	github.com/golang/mock v1.6.0
 	github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510
 	github.com/hashicorp/go-multierror v1.0.0
 	github.com/imdario/mergo v0.3.5

go.sum

@@ -249,6 +249,8 @@ github.com/golang/mock v1.4.1/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt
 github.com/golang/mock v1.4.3/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=
 github.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=
 github.com/golang/mock v1.5.0/go.mod h1:CWnOUgYIOo4TcNZ0wHX3YZCqsaM1I1Jvs6v3mP3KVu8=
+github.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=
+github.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=
 github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=

pkg/scheduler/actions/shuffle/shuffle_test.go

@@ -17,8 +17,10 @@
 package shuffle
 
 import (
+	"github.com/golang/mock/gomock"
 	"testing"
 	"time"
+	mock_framework "volcano.sh/volcano/pkg/scheduler/framework/mock_gen"
 
 	v1 "k8s.io/api/core/v1"
 	schedulingv1 "k8s.io/api/scheduling/v1"
@@ -39,7 +41,15 @@ func TestShuffle(t *testing.T) {
 	highPriority = 100
 	lowPriority = 10
 
-	framework.RegisterPluginBuilder("fake", NewFakePlugin)
+	ctl := gomock.NewController(t)
+	fakePlugin := mock_framework.NewMockPlugin(ctl)
+	fakePlugin.EXPECT().Name().AnyTimes().Return("fake")
+	fakePlugin.EXPECT().OnSessionOpen(gomock.Any()).Return()
+	fakePlugin.EXPECT().OnSessionClose(gomock.Any()).Return()
+	fakePluginBuilder := func(arguments framework.Arguments) framework.Plugin {
+		return fakePlugin
+	}
+	framework.RegisterPluginBuilder("fake", fakePluginBuilder)
 
 	tests := []struct {
 		name      string
@@ -170,8 +180,24 @@ func TestShuffle(t *testing.T) {
 		}, nil)
 		defer framework.CloseSession(ssn)
 
-		shuffle.Execute(ssn)
+		fakePluginVictimFns := func() []api.VictimTasksFn {
+			victimTasksFn := func(candidates []*api.TaskInfo) []*api.TaskInfo {
+				evicts := make([]*api.TaskInfo, 0)
+				for _, task := range candidates {
+					if task.Priority == lowPriority {
+						evicts = append(evicts, task)
+					}
+				}
+				return evicts
+			}
 
+			victimTasksFns := make([]api.VictimTasksFn, 0)
+			victimTasksFns = append(victimTasksFns, victimTasksFn)
+			return victimTasksFns
+		}
+		ssn.AddVictimTasksFns("fake", fakePluginVictimFns())
+
+		shuffle.Execute(ssn)
 		for {
 			select {
 			case <-evictor.Channel: