Implements pod affinity NamespaceSelector feature

pkg/scheduler/framework/plugins/testing/testing.go

@@ -0,0 +1,59 @@
+/*
+Copyright 2021 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 testing
+
+import (
+	"context"
+	"testing"
+
+	"k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/client-go/informers"
+	"k8s.io/client-go/kubernetes/fake"
+	"k8s.io/kubernetes/pkg/scheduler/framework"
+	frameworkruntime "k8s.io/kubernetes/pkg/scheduler/framework/runtime"
+)
+
+// SetupPlugin creates a plugin using a framework handle that includes
+// the provided sharedLister and a SharedInformerFactory with the provided objects.
+// The function also creates an empty namespace (since most tests creates pods with
+// empty namespace), and start informer factory.
+func SetupPlugin(
+	ctx context.Context,
+	t *testing.T,
+	pf frameworkruntime.PluginFactory,
+	config runtime.Object,
+	sharedLister framework.SharedLister,
+	objs []runtime.Object,
+) framework.Plugin {
+	objs = append([]runtime.Object{&v1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: ""}}}, objs...)
+	informerFactory := informers.NewSharedInformerFactory(fake.NewSimpleClientset(objs...), 0)
+	fh, err := frameworkruntime.NewFramework(nil, nil, nil,
+		frameworkruntime.WithSnapshotSharedLister(sharedLister),
+		frameworkruntime.WithInformerFactory(informerFactory))
+	if err != nil {
+		t.Fatalf("Failed creating framework runtime: %v", err)
+	}
+	p, err := pf(config, fh)
+	if err != nil {
+		t.Fatal(err)
+	}
+	informerFactory.Start(ctx.Done())
+	informerFactory.WaitForCacheSync(ctx.Done())
+	return p
+}