Tweak e2e tests for resource creation to verify that aw stays running (…

…#289)
project-codeflare · Dec 16, 2024 · c8e2c4a · c8e2c4a
1 parent cb88f15
commit c8e2c4a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/e2e/appwrapper_test.go b/test/e2e/appwrapper_test.go
@@ -51,27 +51,32 @@ var _ = Describe("AppWrapper E2E Test", func() {
 			aw := createAppWrapper(ctx, pod(250), pod(250))
 			appwrappers = append(appwrappers, aw)
 			Expect(waitAWPodsReady(ctx, aw)).Should(Succeed())
+			Consistently(AppWrapperPhase(ctx, aw), 5*time.Second).Should(Equal(workloadv1beta2.AppWrapperRunning))
 		})
 		It("Deployments", func() {
 			aw := createAppWrapper(ctx, deployment(2, 200))
 			appwrappers = append(appwrappers, aw)
 			Expect(waitAWPodsReady(ctx, aw)).Should(Succeed())
+			Consistently(AppWrapperPhase(ctx, aw), 5*time.Second).Should(Equal(workloadv1beta2.AppWrapperRunning))
 		})
 		It("StatefulSets", func() {
 			aw := createAppWrapper(ctx, statefulset(2, 200))
 			appwrappers = append(appwrappers, aw)
 			Expect(waitAWPodsReady(ctx, aw)).Should(Succeed())
+			Consistently(AppWrapperPhase(ctx, aw), 5*time.Second).Should(Equal(workloadv1beta2.AppWrapperRunning))
 		})
 		It("Batch Jobs", func() {
 			aw := createAppWrapper(ctx, batchjob(250))
 			appwrappers = append(appwrappers, aw)
 			Expect(waitAWPodsReady(ctx, aw)).Should(Succeed())
+			Consistently(AppWrapperPhase(ctx, aw), 5*time.Second).Should(Equal(workloadv1beta2.AppWrapperRunning))
 		})
 
 		It("Mixed Basic Resources", func() {
 			aw := createAppWrapper(ctx, pod(100), deployment(2, 100), statefulset(2, 100), service(), batchjob(100))
 			appwrappers = append(appwrappers, aw)
 			Expect(waitAWPodsReady(ctx, aw)).Should(Succeed())
+			Consistently(AppWrapperPhase(ctx, aw), 10*time.Second).Should(Equal(workloadv1beta2.AppWrapperRunning))
 		})
 	})