feat(tests): add test coverage for replaying LoopUntil

Author: MTarek165

core/src/test/java/io/kestra/core/runners/AbstractRunnerTest.java

@@ -193,8 +193,8 @@ protected void restartFailedWithAfterExecution() throws Exception {
 
     @Test
     @LoadFlows({"flows/valids/loop-until-restart.yaml"})
-    protected void restartLoopUntil() throws Exception{
-        restartCaseTest.restartLoopUntil();
+    protected void restartOrReplayLoopUntil() throws Exception{
+        restartCaseTest.restartOrReplayLoopUntil();
     }
 
     @Test

core/src/test/java/io/kestra/core/runners/RestartCaseTest.java

@@ -286,13 +286,13 @@ public void restartFailedWithAfterExecution() throws Exception {
     }
 
 
-    public void restartLoopUntil() throws Exception{
+    public void restartOrReplayLoopUntil() throws Exception{
         Flow flow = flowRepository.findById(MAIN_TENANT, "io.kestra.tests", "loop-until-restart").orElseThrow();
 
         Execution firstExecution = runnerUtils.runOne(MAIN_TENANT, flow.getNamespace(), flow.getId(), Duration.ofSeconds(60));
 
         assertThat(firstExecution.getState().getCurrent()).isEqualTo(Type.FAILED);
-
+         // restarting case
         Execution restartedExecution = executionService.restart(firstExecution, null);
         assertThat(restartedExecution).isNotNull();
         assertThat(restartedExecution.getId()).isEqualTo(firstExecution.getId());
@@ -301,14 +301,37 @@ public void restartLoopUntil() throws Exception{
         Execution finalRestartedExecution = runnerUtils.restartExecution( execution -> execution.getState().isFailed(), restartedExecution);
         assertThat(finalRestartedExecution.getState().getCurrent()).isEqualTo(Type.FAILED);
 
-        Optional<TaskRun> parentTaskRun = finalRestartedExecution.findTaskRunsByTaskId("loop_test").stream().findFirst();
-        assertThat(parentTaskRun.isPresent());
+        Optional<TaskRun> parentTaskRun1 = finalRestartedExecution.findTaskRunsByTaskId("loop_test").stream().findFirst();
+        assertThat(parentTaskRun1.isPresent());
 
-        State.History lastFailed = parentTaskRun.get().getState().getHistories().getLast();
-        State.History lastRestarted = parentTaskRun.get().getState().getHistories().reversed().stream()
+        State.History lastFailed1 = parentTaskRun1.get().getState().getHistories().getLast();
+        State.History lastRestarted1 = parentTaskRun1.get().getState().getHistories().reversed().stream()
             .filter(history -> history.getState() == Type.RESTARTED).findFirst().get();
-        assertThat(lastRestarted).isNotNull();
-        assertThat(lastRestarted.getDate().plus(10, ChronoUnit.SECONDS).isBefore(lastFailed.getDate()));
+        assertThat(lastRestarted1).isNotNull();
+        assertThat(lastRestarted1.getDate().plus(10, ChronoUnit.SECONDS).isBefore(lastFailed1.getDate()));
+
+        // replaying case
+        Execution replayedExecution = executionService.replay(firstExecution, firstExecution.findTaskRunByTaskIdAndValue("loop_test", List.of()).getId(), null);
+        assertThat(replayedExecution.getState().getCurrent()).isEqualTo(Type.RESTARTED);
+        assertThat(replayedExecution.getId()).isNotEqualTo(firstExecution.getId());
+
+        Execution finalReplayedExecution = runnerUtils.awaitChildExecution(
+            flow,
+            firstExecution,
+            replayedExecution,
+            Duration.ofSeconds(60)
+        );
+        assertThat(finalReplayedExecution.getState().getCurrent()).isEqualTo(Type.FAILED);
+
+        Optional<TaskRun> parentTaskRun2 = finalReplayedExecution.findTaskRunsByTaskId("loop_test").stream().findFirst();
+        assertThat(parentTaskRun2.isPresent());
+
+        State.History lastFailed2 = parentTaskRun2.get().getState().getHistories().getLast();
+        State.History lastRestarted2 = parentTaskRun2.get().getState().getHistories().reversed().stream()
+            .filter(history -> history.getState() == Type.RESTARTED).findFirst().get();
+        assertThat(lastRestarted2).isNotNull();
+        assertThat(lastRestarted2.getDate().plus(10, ChronoUnit.SECONDS).isBefore(lastFailed2.getDate()));
+
     }
 
 }