test and ajust in classes

Signed-off-by: Matheus André Galvão Da Silva <matheusandr2@gmail.com>

Author: matheusandre1

experimental/lambda/src/test/java/io/serverless/workflow/impl/executors/func/FuncDSLDataFlowTransformationHelpersTest.java

@@ -25,6 +25,7 @@
 import io.serverlessworkflow.impl.WorkflowApplication;
 import io.serverlessworkflow.impl.WorkflowContextData;
 import io.serverlessworkflow.impl.WorkflowDefinition;
+import io.serverlessworkflow.impl.WorkflowInstance;
 import io.serverlessworkflow.impl.WorkflowModel;
 import org.assertj.core.api.SoftAssertions;
 import org.junit.jupiter.api.Test;
@@ -160,4 +161,82 @@ void test_input_with_inputFrom_fluent_way() {
 
     softly.assertAll();
   }
+
+  @Test
+  void test_output_with_outputAs() {
+
+    SoftAssertions softly = new SoftAssertions();
+
+    Workflow workflow =
+        FuncWorkflowBuilder.workflow("enrichOutputWithTaskOutputTest")
+            .tasks(
+                function(
+                        "add5",
+                        (Long input) -> {
+                          softly.assertThat(input).isEqualTo(10L);
+                          return input + 5;
+                        },
+                        Long.class)
+                    .outputAs(
+                        (object, workflowContext, taskContextData) -> {
+                          Long taskOutput = output(taskContextData, Long.class);
+                          softly.assertThat(taskOutput).isEqualTo(15L);
+                          Long input = input(workflowContext, Long.class);
+                          softly.assertThat(input).isEqualTo(10L);
+                          return input + taskOutput;
+                        },
+                        Long.class))
+            .build();
+
+    try (WorkflowApplication app = WorkflowApplication.builder().build()) {
+      WorkflowDefinition def = app.workflowDefinition(workflow);
+
+      WorkflowModel model = def.instance(10L).start().join();
+      Number number = model.asNumber().orElseThrow();
+
+      softly.assertThat(number.longValue()).isEqualTo(25L);
+    }
+
+    softly.assertAll();
+  }
+
+  @Test
+  void test_input_with_exportAs() {
+
+    SoftAssertions softly = new SoftAssertions();
+
+    Workflow workflow =
+        FuncWorkflowBuilder.workflow("enrichExportWithInputTest")
+            .tasks(
+                function(
+                        "add5",
+                        (Long input) -> {
+                          softly.assertThat(input).isEqualTo(10L);
+                          return input + 5;
+                        },
+                        Long.class)
+                    .exportAs(
+                        (Long object,
+                            WorkflowContextData workflowContext,
+                            TaskContextData taskContextData) -> {
+                          Long taskOutput = output(taskContextData, Long.class);
+                          softly.assertThat(taskOutput).isEqualTo(15L);
+                          Long input = input(workflowContext, Long.class);
+                          softly.assertThat(input).isEqualTo(10L);
+                          return input + taskOutput;
+                        }))
+            .build();
+
+    try (WorkflowApplication app = WorkflowApplication.builder().build()) {
+      WorkflowDefinition def = app.workflowDefinition(workflow);
+
+      WorkflowInstance instance = def.instance(10L);
+      instance.start().join();
+      Number number = instance.context().asNumber().orElseThrow();
+
+      softly.assertThat(number.longValue()).isEqualTo(25L);
+    }
+
+    softly.assertAll();
+  }
 }

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/BaseTaskItemListBuilder.java

@@ -44,6 +44,7 @@ public abstract class BaseTaskItemListBuilder<SELF extends BaseTaskItemListBuild
   protected final String TYPE_TRY = "try";
   protected final String TYPE_HTTP = "http";
   protected final String TYPE_OPENAPI = "openapi";
+  protected final String TYPE_WORKFLOW = "workflow";
 
   private final List<TaskItem> list;
   private final int offset;

fluent/spec/src/main/java/io/serverlessworkflow/fluent/spec/TaskItemListBuilder.java

@@ -147,7 +147,7 @@ public TaskItemListBuilder openapi(
 
   @Override
   public TaskItemListBuilder workflow(String name, Consumer<WorkflowTaskBuilder> itemsConfigurer) {
-    name = defaultNameAndRequireConfig(name, itemsConfigurer);
+    name = defaultNameAndRequireConfig(name, itemsConfigurer, TYPE_WORKFLOW);
 
     final WorkflowTaskBuilder workflowTaskBuilder = new WorkflowTaskBuilder();
     itemsConfigurer.accept(workflowTaskBuilder);

fluent/spec/src/test/java/io/serverlessworkflow/fluent/spec/dsl/DSLTest.java

@@ -330,6 +330,9 @@ void when_dsl_subflow_workflow_task_with_map_input() {
     assertThat(run.getWorkflow().getInput().getAdditionalProperties().get("extra")).isEqualTo(true);
     assertThat(run.isAwait()).isTrue();
     assertThat(run.getReturn()).isEqualTo(RunTaskConfiguration.ProcessReturnType.NONE);
+  }
+
+  @Test
   public void when_call_openapi_with_explicit_name() {
     Workflow wf =
         WorkflowBuilder.workflow("myFlow", "myNs", "1.2.3")