Skip to content

Commit 7f8410a

Browse files
committed
Improve MP unit test coverage
Signed-off-by: Stefan Büringer [email protected]
1 parent c5b4e93 commit 7f8410a

File tree

3 files changed

+188
-86
lines changed

3 files changed

+188
-86
lines changed

exp/internal/controllers/machinepool_controller.go

Lines changed: 3 additions & 16 deletions
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

Lines changed: 2 additions & 1 deletion
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)