chore: update job tests with retries (GoogleCloudPlatform#6861)

averikitsch · web-flow · commit c5df4758e5fb · 2022-03-08T13:31:20.000-08:00
* chore: update job tests with retries

* minor fixes

* Update
diff --git a/run/jobs/src/test/java/com/example/JobsIntegrationTests.java b/run/jobs/src/test/java/com/example/JobsIntegrationTests.java
@@ -104,15 +104,23 @@ public void generatesLogs() throws Exception {
               + "\" -protoPayload.serviceName=\"run.googleapis.com\"";
 
       System.out.println(logFilter);
-      Page<LogEntry> entries = logging.listLogEntries(EntryListOption.filter(logFilter));
       Boolean found = false;
-      for (LogEntry logEntry : entries.iterateAll()) {
-        if (!logEntry.getLogName().contains("cloudaudit")) {
-          Payload<String> payload = logEntry.getPayload();
-          if (payload.getData().contains("Task")) { 
-            found = true; 
+      // Retry up to 5 times
+      for (int i = 1; i <= 5; i++) {
+        Page<LogEntry> entries = logging.listLogEntries(EntryListOption.filter(logFilter));
+        for (LogEntry logEntry : entries.iterateAll()) {
+          if (!logEntry.getLogName().contains("cloudaudit")) {
+            Payload<String> payload = logEntry.getPayload();
+            if (payload.getData().contains("Task")) { 
+              found = true;
+              break;
+            }
           }
         }
+        if (found) {
+          break;
+        }
+        TimeUnit.SECONDS.sleep(i * 30);
       }
       assertTrue("Log was not found.", found);
     }
