[Fix #897] Fixing flaky test

Signed-off-by: fjtirado <[email protected]>

Author: fjtirado

impl/core/src/main/java/io/serverlessworkflow/impl/executors/RunScriptExecutor.java

@@ -102,22 +102,24 @@ public void init(RunScript taskConfiguration, WorkflowDefinition definition) {
   @Override
   public CompletableFuture<WorkflowModel> apply(
       WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
-    return CompletableFuture.supplyAsync(
-        () ->
-            taskRunner.runScript(
-                new ScriptContext(
-                    argumentExpr
-                        .map(m -> m.apply(workflowContext, taskContext, input))
-                        .orElse(Map.of()),
-                    environmentExpr
-                        .map(m -> m.apply(workflowContext, taskContext, input))
-                        .orElse(Map.of()),
-                    codeSupplier.apply(workflowContext, taskContext, input),
-                    isAwait,
-                    returnType),
-                workflowContext,
-                taskContext,
-                input));
+    ScriptContext scriptContext =
+        new ScriptContext(
+            argumentExpr.map(m -> m.apply(workflowContext, taskContext, input)).orElse(Map.of()),
+            environmentExpr.map(m -> m.apply(workflowContext, taskContext, input)).orElse(Map.of()),
+            codeSupplier.apply(workflowContext, taskContext, input),
+            returnType);
+    if (isAwait) {
+      return CompletableFuture.supplyAsync(
+          () -> taskRunner.runScript(scriptContext, workflowContext, taskContext, input),
+          workflowContext.definition().application().executorService());
+    } else {
+      workflowContext
+          .definition()
+          .application()
+          .executorService()
+          .submit(() -> taskRunner.runScript(scriptContext, workflowContext, taskContext, input));
+      return CompletableFuture.completedFuture(input);
+    }
   }
 
   @Override

impl/core/src/main/java/io/serverlessworkflow/impl/scripts/ScriptContext.java

@@ -22,5 +22,4 @@ public record ScriptContext(
     Map<String, Object> args,
     Map<String, Object> envs,
     String code,
-    boolean isAwait,
     RunTaskConfiguration.ProcessReturnType returnType) {}

impl/script-js/src/main/java/io/serverlessworkflow/impl/executors/script/js/JavaScriptScriptTaskRunner.java

@@ -55,7 +55,6 @@ public WorkflowModel runScript(
     WorkflowApplication application = workflowContext.definition().application();
     ByteArrayOutputStream stderr = new ByteArrayOutputStream();
     ByteArrayOutputStream stdout = new ByteArrayOutputStream();
-
     try (Context ctx =
         Context.newBuilder()
             .err(stderr)
@@ -71,21 +70,8 @@ public WorkflowModel runScript(
               (key, val) -> {
                 ctx.getBindings(identifier().getLang()).putMember(key, val);
               });
-
       configureProcessEnv(ctx, script.envs());
-
-      if (!script.isAwait()) {
-        application
-            .executorService()
-            .submit(
-                () -> {
-                  ctx.eval(identifier().getLang(), script.code());
-                });
-        return application.modelFactory().fromAny(input);
-      }
-
       ctx.eval(Source.create(identifier().getLang(), script.code()));
-
       return modelFromOutput(
           script.returnType(), application.modelFactory(), stdout, () -> stderr.toString());
     } catch (PolyglotException e) {

impl/test/src/test/java/io/serverlessworkflow/impl/test/RunShellExecutorTest.java

@@ -21,8 +21,6 @@
 import io.serverlessworkflow.impl.WorkflowModel;
 import io.serverlessworkflow.impl.executors.ProcessResult;
 import java.io.IOException;
-import java.nio.file.Files;
-import java.nio.file.Path;
 import java.util.Map;
 import org.assertj.core.api.SoftAssertions;
 import org.junit.jupiter.api.Test;
@@ -127,15 +125,10 @@ void testAwaitBehavior() throws IOException {
             "workflows-samples/run-shell/echo-not-awaiting.yaml");
     try (WorkflowApplication appl = WorkflowApplication.builder().build()) {
       Map<String, String> inputMap = Map.of("full_name", "Matheus Cruz");
-
       WorkflowModel outputModel =
           appl.workflowDefinition(workflow).instance(inputMap).start().join();
-
-      String content = Files.readString(Path.of("/tmp/hello.txt"));
-
       SoftAssertions.assertSoftly(
           softly -> {
-            softly.assertThat(content).contains("hello world not awaiting (Matheus Cruz)");
             softly.assertThat(outputModel.asMap().get()).isEqualTo(inputMap);
           });
     }