Make asserts on schedules existing (having been created) eventual, as…

… this is done by design eventually.
thelastpickle · Sep 13, 2017 · 3699509 · 3699509
1 parent d85d438
commit 3699509
Showing 1 changed file with 23 additions and 6 deletions.
diff --git a/src/server/src/test/java/com/spotify/reaper/acceptance/BasicSteps.java b/src/server/src/test/java/com/spotify/reaper/acceptance/BasicSteps.java
@@ -452,9 +452,18 @@ public void a_second_daily_repair_schedule_is_added_for_and_keyspace(String clus
   public void reaper_has_scheduled_repairs_for_cluster_called(int repairAmount, String clusterName) throws Throwable {
     synchronized (BasicSteps.class) {
       CLIENTS.parallelStream().forEach(client -> {
-        List<RepairScheduleStatus> schedules = client.getRepairSchedulesForCluster(clusterName);
-        LOG.info("Got " + schedules.size() + " schedules");
-        assertEquals(repairAmount, schedules.size());
+
+        await().with().pollInterval(1, SECONDS).atMost(1, MINUTES).until(() -> {
+            try {
+                List<RepairScheduleStatus> schedules = client.getRepairSchedulesForCluster(clusterName);
+                LOG.info("Got " + schedules.size() + " schedules");
+                assertEquals(repairAmount, schedules.size());
+            } catch (AssertionError ex) {
+                LOG.warn(ex.getMessage());
+                return false;
+            }
+            return true;
+        });
       });
     }
   }
@@ -464,9 +473,17 @@ public void reaper_has_scheduled_repairs_for_cluster_called(int repairAmount, St
   public void reaper_has_scheduled_repairs_for_the_last_added_cluster(int repairAmount) throws Throwable {
     synchronized (BasicSteps.class) {
       CLIENTS.parallelStream().forEach(client -> {
-        List<RepairScheduleStatus> schedules = client.getRepairSchedulesForCluster(TestContext.TEST_CLUSTER);
-        LOG.info("Got " + schedules.size() + " schedules");
-        assertEquals(repairAmount, schedules.size());
+        await().with().pollInterval(1, SECONDS).atMost(1, MINUTES).until(() -> {
+            try {
+                List<RepairScheduleStatus> schedules = client.getRepairSchedulesForCluster(TestContext.TEST_CLUSTER);
+                LOG.info("Got " + schedules.size() + " schedules");
+                assertEquals(repairAmount, schedules.size());
+            } catch (AssertionError ex) {
+                LOG.warn(ex.getMessage());
+                return false;
+            }
+            return true;
+        });
       });
     }
   }