Skip to content

Commit d36541f

Browse files
authored
Merge pull request #10939 from sbueringer/pr-improve-mp-unit-tests
🌱 Improve MP unit test coverage
2 parents c5b4e93 + 7f8410a commit d36541f

File tree

3 files changed

+188
-86
lines changed

3 files changed

+188
-86
lines changed

exp/internal/controllers/machinepool_controller.go

+3-16
Original file line numberDiff line numberDiff line change
@@ -302,10 +302,6 @@ func (r *MachinePoolReconciler) reconcileDeleteNodes(ctx context.Context, cluste
302302
return nil
303303
}
304304

305-
if r.Tracker == nil {
306-
return errors.New("Cannot establish cluster client to delete nodes")
307-
}
308-
309305
clusterClient, err := r.Tracker.GetClient(ctx, util.ObjectKey(cluster))
310306
if err != nil {
311307
return err
@@ -328,13 +324,9 @@ func (r *MachinePoolReconciler) isMachinePoolNodeDeleteTimeoutPassed(machinePool
328324
// reconcileDeleteExternal tries to delete external references, returning true if it cannot find any.
329325
func (r *MachinePoolReconciler) reconcileDeleteExternal(ctx context.Context, machinePool *expv1.MachinePool) (bool, error) {
330326
objects := []*unstructured.Unstructured{}
331-
references := []*corev1.ObjectReference{}
332-
// check for external ref
333-
if machinePool.Spec.Template.Spec.Bootstrap.ConfigRef != nil {
334-
references = append(references, machinePool.Spec.Template.Spec.Bootstrap.ConfigRef)
335-
}
336-
if machinePool.Spec.Template.Spec.InfrastructureRef != (corev1.ObjectReference{}) {
337-
references = append(references, &machinePool.Spec.Template.Spec.InfrastructureRef)
327+
references := []*corev1.ObjectReference{
328+
machinePool.Spec.Template.Spec.Bootstrap.ConfigRef,
329+
&machinePool.Spec.Template.Spec.InfrastructureRef,
338330
}
339331

340332
// Loop over the references and try to retrieve it with the client.
@@ -374,11 +366,6 @@ func (r *MachinePoolReconciler) watchClusterNodes(ctx context.Context, cluster *
374366
return nil
375367
}
376368

377-
// If there is no tracker, don't watch remote nodes
378-
if r.Tracker == nil {
379-
return nil
380-
}
381-
382369
return r.Tracker.Watch(ctx, remote.WatchInput{
383370
Name: "machinepool-watchNodes",
384371
Cluster: util.ObjectKey(cluster),

exp/internal/controllers/machinepool_controller_noderef.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ import (
2323

2424
"github.com/pkg/errors"
2525
corev1 "k8s.io/api/core/v1"
26+
apierrors "k8s.io/apimachinery/pkg/api/errors"
2627
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
2728
"k8s.io/klog/v2"
2829
ctrl "sigs.k8s.io/controller-runtime"
@@ -153,7 +154,7 @@ func (r *MachinePoolReconciler) deleteRetiredNodes(ctx context.Context, c client
153154
delete(nodeRefsMap, providerID)
154155
}
155156
for _, node := range nodeRefsMap {
156-
if err := c.Delete(ctx, node); err != nil {
157+
if err := c.Delete(ctx, node); err != nil && !apierrors.IsNotFound(err) {
157158
return errors.Wrapf(err, "failed to delete Node")
158159
}
159160
}

0 commit comments

Comments
 (0)