[Fix #1018] Initializing context (#1019)

Signed-off-by: fjtirado <ftirados@redhat.com>

Author: fjtirado

experimental/lambda/src/test/java/io/serverless/workflow/impl/CallTest.java

@@ -116,7 +116,7 @@ void testSwitch() throws InterruptedException, ExecutionException {
       Workflow workflow =
           new Workflow()
               .withDocument(
-                  new Document().withNamespace("test").withName("testSwith").withVersion("1.0"))
+                  new Document().withNamespace("test").withName("testSwitch").withVersion("1.0"))
               .withDo(
                   List.of(
                       new TaskItem(

experimental/lambda/src/test/java/io/serverless/workflow/impl/FluentDSLCallTest.java

@@ -15,13 +15,15 @@
  */
 package io.serverless.workflow.impl;
 
+import static io.serverlessworkflow.fluent.func.dsl.FuncDSL.function;
 import static org.assertj.core.api.Assertions.assertThat;
 
 import io.serverlessworkflow.api.types.FlowDirectiveEnum;
 import io.serverlessworkflow.api.types.Workflow;
 import io.serverlessworkflow.fluent.func.FuncWorkflowBuilder;
 import io.serverlessworkflow.impl.WorkflowApplication;
 import io.serverlessworkflow.impl.WorkflowDefinition;
+import io.serverlessworkflow.impl.WorkflowInstance;
 import java.util.Collection;
 import java.util.List;
 import java.util.concurrent.ExecutionException;
@@ -47,6 +49,23 @@ void testJavaFunction() throws InterruptedException, ExecutionException {
     }
   }
 
+  @Test
+  void textExportFunction() {
+    try (WorkflowApplication app = WorkflowApplication.builder().build()) {
+      final Workflow workflow =
+          FuncWorkflowBuilder.workflow("testExportFunction")
+              .tasks(
+                  function(JavaFunctions::getName, Person.class)
+                      .exportAs(
+                          (prevContext, w, t) -> t.output().asText().orElseThrow() + " Tirado"))
+              .build();
+      WorkflowInstance instance =
+          app.workflowDefinition(workflow).instance(new Person("Francisco", 33));
+      assertThat(instance.start().join().asText().orElseThrow()).isEqualTo("Francisco Javierito");
+      assertThat(instance.context().asText().orElseThrow()).isEqualTo("Francisco Javierito Tirado");
+    }
+  }
+
   @Test
   void testForLoop() throws InterruptedException, ExecutionException {
     try (WorkflowApplication app = WorkflowApplication.builder().build()) {

impl/core/src/main/java/io/serverlessworkflow/impl/WorkflowContext.java

@@ -23,6 +23,7 @@ public class WorkflowContext implements WorkflowContextData {
   WorkflowContext(WorkflowDefinition definition, WorkflowMutableInstance instance) {
     this.definition = definition;
     this.instance = instance;
+    this.context = definition.application().modelFactory().fromNull();
   }
 
   @Override

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

@@ -25,12 +25,14 @@
 import io.serverlessworkflow.impl.WorkflowApplication;
 import io.serverlessworkflow.impl.WorkflowDefinition;
 import io.serverlessworkflow.impl.WorkflowException;
+import io.serverlessworkflow.impl.WorkflowInstance;
 import io.serverlessworkflow.impl.WorkflowModel;
 import io.serverlessworkflow.impl.jackson.JsonUtils;
 import java.io.IOException;
 import java.time.Instant;
 import java.util.Arrays;
 import java.util.Map;
+import java.util.Optional;
 import java.util.concurrent.CompletionException;
 import java.util.function.Consumer;
 import java.util.stream.Stream;
@@ -76,7 +78,8 @@ private static Stream<Arguments> provideParameters() {
         args(
             "workflows-samples/for-sum.yaml",
             Map.of("input", Arrays.asList(1, 2, 3)),
-            o -> assertThat(o).isEqualTo(6)),
+            o -> assertThat(o).isEqualTo(6),
+            c -> assertThat(c).isEqualTo(Map.of("incr", Arrays.asList(2, 3, 4)))),
         args(
             "workflows-samples/switch-then-loop.yaml",
             Map.of("count", 1),
@@ -116,16 +119,36 @@ private static Stream<Arguments> provideParameters() {
   }
 
   private static Arguments args(
-      String fileName, Map<String, Object> input, Consumer<Object> instance) {
+      String fileName,
+      Map<String, Object> input,
+      Consumer<Object> modelAssert,
+      Optional<Consumer<Object>> context) {
     return Arguments.of(
         fileName,
         (Consumer<WorkflowDefinition>)
-            d ->
-                instance.accept(
-                    d.instance(input)
-                        .start()
-                        .thenApply(model -> JsonUtils.toJavaValue(JsonUtils.modelToJson(model)))
-                        .join()));
+            d -> {
+              WorkflowInstance instance = d.instance(input);
+              modelAssert.accept(
+                  instance
+                      .start()
+                      .thenApply(model -> JsonUtils.toJavaValue(JsonUtils.modelToJson(model)))
+                      .join());
+              context.ifPresent(
+                  c -> c.accept(JsonUtils.toJavaValue(JsonUtils.fromValue(instance.context()))));
+            });
+  }
+
+  private static Arguments args(
+      String fileName,
+      Map<String, Object> input,
+      Consumer<Object> instance,
+      Consumer<Object> context) {
+    return args(fileName, input, instance, Optional.of(context));
+  }
+
+  private static Arguments args(
+      String fileName, Map<String, Object> input, Consumer<Object> instance) {
+    return args(fileName, input, instance, Optional.empty());
   }
 
   private static Arguments argsJson(

impl/test/src/test/resources/workflows-samples/for-sum.yaml

@@ -12,5 +12,8 @@ do:
         - accumulate:
             set:
               counter: ${.counter+$number}
+            export: 
+             as: if .incr==null then {incr:[$number+1]} else .incr+=[$number+1] end 
       output:
         as: .counter
+