Fix flaky e2e test

Signed-off-by: Shingo Omura <[email protected]>

Author: everpeace

test/e2e/sparkapplication_test.go

@@ -439,21 +439,27 @@ var _ = Describe("Example SparkApplication", func() {
 			}).WithTimeout(5 * time.Second).Should(Succeed())
 
 			By("Suspending Spark Application")
-			Expect(k8sClient.Get(ctx, key, app)).To(Succeed())
-			app.Spec.Suspend = ptr.To(true)
-			Expect(k8sClient.Update(ctx, app)).To(Succeed())
+			Eventually(func(g Gomega) {
+				app := &v1beta2.SparkApplication{}
+				Expect(k8sClient.Get(ctx, key, app)).To(Succeed())
+				app.Spec.Suspend = ptr.To(true)
+				Expect(k8sClient.Update(ctx, app)).To(Succeed())
+			}).WithTimeout(5 * time.Second).Should(Succeed())
 
 			By("Waiting for SparkApplication to Suspended")
 			Eventually(func(g Gomega) {
 				app := &v1beta2.SparkApplication{}
 				g.Expect(k8sClient.Get(ctx, key, app)).To(Succeed())
 				g.Expect(app.Status.AppState.State).To(Equal(v1beta2.ApplicationStateSuspended))
-			}).WithTimeout(3 * time.Minute).Should(Succeed())
+			}).WithTimeout(1 * time.Minute).Should(Succeed())
 
 			By("Resuming for SparkApplication")
-			Expect(k8sClient.Get(ctx, key, app)).To(Succeed())
-			app.Spec.Suspend = ptr.To(false)
-			Expect(k8sClient.Update(ctx, app)).To(Succeed())
+			Eventually(func(g Gomega) {
+				app := &v1beta2.SparkApplication{}
+				Expect(k8sClient.Get(ctx, key, app)).To(Succeed())
+				app.Spec.Suspend = ptr.To(false)
+				Expect(k8sClient.Update(ctx, app)).To(Succeed())
+			}).WithTimeout(5 * time.Second).Should(Succeed())
 
 			By("Waiting for SparkApplication to Running")
 			Eventually(func(g Gomega) {