[WIP] 🌱 Enable webhook envtests

api/v1alpha3/suite_test.go

@@ -17,13 +17,16 @@ limitations under the License.
 package v1alpha3
 
 import (
+	"fmt"
 	"os"
 	"testing"
 
 	// +kubebuilder:scaffold:imports
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/kubernetes/scheme"
+	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/webhooks"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -38,5 +41,13 @@ func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
 		M:        m,
 		SetupEnv: func(e *envtest.Environment) { env = e },
+		SetupWebhooks: func(mgr ctrl.Manager) {
+			if err := clusterv1.SetupWebhooksWithManager(mgr); err != nil {
+				panic(fmt.Sprintf("Failed to set up webhooks: %v", err))
+			}
+			if err := webhooks.SetupWebhooksWithManager(mgr); err != nil {
+				panic(fmt.Sprintf("failed to set up webhooks: %v", err))
+			}
+		},
 	}))
 }

api/v1beta1/webhooks.go

@@ -0,0 +1,42 @@
+/*
+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 v1beta1
+
+import (
+	"github.com/pkg/errors"
+	ctrl "sigs.k8s.io/controller-runtime"
+)
+
+// SetupWebhooksWithManager sets up all webhooks of the current package.
+func SetupWebhooksWithManager(mgr ctrl.Manager) error {
+	if err := (&Machine{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create Machine webhook")
+	}
+
+	if err := (&MachineSet{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create MachineSet webhook")
+	}
+
+	if err := (&MachineDeployment{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create MachineDeployment webhook")
+	}
+
+	if err := (&MachineHealthCheck{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create MachineHealthCheck webhook")
+	}
+	return nil
+}

bootstrap/kubeadm/api/v1alpha3/suite_test.go

@@ -17,12 +17,14 @@ limitations under the License.
 package v1alpha3
 
 import (
+	"fmt"
 	"os"
 	"testing"
 
 	// +kubebuilder:scaffold:imports
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/kubernetes/scheme"
+	kubeadmbootstrapv1 "sigs.k8s.io/cluster-api/bootstrap/kubeadm/api/v1beta1"
 	"sigs.k8s.io/cluster-api/internal/envtest"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
@@ -38,5 +40,10 @@ func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
 		M:        m,
 		SetupEnv: func(e *envtest.Environment) { env = e },
+		SetupWebhooks: func(mgr ctrl.Manager) {
+			if err := kubeadmbootstrapv1.SetupWebhooksWithManager(mgr); err != nil {
+				panic(fmt.Sprintf("Failed to set up webhooks: %v", err))
+			}
+		},
 	}))
 }

bootstrap/kubeadm/api/v1beta1/webhooks.go

@@ -0,0 +1,33 @@
+/*
+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 v1beta1
+
+import (
+	"github.com/pkg/errors"
+	ctrl "sigs.k8s.io/controller-runtime"
+)
+
+// SetupWebhooksWithManager sets up all webhooks of the current package.
+func SetupWebhooksWithManager(mgr ctrl.Manager) error {
+	if err := (&KubeadmConfig{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create KubeadmConfig webhook")
+	}
+	if err := (&KubeadmConfigTemplate{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create KubeadmConfigTemplate webhook")
+	}
+	return nil
+}

bootstrap/kubeadm/internal/controllers/suite_test.go

@@ -22,6 +22,7 @@ import (
 
 	// +kubebuilder:scaffold:imports
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -32,7 +33,8 @@ var (
 
 func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

bootstrap/kubeadm/main.go

@@ -216,18 +216,10 @@ func setupWebhooks(mgr ctrl.Manager) {
 		setupLog.Error(err, "unable to create webhook", "webhook", "KubeadmConfig")
 		os.Exit(1)
 	}
-	if err := (&kubeadmbootstrapv1.KubeadmConfigList{}).SetupWebhookWithManager(mgr); err != nil {
-		setupLog.Error(err, "unable to create webhook", "webhook", "KubeadmConfigList")
-		os.Exit(1)
-	}
 	if err := (&kubeadmbootstrapv1.KubeadmConfigTemplate{}).SetupWebhookWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create webhook", "webhook", "KubeadmConfigTemplate")
 		os.Exit(1)
 	}
-	if err := (&kubeadmbootstrapv1.KubeadmConfigTemplateList{}).SetupWebhookWithManager(mgr); err != nil {
-		setupLog.Error(err, "unable to create webhook", "webhook", "KubeadmConfigTemplateList")
-		os.Exit(1)
-	}
 }
 
 func concurrency(c int) controller.Options {

bootstrap/util/suite_test.go

@@ -22,6 +22,7 @@ import (
 
 	// +kubebuilder:scaffold:imports
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -32,7 +33,8 @@ var (
 
 func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

controllers/remote/suite_test.go

@@ -23,6 +23,7 @@ import (
 
 	// +kubebuilder:scaffold:imports
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -37,7 +38,8 @@ var (
 
 func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

controllers/suite_test.go

@@ -36,6 +36,7 @@ import (
 	"sigs.k8s.io/cluster-api/api/v1beta1/index"
 	"sigs.k8s.io/cluster-api/controllers/remote"
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/controller"
 )
@@ -127,6 +128,7 @@ func TestMain(m *testing.M) {
 		M:                m,
 		SetupEnv:         func(e *envtest.Environment) { env = e },
 		SetupIndexes:     setupIndexes,
+		SetupWebhooks:    setup.AllWebhooksWithManager,
 		SetupReconcilers: setupReconcilers,
 	}))
 }

controllers/topology/suite_test.go

@@ -30,6 +30,7 @@ import (
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/cluster-api/api/v1beta1/index"
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/controller"
@@ -52,6 +53,7 @@ func TestMain(m *testing.M) {
 			panic(fmt.Sprintf("unable to setup index: %v", err))
 		}
 	}
+
 	setupReconcilers := func(ctx context.Context, mgr ctrl.Manager) {
 		unstructuredCachingClient, err := client.NewDelegatingClient(
 			client.NewDelegatingClientInput{
@@ -87,6 +89,7 @@ func TestMain(m *testing.M) {
 		ManagerUncachedObjs: []client.Object{},
 		SetupEnv:            func(e *envtest.Environment) { env = e },
 		SetupIndexes:        setupIndexes,
+		SetupWebhooks:       setup.AllWebhooksWithManager,
 		SetupReconcilers:    setupReconcilers,
 	}))
 }

controlplane/kubeadm/api/v1alpha3/suite_test.go

@@ -24,6 +24,7 @@ import (
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/kubernetes/scheme"
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -36,7 +37,8 @@ func TestMain(m *testing.M) {
 	utilruntime.Must(AddToScheme(scheme.Scheme))
 
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

controlplane/kubeadm/api/v1beta1/webhooks.go

@@ -0,0 +1,33 @@
+/*
+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 v1beta1
+
+import (
+	"github.com/pkg/errors"
+	ctrl "sigs.k8s.io/controller-runtime"
+)
+
+// SetupWebhooksWithManager sets up all webhooks of the current package.
+func SetupWebhooksWithManager(mgr ctrl.Manager) error {
+	if err := (&KubeadmControlPlane{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create KubeadmControlPlane webhook")
+	}
+	if err := (&KubeadmControlPlaneTemplate{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create KubeadmControlPlaneTemplate webhook")
+	}
+	return nil
+}

controlplane/kubeadm/controllers/suite_test.go

@@ -24,6 +24,7 @@ import (
 	apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -53,7 +54,8 @@ var (
 
 func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

controlplane/kubeadm/internal/suite_test.go

@@ -21,6 +21,7 @@ import (
 	"testing"
 
 	"sigs.k8s.io/cluster-api/internal/envtest"
+	"sigs.k8s.io/cluster-api/internal/envtest/setup"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
 
@@ -31,7 +32,8 @@ var (
 
 func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
-		M:        m,
-		SetupEnv: func(e *envtest.Environment) { env = e },
+		M:             m,
+		SetupEnv:      func(e *envtest.Environment) { env = e },
+		SetupWebhooks: setup.AllWebhooksWithManager,
 	}))
 }

exp/addons/api/v1alpha3/suite_test.go

@@ -17,12 +17,14 @@ limitations under the License.
 package v1alpha3
 
 import (
+	"fmt"
 	"os"
 	"testing"
 
 	// +kubebuilder:scaffold:imports
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/kubernetes/scheme"
+	addonsv1 "sigs.k8s.io/cluster-api/exp/addons/api/v1beta1"
 	"sigs.k8s.io/cluster-api/internal/envtest"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
@@ -38,5 +40,10 @@ func TestMain(m *testing.M) {
 	os.Exit(envtest.Run(ctx, envtest.RunInput{
 		M:        m,
 		SetupEnv: func(e *envtest.Environment) { env = e },
+		SetupWebhooks: func(mgr ctrl.Manager) {
+			if err := addonsv1.SetupWebhooksWithManager(mgr); err != nil {
+				panic(fmt.Sprintf("Failed to set up webhooks: %v", err))
+			}
+		},
 	}))
 }

exp/addons/api/v1beta1/webhooks.go

@@ -0,0 +1,30 @@
+/*
+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 v1beta1
+
+import (
+	"github.com/pkg/errors"
+	ctrl "sigs.k8s.io/controller-runtime"
+)
+
+// SetupWebhooksWithManager sets up all webhooks of the current package.
+func SetupWebhooksWithManager(mgr ctrl.Manager) error {
+	if err := (&ClusterResourceSet{}).SetupWebhookWithManager(mgr); err != nil {
+		return errors.Wrapf(err, "failed to create Machine webhook")
+	}
+	return nil
+}