Fix flake by changing image behavior of high-priority-group (#4438)

kubernetes-sigs · Mar 3, 2025 · d135c1d · d135c1d
1 parent efe98ae
commit d135c1d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/e2e/singlecluster/pod_test.go b/test/e2e/singlecluster/pod_test.go
@@ -443,7 +443,7 @@ var _ = ginkgo.Describe("Pod groups", func() {
 			})
 
 			highPriorityGroup := podtesting.MakePod("high-priority-group", ns.Name).
-				Image(util.E2eTestAgnHostImage, util.BehaviorExitFast).
+				Image(util.E2eTestAgnHostImage, util.BehaviorWaitForDeletion).
 				Queue(lq.Name).
 				PriorityClass("high").
 				Request(corev1.ResourceCPU, "1").
@@ -517,6 +517,10 @@ var _ = ginkgo.Describe("Pod groups", func() {
 				}, util.Timeout, util.Interval).Should(gomega.Succeed())
 			})
 
+			ginkgo.By("Call high priority group pods to complete", func() {
+				util.WaitForActivePodsAndTerminate(ctx, k8sClient, restClient, cfg, ns.Name, 2, 0)
+			})
+
 			ginkgo.By("Verify the high priority group completes", func() {
 				util.ExpectWorkloadToFinish(ctx, k8sClient, highGroupKey)
 			})