[node operator] controller client doubles

Signed-off-by: Malte Poll <mp@edgeless.systems>

operators/constellation-node-operator/controllers/client_test.go

@@ -0,0 +1,172 @@
+package controllers
+
+import (
+	"context"
+	"reflect"
+	"testing"
+
+	apimeta "k8s.io/apimachinery/pkg/api/meta"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+type stubReaderClient struct {
+	objects map[schema.GroupVersionKind]map[client.ObjectKey]runtime.Object
+	getErr  error
+	listErr error
+	scheme  *runtime.Scheme
+	client.Client
+}
+
+func newStubReaderClient(t *testing.T, objects []runtime.Object, getErr, listErr error) *stubReaderClient {
+	scheme := getScheme(t)
+	objectsMap := make(map[schema.GroupVersionKind]map[client.ObjectKey]runtime.Object)
+	for _, obj := range objects {
+		if obj == nil {
+			continue
+		}
+		gvks, _, err := scheme.ObjectKinds(obj)
+		if err != nil {
+			panic(err)
+		}
+		gvk := gvks[0]
+		if _, ok := objectsMap[gvk]; !ok {
+			objectsMap[gvk] = make(map[client.ObjectKey]runtime.Object)
+		}
+		objectsMap[gvk][client.ObjectKey{Namespace: obj.(client.Object).GetNamespace(), Name: obj.(client.Object).GetName()}] = obj
+	}
+	return &stubReaderClient{
+		objects: objectsMap,
+		getErr:  getErr,
+		listErr: listErr,
+		scheme:  scheme,
+	}
+}
+
+func (c *stubReaderClient) Get(_ context.Context, key client.ObjectKey, out client.Object) error {
+	gvks, _, err := c.scheme.ObjectKinds(out)
+	if err != nil {
+		panic(err)
+	}
+	gvk := gvks[0]
+	result := c.objects[gvk][key]
+	if result == nil {
+		return c.getErr
+	}
+	obj := result.DeepCopyObject()
+	outVal := reflect.ValueOf(out)
+	objVal := reflect.ValueOf(obj)
+	if !objVal.Type().AssignableTo(outVal.Type()) {
+		panic("type mismatch")
+	}
+	reflect.Indirect(outVal).Set(reflect.Indirect(objVal))
+	out.GetObjectKind().SetGroupVersionKind(gvk)
+	return c.getErr
+}
+
+func (c *stubReaderClient) List(_ context.Context, out client.ObjectList, opts ...client.ListOption) error {
+	gvks, _, err := c.scheme.ObjectKinds(out)
+	if err != nil {
+		panic(err)
+	}
+	listGVK := gvks[0]
+	itemGVK := listGVK.GroupVersion().WithKind(listGVK.Kind[:len(listGVK.Kind)-len("List")])
+	results := c.objects[itemGVK]
+	runtimeObjs := make([]runtime.Object, 0, len(results))
+	for _, item := range results {
+		outObj := item.DeepCopyObject()
+		outObj.GetObjectKind().SetGroupVersionKind(itemGVK)
+		runtimeObjs = append(runtimeObjs, outObj)
+	}
+	if err := apimeta.SetList(out, runtimeObjs); err != nil {
+		panic(err)
+	}
+	out.GetObjectKind().SetGroupVersionKind(listGVK)
+	return c.listErr
+}
+
+type stubWriterClient struct {
+	createErr      error
+	deleteErr      error
+	updateErr      error
+	patchErr       error
+	deleteAllOfErr error
+	statusWriter   stubStatusWriter
+	client.Client
+}
+
+func (c *stubWriterClient) Create(ctx context.Context, obj client.Object, opts ...client.CreateOption) error {
+	return c.createErr
+}
+
+func (c *stubWriterClient) Delete(ctx context.Context, obj client.Object, opts ...client.DeleteOption) error {
+	return c.deleteErr
+}
+
+func (c *stubWriterClient) Update(ctx context.Context, obj client.Object, opts ...client.UpdateOption) error {
+	return c.updateErr
+}
+
+func (c *stubWriterClient) Patch(ctx context.Context, obj client.Object, patch client.Patch, opts ...client.PatchOption) error {
+	return c.patchErr
+}
+
+func (c *stubWriterClient) DeleteAllOf(ctx context.Context, obj client.Object, opts ...client.DeleteAllOfOption) error {
+	return c.deleteAllOfErr
+}
+
+func (c *stubWriterClient) Status() client.StatusWriter {
+	return &c.statusWriter
+}
+
+type stubReadWriterClient struct {
+	stubReaderClient
+	stubWriterClient
+	client.Client
+}
+
+func (c *stubReadWriterClient) Get(ctx context.Context, key client.ObjectKey, obj client.Object) error {
+	return c.stubReaderClient.Get(ctx, key, obj)
+}
+
+func (c *stubReadWriterClient) List(ctx context.Context, list client.ObjectList, opts ...client.ListOption) error {
+	return c.stubReaderClient.List(ctx, list, opts...)
+}
+
+func (c *stubReadWriterClient) Create(ctx context.Context, obj client.Object, opts ...client.CreateOption) error {
+	return c.stubWriterClient.Create(ctx, obj, opts...)
+}
+
+func (c *stubReadWriterClient) Delete(ctx context.Context, obj client.Object, opts ...client.DeleteOption) error {
+	return c.stubWriterClient.Delete(ctx, obj, opts...)
+}
+
+func (c *stubReadWriterClient) Update(ctx context.Context, obj client.Object, opts ...client.UpdateOption) error {
+	return c.stubWriterClient.Update(ctx, obj, opts...)
+}
+
+func (c *stubReadWriterClient) Patch(ctx context.Context, obj client.Object, patch client.Patch, opts ...client.PatchOption) error {
+	return c.stubWriterClient.Patch(ctx, obj, patch, opts...)
+}
+
+func (c *stubReadWriterClient) DeleteAllOf(ctx context.Context, obj client.Object, opts ...client.DeleteAllOfOption) error {
+	return c.stubWriterClient.DeleteAllOf(ctx, obj, opts...)
+}
+
+func (c *stubReadWriterClient) Status() client.StatusWriter {
+	return c.stubWriterClient.Status()
+}
+
+type stubStatusWriter struct {
+	updateErr error
+	patchErr  error
+}
+
+func (w *stubStatusWriter) Update(ctx context.Context, obj client.Object, opts ...client.UpdateOption) error {
+	return w.updateErr
+}
+
+func (w *stubStatusWriter) Patch(ctx context.Context, obj client.Object, patch client.Patch, opts ...client.PatchOption) error {
+	return w.patchErr
+}

operators/constellation-node-operator/controllers/schemes_test.go

@@ -0,0 +1,20 @@
+package controllers
+
+import (
+	"testing"
+
+	"k8s.io/apimachinery/pkg/runtime"
+
+	updatev1alpha1 "github.com/edgelesssys/constellation/operators/constellation-node-operator/api/v1alpha1"
+	nodemaintenancev1beta1 "github.com/medik8s/node-maintenance-operator/api/v1beta1"
+	"github.com/stretchr/testify/require"
+	clientgoscheme "k8s.io/client-go/kubernetes/scheme"
+)
+
+func getScheme(t *testing.T) *runtime.Scheme {
+	scheme := runtime.NewScheme()
+	require.NoError(t, clientgoscheme.AddToScheme(scheme))
+	require.NoError(t, updatev1alpha1.AddToScheme(scheme))
+	require.NoError(t, nodemaintenancev1beta1.AddToScheme(scheme))
+	return scheme
+}