test: deflake by waiting for healthy pods

Azure · Jan 16, 2025 · deaa7ae · deaa7ae
1 parent f6d20e0
commit deaa7ae
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/suites/drift/suite_test.go b/test/suites/drift/suite_test.go
@@ -580,11 +580,13 @@ var _ = Describe("Drift", func() {
 					},
 				},
 			})
+			selector = labels.SelectorFromSet(dep.Spec.Selector.MatchLabels)
 			env.ExpectCreated(dep, nodeClass, nodePool)
 
 			By("deploying multiple replicas, pod per node")
 			startingNodeClaimState := env.EventuallyExpectCreatedNodeClaimCount("==", int(numPods))
 			env.EventuallyExpectCreatedNodeCount("==", int(numPods))
+			env.EventuallyExpectHealthyPodCount(selector, int(numPods))
 
 			By("drifting the nodeClaim with bad configuration that never registers")
 			nodeClass.Spec.VNETSubnetID = lo.ToPtr("/subscriptions/12345678-1234-1234-1234-123456789012/resourceGroups/sillygeese/providers/Microsoft.Network/virtualNetworks/karpenter/subnets/nodeclassSubnet2")
@@ -626,11 +628,13 @@ var _ = Describe("Drift", func() {
 					},
 				},
 			})
+			selector = labels.SelectorFromSet(dep.Spec.Selector.MatchLabels)
 			env.ExpectCreated(dep, nodeClass, nodePool)
 
 			By("deploying multiple replicas, pod per node")
 			startingNodeClaimState := env.EventuallyExpectCreatedNodeClaimCount("==", int(numPods))
 			env.EventuallyExpectCreatedNodeCount("==", int(numPods))
+			env.EventuallyExpectHealthyPodCount(selector, int(numPods))
 
 			By("drifting the nodeClaim with bad configuration that never initializes")
 			nodePool.Spec.Template.Spec.StartupTaints = []corev1.Taint{{Key: "example.com/taint", Effect: corev1.TaintEffectPreferNoSchedule}}