fix(test): backport flacky tests fixed on develop

Author: nKwiatkowski

core/src/test/java/io/kestra/core/schedulers/SchedulerScheduleOnDatesTest.java

@@ -53,9 +53,9 @@ public class SchedulerScheduleOnDatesTest extends AbstractSchedulerTest {
 
     private Flow createScheduleFlow(String zone, String triggerId) {
         var now = ZonedDateTime.now();
-        var before = now.minusSeconds(1).truncatedTo(ChronoUnit.SECONDS);
-        var after = now.plusSeconds(1).truncatedTo(ChronoUnit.SECONDS);
-        var later = now.plusSeconds(2).truncatedTo(ChronoUnit.SECONDS);
+        var before = now.minusSeconds(3).truncatedTo(ChronoUnit.SECONDS);
+        var after = now.plusSeconds(3).truncatedTo(ChronoUnit.SECONDS);
+        var later = now.plusSeconds(6).truncatedTo(ChronoUnit.SECONDS);
         ScheduleOnDates schedule = createScheduleOnDatesTrigger(zone, List.of(before, after, later), triggerId).build();
 
         return createFlow(Collections.singletonList(schedule));
@@ -106,11 +106,11 @@ void scheduleOnDates() throws Exception {
                 date.add((String) execution.getTrigger().getVariables().get("date"));
                 executionId.add(execution.getId());
 
-                queueCount.countDown();
                 if (execution.getState().getCurrent() == State.Type.CREATED) {
                     executionQueue.emit(execution.withState(State.Type.SUCCESS));
                 }
                 assertThat(execution.getFlowId(), is(flow.getId()));
+                queueCount.countDown();
             }));
 
             scheduler.run();

jdbc/src/test/java/io/kestra/jdbc/runner/JdbcServiceLivenessCoordinatorTest.java

@@ -243,14 +243,11 @@ void shouldReEmitTriggerWhenWorkerIsDetectedAsNonResponding() throws Exception {
         worker.shutdown();
 
         Worker newWorker = applicationContext.createBean(Worker.class, IdUtils.create(), 1, null);
-        applicationContext.registerSingleton(newWorker);
         newWorker.run();
+        assertThat(countDownLatch.await(30, TimeUnit.SECONDS), is(true));
 
-        boolean lastAwait = countDownLatch.await(10, TimeUnit.SECONDS);
-
-        newWorker.shutdown();
         receive.blockLast();
-        assertThat(lastAwait, is(true));
+        newWorker.shutdown();
     }
 
     @Test
@@ -260,11 +257,11 @@ void shouldReEmitTriggerToTheSameWorkerGroup() throws Exception {
 
         WorkerTrigger workerTrigger = workerTrigger(Duration.ofSeconds(5), "workerGroupKey");
 
-        // 2 trigger should happen because of the resubmit
+        // 2 triggers should happen because of the resubmit
         CountDownLatch countDownLatch = new CountDownLatch(2);
         Flux<WorkerTriggerResult> receive = TestsUtils.receive(workerTriggerResultQueue, workerTriggerResult -> countDownLatch.countDown());
 
-        // we wait that the worker receive the trigger
+        // we wait that the worker receives the trigger
         CountDownLatch triggerCountDownLatch = new CountDownLatch(1);
         Flux<Trigger> receiveTrigger = TestsUtils.receive(triggerQueue, either -> {
             if (either.getLeft().getWorkerId().equals(worker.getId())) {
@@ -277,14 +274,11 @@ void shouldReEmitTriggerToTheSameWorkerGroup() throws Exception {
         worker.shutdown();
 
         Worker newWorker = applicationContext.createBean(Worker.class, IdUtils.create(), 1, "workerGroupKey");
-        applicationContext.registerSingleton(newWorker);
         newWorker.run();
+        assertThat(countDownLatch.await(30, TimeUnit.SECONDS), is(true));
 
-        boolean lastAwait = countDownLatch.await(10, TimeUnit.SECONDS);
-
-        newWorker.shutdown();
         receive.blockLast();
-        assertThat(lastAwait, is(true));
+        newWorker.shutdown();
     }
 
     @MockBean(WorkerGroupService.class)