fix(*): polishing

core/src/main/java/io/kestra/plugin/core/flow/Sleep.java

@@ -3,7 +3,9 @@
 import io.kestra.core.models.annotations.Example;
 import io.kestra.core.models.annotations.Plugin;
 import io.kestra.core.models.annotations.PluginProperty;
+import io.kestra.core.models.tasks.RunnableTask;
 import io.kestra.core.models.tasks.Task;
+import io.kestra.core.models.tasks.VoidOutput;
 import io.kestra.core.runners.RunContext;
 import io.swagger.v3.oas.annotations.media.Schema;
 import jakarta.validation.constraints.NotNull;
@@ -22,29 +24,29 @@
 @Getter
 @NoArgsConstructor
 @Schema(
-    title = "A task that waits for a specified duration before proceeding."
+    title = "A task that sleep for a specified duration before proceeding."
 )
 @Plugin(
     examples = {
         @Example(
             code = """
-                id: wait
-                type: io.kestra.plugin.core.flow.Wait
-                duration: "PT5S"
-            """
+                    id: sleep
+                    type: io.kestra.plugin.core.flow.Sleep
+                    duration: "PT5S"
+                """
         )
     }
 )
-public class Sleep extends Task {
+public class Sleep extends Task implements RunnableTask<VoidOutput> {
     @Schema(
-        title = "Duration to wait",
+        title = "Duration to sleep",
         description = "The time duration in ISO-8601 format (e.g., `PT5S` for 5 seconds)."
     )
     @PluginProperty
     @NotNull
     private Duration duration;
 
-    public Void run(RunContext runContext) throws Exception {
+    public VoidOutput run(RunContext runContext) throws Exception {
         runContext.logger().info("Waiting for {}", duration);
 
         // Wait for the specified duration

core/src/test/java/io/kestra/plugin/core/flow/SequentialTest.java

@@ -21,15 +21,6 @@ void sequential() throws TimeoutException, QueueException {
         assertThat(execution.getState().getCurrent(), is(State.Type.SUCCESS));
     }
 
-    @Test
-    void sleepTaskTest() throws TimeoutException, QueueException {
-        Execution execution = runnerUtils.runOne(null, "io.kestra.tests", "wait-task-flow");
-
-        // Assert that the execution state is SUCCESS after the wait task completes
-        assertThat(execution.getState().getCurrent(), is(State.Type.SUCCESS));
-    }
-
-
     @Test
     void sequentialWithGlobalErrors() throws TimeoutException, QueueException {
         Execution execution = runnerUtils.runOne(null, "io.kestra.tests", "sequential-with-global-errors");

core/src/test/java/io/kestra/plugin/core/flow/SleepTest.java

@@ -0,0 +1,23 @@
+package io.kestra.plugin.core.flow;
+
+import io.kestra.core.models.executions.Execution;
+import io.kestra.core.models.flows.State;
+import io.kestra.core.queues.QueueException;
+import io.kestra.core.runners.AbstractMemoryRunnerTest;
+import org.junit.jupiter.api.Test;
+
+import java.util.concurrent.TimeoutException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+
+public class SleepTest extends AbstractMemoryRunnerTest {
+    @Test
+    void sleepTaskTest() throws TimeoutException, QueueException {
+        Execution execution = runnerUtils.runOne(null, "io.kestra.tests", "sleep-task-flow");
+
+        assertThat(execution.getState().getCurrent(), is(State.Type.SUCCESS));
+    }
+
+
+}

core/src/test/resources/flows/valids/sleep-task-flow.yaml

@@ -4,11 +4,6 @@ namespace: io.kestra.tests
 tasks:
   - id: sleep
     type: io.kestra.plugin.core.flow.Sleep
-    delay: PT10S # Waits for 10 seconds before proceeding
-
-  - id: next-task
-    type: io.kestra.plugin.core.scripts.Shell
-    commands:
-      - echo "Wait task completed"
+    duration: PT2S
 
 

core/src/test/resources/flows/valids/sleep.yml

@@ -3,5 +3,5 @@ namespace: io.kestra.tests
 
 tasks:
   - id: sleep
-    type: io.kestra.plugin.core.flow.Sleep
+    type: io.kestra.core.tasks.test.Sleep
     duration: 1000