mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
9e1cb8ec47
Signed-off-by: Malte Poll <mp@edgeless.systems>
173 lines
5.0 KiB
Go
173 lines
5.0 KiB
Go
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
|
|
}
|