Include Actions of ForEachState for GetFunctionDefinitionForAction

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

Author: manick02

utils/src/main/java/io/serverlessworkflow/utils/WorkflowUtils.java

@@ -318,6 +318,18 @@ private static FunctionRef getFunctionRefFromAction(Workflow workflow, String ac
             }
           }
         }
+      } else if (state instanceof ForEachState) {
+        ForEachState forEachState = (ForEachState) state;
+        List<Action> forEachStateActions = forEachState.getActions();
+        if (forEachStateActions != null) {
+          for (Action forEachStateAction : forEachStateActions) {
+            if (forEachStateAction != null
+                && forEachStateAction.getName() != null
+                && forEachStateAction.getName().equals(action)) {
+              return forEachStateAction.getFunctionRef();
+            }
+          }
+        }
       }
     }