feat: implement initial benchmark tests

Signed-off-by: Edmund Ochieng <[email protected]>

Author: OchiengEd

Diff for: internal/authentication/tokengetter_test.go

@@ -85,3 +85,65 @@ func TestTokenGetterGet(t *testing.T) {
 		}
 	}
 }
+
+func BenchmarkTokenGetterGet(b *testing.B) {
+	fakeClient := fake.NewSimpleClientset()
+	fakeClient.PrependReactor("create", "serviceaccounts/token",
+		func(action ctest.Action) (bool, runtime.Object, error) {
+			act, ok := action.(ctest.CreateActionImpl)
+			if !ok {
+				return false, nil, nil
+			}
+			tokenRequest := act.GetObject().(*authenticationv1.TokenRequest)
+			var err error
+			if act.Name == "test-service-account-1" {
+				tokenRequest.Status = authenticationv1.TokenRequestStatus{
+					Token:               "test-token-1",
+					ExpirationTimestamp: metav1.NewTime(metav1.Now().Add(DefaultExpirationDuration)),
+				}
+			}
+			if act.Name == "test-service-account-2" {
+				tokenRequest.Status = authenticationv1.TokenRequestStatus{
+					Token:               "test-token-2",
+					ExpirationTimestamp: metav1.NewTime(metav1.Now().Add(1 * time.Second)),
+				}
+			}
+			if act.Name == "test-service-account-3" {
+				tokenRequest.Status = authenticationv1.TokenRequestStatus{
+					Token:               "test-token-3",
+					ExpirationTimestamp: metav1.NewTime(metav1.Now().Add(-10 * time.Second)),
+				}
+			}
+			if act.Name == "test-service-account-4" {
+				tokenRequest = nil
+				err = fmt.Errorf("error when fetching token")
+			}
+			return true, tokenRequest, err
+		},
+	)
+
+	tg := NewTokenGetter(fakeClient.CoreV1(),
+		WithExpirationDuration(DefaultExpirationDuration),
+	)
+
+	test := struct {
+		serviceAccountName string
+		namespace          string
+	}{
+		serviceAccountName: "test-service-account-1",
+		namespace:          "test-namespace-1",
+	}
+
+	for n := 0; n < b.N; n++ {
+		token, err := tg.Get(context.Background(),
+			types.NamespacedName{
+				Namespace: test.namespace,
+				Name:      test.serviceAccountName,
+			},
+		)
+		if err != nil {
+			b.Logf("token creation; %v", err)
+		}
+		_ = token
+	}
+}

Diff for: internal/contentmanager/contentmanager_test.go

@@ -108,6 +108,48 @@ func TestWatch(t *testing.T) {
 	}
 }
 
+func BenchmarkWatch(b *testing.B) {
+	rcm := func(_ context.Context, _ client.Object, cfg *rest.Config) (*rest.Config, error) {
+		return cfg, nil
+	}
+	config := &rest.Config{}
+
+	ce := &v1alpha1.ClusterExtension{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: "test-cluster-extension",
+		},
+	}
+	objs := []client.Object{
+		&corev1.Pod{
+			TypeMeta: metav1.TypeMeta{
+				APIVersion: "v1",
+				Kind:       "Pod",
+			},
+			ObjectMeta: metav1.ObjectMeta{
+				Name: "webserver",
+			},
+		},
+	}
+
+	for n := 0; n < b.N; n++ {
+		mgr, _ := manager.New(config, manager.Options{})
+		ctrl, err := controller.New("test-controller", mgr, controller.Options{
+			Reconciler: reconcile.Func(func(context.Context, reconcile.Request) (reconcile.Result, error) {
+				return reconcile.Result{}, nil
+			}),
+		})
+		if err != nil {
+			b.Logf("creating controller; %v", err)
+		}
+
+		instance := New(rcm, config, mgr.GetRESTMapper())
+		err = instance.Watch(context.Background(), ctrl, ce, objs)
+		if err != nil {
+			b.Logf("setup watcher; %v", err)
+		}
+	}
+}
+
 func TestBuildScheme(t *testing.T) {
 	type validation struct {
 		gvks  []schema.GroupVersionKind
@@ -205,3 +247,27 @@ func TestBuildScheme(t *testing.T) {
 		})
 	}
 }
+
+func BenchmarkBuildScheme(b *testing.B) {
+	objects := []client.Object{
+		&appsv1.Deployment{
+			TypeMeta: metav1.TypeMeta{
+				APIVersion: "apps/v1",
+				Kind:       "Deployment",
+			},
+			ObjectMeta: metav1.ObjectMeta{
+				Name: "webserver",
+			},
+		},
+	}
+
+	gvk := appsv1.SchemeGroupVersion.WithKind("Deployment")
+
+	for n := 0; n < b.N; n++ {
+		scheme, err := buildScheme(objects)
+		if err != nil {
+			b.Logf("failed creating scheme; %v", err)
+		}
+		_ = scheme.Recognizes(gvk)
+	}
+}