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 index 397515af2..575492ff9 100644 --- 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 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 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 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 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; + }); }); } }