Skip to content

Commit a57aee3

Browse files
committed
Handle controller name validation in testing
1 parent a198440 commit a57aee3

File tree

5 files changed

+19
-11
lines changed

5 files changed

+19
-11
lines changed

pkg/controller/machine/controller.go

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,6 @@ func newReconciler(mgr manager.Manager, actuator Actuator, gate featuregate.Muta
121121
return r
122122
}
123123

124-
// add adds a new Controller to mgr with r as the reconcile.Reconciler
125-
func add(mgr manager.Manager, r reconcile.Reconciler, controllerName string) error {
126-
return addWithOpts(mgr, controller.Options{Reconciler: r}, controllerName)
127-
}
128-
129124
// add adds a new Controller to mgr with r as the reconcile.Reconciler
130125
func addWithOpts(mgr manager.Manager, opts controller.Options, controllerName string) error {
131126
// Create a new controller

pkg/controller/machine/machine_controller_test.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,14 @@ import (
2626
"golang.org/x/net/context"
2727
corev1 "k8s.io/api/core/v1"
2828
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
29+
"k8s.io/utils/ptr"
2930

3031
machinev1resourcebuilder "github.com/openshift/cluster-api-actuator-pkg/testutils/resourcebuilder/machine/v1beta1"
3132
"github.com/openshift/cluster-control-plane-machine-set-operator/test/e2e/framework"
3233
testutils "github.com/openshift/machine-api-operator/pkg/util/testing"
3334

3435
"sigs.k8s.io/controller-runtime/pkg/client"
36+
"sigs.k8s.io/controller-runtime/pkg/controller"
3537
"sigs.k8s.io/controller-runtime/pkg/envtest/komega"
3638
"sigs.k8s.io/controller-runtime/pkg/manager"
3739
"sigs.k8s.io/controller-runtime/pkg/metrics/server"
@@ -64,7 +66,10 @@ var _ = Describe("Machine Reconciler", func() {
6466
act := newTestActuator()
6567
reconciler := newReconciler(mgr, act, gate)
6668

67-
Expect(add(mgr, reconciler, "testing")).To(Succeed())
69+
Expect(addWithOpts(mgr, controller.Options{
70+
Reconciler: reconciler,
71+
SkipNameValidation: ptr.To(true),
72+
}, "testing")).To(Succeed())
6873

6974
var mgrCtx context.Context
7075
mgrCtx, mgrCtxCancel = context.WithCancel(ctx)

pkg/controller/machineset/controller.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ var (
7373
// The Manager will set fields on the Controller and Start it when the Manager is Started.
7474
func Add(mgr manager.Manager, opts manager.Options, gate featuregate.MutableFeatureGate) error {
7575
r := newReconciler(mgr, gate)
76-
return add(mgr, r, r.MachineToMachineSets)
76+
return addWithOpts(mgr, controller.Options{Reconciler: r}, r.MachineToMachineSets)
7777
}
7878

7979
// newReconciler returns a new reconcile.Reconciler.
@@ -86,9 +86,9 @@ func newReconciler(mgr manager.Manager, gate featuregate.MutableFeatureGate) *Re
8686
}
8787

8888
// add adds a new Controller to mgr with r as the reconcile.Reconciler.
89-
func add(mgr manager.Manager, r reconcile.Reconciler, mapFn handler.TypedMapFunc[*machinev1.Machine, reconcile.Request]) error {
89+
func addWithOpts(mgr manager.Manager, opts controller.Options, mapFn handler.TypedMapFunc[*machinev1.Machine, reconcile.Request]) error {
9090
// Create a new controller.
91-
c, err := controller.New(controllerName, mgr, controller.Options{Reconciler: r})
91+
c, err := controller.New(controllerName, mgr, opts)
9292
if err != nil {
9393
return err
9494
}

pkg/controller/machineset/machineset_controller_test.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,9 @@ import (
3030
testutils "github.com/openshift/machine-api-operator/pkg/util/testing"
3131
corev1 "k8s.io/api/core/v1"
3232
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
33+
"k8s.io/utils/ptr"
3334
"sigs.k8s.io/controller-runtime/pkg/client"
35+
"sigs.k8s.io/controller-runtime/pkg/controller"
3436
"sigs.k8s.io/controller-runtime/pkg/envtest/komega"
3537
"sigs.k8s.io/controller-runtime/pkg/manager"
3638

@@ -64,7 +66,10 @@ var _ = Describe("MachineSet Reconciler", func() {
6466
By("Setting up a new reconciler")
6567
reconciler := newReconciler(mgr, gate)
6668

67-
Expect(add(mgr, reconciler, reconciler.MachineToMachineSets)).To(Succeed())
69+
Expect(addWithOpts(mgr, controller.Options{
70+
Reconciler: reconciler,
71+
SkipNameValidation: ptr.To(true),
72+
}, reconciler.MachineToMachineSets)).To(Succeed())
6873

6974
var mgrCtx context.Context
7075
mgrCtx, mgrCtxCancel = context.WithCancel(ctx)

pkg/controller/vsphere/machineset/controller_test.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ import (
2828
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
2929
"k8s.io/apimachinery/pkg/runtime"
3030
"k8s.io/client-go/tools/record"
31+
"k8s.io/utils/ptr"
3132
"sigs.k8s.io/controller-runtime/pkg/client"
3233
"sigs.k8s.io/controller-runtime/pkg/controller"
3334
"sigs.k8s.io/controller-runtime/pkg/log"
@@ -53,7 +54,9 @@ var _ = Describe("Reconciler", func() {
5354
Client: mgr.GetClient(),
5455
Log: log.Log,
5556
}
56-
Expect(r.SetupWithManager(mgr, controller.Options{})).To(Succeed())
57+
Expect(r.SetupWithManager(mgr, controller.Options{
58+
SkipNameValidation: ptr.To(true),
59+
})).To(Succeed())
5760

5861
fakeRecorder = record.NewFakeRecorder(1)
5962
r.recorder = fakeRecorder

0 commit comments

Comments
 (0)