[tests] de-duplicate framework handle initialization

pkg/framework/testing/utils.go

@@ -0,0 +1,66 @@
+package testing
+
+import (
+	"context"
+	"fmt"
+
+	v1 "k8s.io/api/core/v1"
+	"k8s.io/client-go/informers"
+	"k8s.io/client-go/tools/events"
+
+	clientset "k8s.io/client-go/kubernetes"
+	"sigs.k8s.io/descheduler/pkg/descheduler/evictions"
+	podutil "sigs.k8s.io/descheduler/pkg/descheduler/pod"
+	frameworkfake "sigs.k8s.io/descheduler/pkg/framework/fake"
+	"sigs.k8s.io/descheduler/pkg/framework/plugins/defaultevictor"
+	frameworktypes "sigs.k8s.io/descheduler/pkg/framework/types"
+)
+
+func InitFrameworkHandle(
+	ctx context.Context,
+	client clientset.Interface,
+	evictionOptions *evictions.Options,
+	defaultEvictorArgs defaultevictor.DefaultEvictorArgs,
+	getPodsAssignedToNodeSorter func([]*v1.Pod),
+) (*frameworkfake.HandleImpl, *evictions.PodEvictor, error) {
+	sharedInformerFactory := informers.NewSharedInformerFactory(client, 0)
+	podInformer := sharedInformerFactory.Core().V1().Pods().Informer()
+	podsAssignedToNode, err := podutil.BuildGetPodsAssignedToNodeFunc(podInformer)
+	if err != nil {
+		return nil, nil, fmt.Errorf("Build get pods assigned to node function error: %v", err)
+	}
+
+	var getPodsAssignedToNode func(s string, filterFunc podutil.FilterFunc) ([]*v1.Pod, error)
+	if getPodsAssignedToNodeSorter != nil {
+		getPodsAssignedToNode = func(s string, filterFunc podutil.FilterFunc) ([]*v1.Pod, error) {
+			pods, err := podsAssignedToNode(s, filterFunc)
+			getPodsAssignedToNodeSorter(pods)
+			return pods, err
+		}
+	} else {
+		getPodsAssignedToNode = podsAssignedToNode
+	}
+
+	sharedInformerFactory.Start(ctx.Done())
+	sharedInformerFactory.WaitForCacheSync(ctx.Done())
+	eventRecorder := &events.FakeRecorder{}
+	podEvictor := evictions.NewPodEvictor(client, eventRecorder, evictionOptions)
+	evictorFilter, err := defaultevictor.New(
+		&defaultEvictorArgs,
+		&frameworkfake.HandleImpl{
+			ClientsetImpl:                 client,
+			GetPodsAssignedToNodeFuncImpl: getPodsAssignedToNode,
+			SharedInformerFactoryImpl:     sharedInformerFactory,
+		},
+	)
+	if err != nil {
+		return nil, nil, fmt.Errorf("Unable to initialize the plugin: %v", err)
+	}
+	return &frameworkfake.HandleImpl{
+		ClientsetImpl:                 client,
+		GetPodsAssignedToNodeFuncImpl: getPodsAssignedToNode,
+		PodEvictorImpl:                podEvictor,
+		EvictorFilterImpl:             evictorFilter.(frameworktypes.EvictorPlugin),
+		SharedInformerFactoryImpl:     sharedInformerFactory,
+	}, podEvictor, nil
+}