fix: SampleTests: customAttributeSampleTest (#2411)

GoogleCloudPlatform · Mar 13, 2020 · 34fbfe3 · 34fbfe3
1 parent 9612e8b
commit 34fbfe3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/jobs/v3/src/test/java/SampleTests.java b/jobs/v3/src/test/java/SampleTests.java
@@ -39,6 +39,7 @@
 public class SampleTests {
 
   private static ByteArrayOutputStream bout;
+  private long timeInMillis;
 
   @BeforeClass
   public static void setUp() {
@@ -111,6 +112,13 @@ public void commuteSearchSampleTest() throws Exception {
   @Test
   public void customAttributeSampleTest() throws Exception {
     CustomAttributeSample.main();
+
+    // wait for 10 seconds to elapse and then run it.
+    timeInMillis = System.currentTimeMillis();
+    while (System.currentTimeMillis() < timeInMillis + 10000) {
+      Thread.sleep(1000);
+    }
+
     assertThat(bout.toString())
         .containsMatch(
             ".*Job created:.*jobWithACustomAttribute.*\n"