[Fix #1002] Adding test

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

Author: fjtirado

experimental/lambda/src/main/java/io/serverlessworkflow/impl/executors/func/JavaConsumerCallExecutor.java

@@ -42,7 +42,7 @@ public void init(
     inputClass = task.inputClass();
   }
 
-  public CompletableFuture<WorkflowModel> apply(
+  private CompletableFuture<WorkflowModel> apply(
       WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
     T typed = JavaFuncUtils.convertT(input, inputClass);
     consumer.accept(typed);

experimental/lambda/src/main/java/io/serverlessworkflow/impl/executors/func/JavaContextFunctionCallExecutor.java

@@ -35,17 +35,6 @@ public class JavaContextFunctionCallExecutor<T, V>
   private JavaContextFunction<T, V> function;
   private Optional<Class<T>> inputClass;
 
-  public CompletableFuture<WorkflowModel> apply(
-      WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
-    return CompletableFuture.completedFuture(
-        workflowContext
-            .definition()
-            .application()
-            .modelFactory()
-            .fromAny(
-                input, function.apply(JavaFuncUtils.convertT(input, inputClass), workflowContext)));
-  }
-
   @Override
   public boolean accept(Class<? extends TaskBase> clazz) {
     return CallJava.CallJavaContextFunction.class.isAssignableFrom(clazz);
@@ -64,4 +53,15 @@ public void init(
   public CallableTask build() {
     return this::apply;
   }
+
+  private CompletableFuture<WorkflowModel> apply(
+      WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
+    return CompletableFuture.completedFuture(
+        workflowContext
+            .definition()
+            .application()
+            .modelFactory()
+            .fromAny(
+                input, function.apply(JavaFuncUtils.convertT(input, inputClass), workflowContext)));
+  }
 }

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

@@ -16,20 +16,31 @@
 package io.serverlessworkflow.impl.executors;
 
 import io.serverlessworkflow.api.types.CallFunction;
+import io.serverlessworkflow.api.types.FunctionArguments;
 import io.serverlessworkflow.api.types.Task;
 import io.serverlessworkflow.api.types.TaskBase;
 import io.serverlessworkflow.impl.WorkflowDefinition;
 import io.serverlessworkflow.impl.WorkflowMutablePosition;
+import io.serverlessworkflow.impl.WorkflowUtils;
+import io.serverlessworkflow.impl.WorkflowValueResolver;
+import java.util.Map;
 import java.util.Optional;
 
 public class CallFunctionExecutor implements CallableTaskBuilder<CallFunction> {
 
   private TaskExecutorBuilder<? extends TaskBase> executorBuilder;
+  private WorkflowValueResolver<Map<String, Object>> args;
 
   @Override
   public void init(
       CallFunction task, WorkflowDefinition definition, WorkflowMutablePosition position) {
     String functionName = task.getCall();
+    FunctionArguments functionArgs = task.getWith();
+    args =
+        functionArgs != null
+            ? WorkflowUtils.buildMapResolver(
+                definition.application(), functionArgs.getAdditionalProperties())
+            : (w, t, m) -> Map.of();
     Task function = null;
     if (definition.workflow().getUse() != null
         && definition.workflow().getUse().getFunctions() != null
@@ -53,6 +64,12 @@ public boolean accept(Class<? extends TaskBase> clazz) {
   @Override
   public CallableTask build() {
     TaskExecutor<? extends TaskBase> executor = executorBuilder.build();
-    return (w, t, m) -> executor.apply(w, Optional.of(t), m).thenApply(o -> o.output());
+    return (w, t, m) ->
+        executor
+            .apply(
+                w,
+                Optional.of(t),
+                w.definition().application().modelFactory().fromAny(args.apply(w, t, m)))
+            .thenApply(o -> o.output());
   }
 }

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

@@ -121,4 +121,19 @@ void testTimeout() throws IOException {
             .orElseThrow();
     assertThat(result.get("message")).isEqualTo("Viva er Beti Balompie");
   }
+
+  @Test
+  void testCustomFunction() {
+    assertThatThrownBy(
+            () ->
+                app.workflowDefinition(
+                        readWorkflowFromClasspath(
+                            "workflows-samples/call-custom-function-inline.yaml"))
+                    .instance(Map.of())
+                    .start()
+                    .join())
+        .hasCauseInstanceOf(WorkflowException.class)
+        .extracting(w -> ((WorkflowException) w.getCause()).getWorkflowError().status())
+        .isEqualTo(404);
+  }
 }

impl/test/src/test/resources/workflows-samples/call-custom-function-inline.yaml

@@ -1,6 +1,6 @@
 document:
   dsl: '1.0.1'
-  namespace: samples
+  namespace: test
   name: call-custom-function-inline
   version: '0.1.0'
 use:
@@ -12,7 +12,7 @@ use:
             type: object
             properties:
               petId:
-                type: string
+                type: integer
             required: [ petId ]
       call: http
       with: