@@ -452,9 +452,18 @@ public void a_second_daily_repair_schedule_is_added_for_and_keyspace(String clus
452
452
public void reaper_has_scheduled_repairs_for_cluster_called (int repairAmount , String clusterName ) throws Throwable {
453
453
synchronized (BasicSteps .class ) {
454
454
CLIENTS .parallelStream ().forEach (client -> {
455
- List <RepairScheduleStatus > schedules = client .getRepairSchedulesForCluster (clusterName );
456
- LOG .info ("Got " + schedules .size () + " schedules" );
457
- assertEquals (repairAmount , schedules .size ());
455
+
456
+ await ().with ().pollInterval (1 , SECONDS ).atMost (1 , MINUTES ).until (() -> {
457
+ try {
458
+ List <RepairScheduleStatus > schedules = client .getRepairSchedulesForCluster (clusterName );
459
+ LOG .info ("Got " + schedules .size () + " schedules" );
460
+ assertEquals (repairAmount , schedules .size ());
461
+ } catch (AssertionError ex ) {
462
+ LOG .warn (ex .getMessage ());
463
+ return false ;
464
+ }
465
+ return true ;
466
+ });
458
467
});
459
468
}
460
469
}
@@ -464,9 +473,17 @@ public void reaper_has_scheduled_repairs_for_cluster_called(int repairAmount, St
464
473
public void reaper_has_scheduled_repairs_for_the_last_added_cluster (int repairAmount ) throws Throwable {
465
474
synchronized (BasicSteps .class ) {
466
475
CLIENTS .parallelStream ().forEach (client -> {
467
- List <RepairScheduleStatus > schedules = client .getRepairSchedulesForCluster (TestContext .TEST_CLUSTER );
468
- LOG .info ("Got " + schedules .size () + " schedules" );
469
- assertEquals (repairAmount , schedules .size ());
476
+ await ().with ().pollInterval (1 , SECONDS ).atMost (1 , MINUTES ).until (() -> {
477
+ try {
478
+ List <RepairScheduleStatus > schedules = client .getRepairSchedulesForCluster (TestContext .TEST_CLUSTER );
479
+ LOG .info ("Got " + schedules .size () + " schedules" );
480
+ assertEquals (repairAmount , schedules .size ());
481
+ } catch (AssertionError ex ) {
482
+ LOG .warn (ex .getMessage ());
483
+ return false ;
484
+ }
485
+ return true ;
486
+ });
470
487
});
471
488
}
472
489
}
0 commit comments