Move test/* to internal/test/*

This ensures this code cannot be imported by other repositories as this
is only intended for internal testing packages.

internal/test/util/kubernetes.go

@@ -0,0 +1,76 @@
+package util
+
+import (
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/tools/record"
+)
+
+// FakeConfigMap returns a configmap with the specified namespace, name and data.
+func FakeConfigMap(namespace, name string, data map[string]string) *corev1.ConfigMap {
+	return &corev1.ConfigMap{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      name,
+		},
+		Data: data,
+	}
+}
+
+// FakeEventRecorder returns an event recorder that can be used to capture events.
+func FakeEventRecorder(bufferSize int) *record.FakeRecorder {
+	return record.NewFakeRecorder(bufferSize)
+}
+
+// FakePodWithSingleContainer returns a pod with the specified namespace and name, and having a single container with the specified image.
+func FakePodWithSingleContainer(namespace, name, image string) *corev1.Pod {
+	enableServiceLink := corev1.DefaultEnableServiceLinks
+
+	return &corev1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      name,
+		},
+		Spec: corev1.PodSpec{
+			Containers: []corev1.Container{
+				{
+					Name:  name,
+					Image: image,
+				},
+			},
+			EnableServiceLinks: &enableServiceLink,
+		},
+	}
+}
+
+// FakeSecret returns a secret with the specified namespace, name and data.
+func FakeSecret(namespace, name string, data map[string]string) *corev1.Secret {
+	res := &corev1.Secret{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      name,
+		},
+		Data: make(map[string][]byte),
+	}
+	for key, val := range data {
+		res.Data[key] = []byte(val)
+	}
+	return res
+}
+
+// FakeService returns a service with the specified namespace and name and service info.
+func FakeService(namespace, name, clusterIP, protocol string, port int32) *corev1.Service {
+	return &corev1.Service{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespace,
+			Name:      name,
+		},
+		Spec: corev1.ServiceSpec{
+			Ports: []corev1.ServicePort{{
+				Protocol: corev1.Protocol(protocol),
+				Port:     port,
+			}},
+			ClusterIP: clusterIP,
+		},
+	}
+}

internal/test/util/resource_manager.go

@@ -0,0 +1,42 @@
+package util
+
+import (
+	"time"
+
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/util/wait"
+	"k8s.io/client-go/informers"
+	"k8s.io/client-go/kubernetes/fake"
+	"k8s.io/client-go/tools/cache"
+
+	"github.com/virtual-kubelet/virtual-kubelet/manager"
+)
+
+// FakeResourceManager returns an instance of the resource manager that will return the specified objects when its "GetX" methods are called.
+// Objects can be any valid Kubernetes object (corev1.Pod, corev1.ConfigMap, corev1.Secret, ...).
+func FakeResourceManager(objects ...runtime.Object) *manager.ResourceManager {
+	// Create a fake Kubernetes client that will list the specified objects.
+	kubeClient := fake.NewSimpleClientset(objects...)
+	// Create a shared informer factory from where we can grab informers and listers for pods, configmaps, secrets and services.
+	kubeInformerFactory := informers.NewSharedInformerFactory(kubeClient, 30*time.Second)
+	// Grab informers for pods, configmaps and secrets.
+	pInformer := kubeInformerFactory.Core().V1().Pods()
+	mInformer := kubeInformerFactory.Core().V1().ConfigMaps()
+	sInformer := kubeInformerFactory.Core().V1().Secrets()
+	svcInformer := kubeInformerFactory.Core().V1().Services()
+	// Start all the required informers.
+	go pInformer.Informer().Run(wait.NeverStop)
+	go mInformer.Informer().Run(wait.NeverStop)
+	go sInformer.Informer().Run(wait.NeverStop)
+	go svcInformer.Informer().Run(wait.NeverStop)
+	// Wait for the caches to be synced.
+	if !cache.WaitForCacheSync(wait.NeverStop, pInformer.Informer().HasSynced, mInformer.Informer().HasSynced, sInformer.Informer().HasSynced, svcInformer.Informer().HasSynced) {
+		panic("failed to wait for caches to be synced")
+	}
+	// Create a new instance of the resource manager using the listers for pods, configmaps and secrets.
+	r, err := manager.NewResourceManager(pInformer.Lister(), sInformer.Lister(), mInformer.Lister(), svcInformer.Lister())
+	if err != nil {
+		panic(err)
+	}
+	return r
+}