GetFunctionDefs for action #127

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

Author: manick02

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

@@ -19,6 +19,7 @@
 import io.serverlessworkflow.api.actions.Action;
 import io.serverlessworkflow.api.branches.Branch;
 import io.serverlessworkflow.api.events.EventDefinition;
+import io.serverlessworkflow.api.functions.FunctionDefinition;
 import io.serverlessworkflow.api.interfaces.State;
 import io.serverlessworkflow.api.start.Start;
 import io.serverlessworkflow.api.states.*;
@@ -157,12 +158,12 @@ private static List<EventDefinition> getWorkflowEventDefinitions(
 
     List<String> uniqueWorkflowEventsFromStates = getUniqueWorkflowEventsFromStates(workflow);
     List<EventDefinition> definedConsumedEvents = getDefinedEvents(workflow, eventKind);
-    if(definedConsumedEvents == null) {
+    if (definedConsumedEvents == null) {
       return null;
     }
     return definedConsumedEvents.stream()
-            .filter(definedEvent -> uniqueWorkflowEventsFromStates.contains(definedEvent.getName()))
-            .collect(Collectors.toList());
+        .filter(definedEvent -> uniqueWorkflowEventsFromStates.contains(definedEvent.getName()))
+        .collect(Collectors.toList());
   }
 
   /** Returns a list of unique event names from workflow states */
@@ -240,6 +241,24 @@ public static int getWorkflowProducedEventsCount(Workflow workflow) {
     return workflowProducedEvents == null ? 0 : workflowProducedEvents.size();
   }
 
+  /** @return Returns function definition for actions */
+  public static List<FunctionDefinition> getFunctionDefinitionsForAction(
+      Workflow workflow, String action) {
+    if (hasFunctionDefs(workflow)) {
+      return workflow.getFunctions().getFunctionDefs().stream()
+          .filter(functionDef -> functionDef.getName().equals(action))
+          .collect(Collectors.toList());
+    }
+    return null;
+  }
+
+  private static boolean hasFunctionDefs(Workflow workflow) {
+    return workflow != null
+        && workflow.getFunctions() != null
+        && workflow.getFunctions().getFunctionDefs() != null
+        && !workflow.getFunctions().getFunctionDefs().isEmpty();
+  }
+
   /** Returns true if workflow has states, otherwise false */
   private static boolean hasStates(Workflow workflow) {
     return workflow != null && workflow.getStates() != null && !workflow.getStates().isEmpty();

utils/src/test/java/io/serverlessworkflow/util/FunctionDefinitionsTest.java

@@ -0,0 +1,59 @@
+/*
+ * Copyright 2020-Present The Serverless Workflow Specification Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.serverlessworkflow.util;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+
+import io.serverlessworkflow.api.Workflow;
+import io.serverlessworkflow.api.functions.FunctionDefinition;
+import io.serverlessworkflow.util.testutil.TestUtils;
+import io.serverlessworkflow.utils.WorkflowUtils;
+import java.util.List;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.ValueSource;
+
+public class FunctionDefinitionsTest {
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/funcdefinitiontest/functiondefinition.yml"})
+  public void testFunctionDefsForAction(String funcDefinitions) {
+    String actionLookUp = "finalizeApplicationFunction";
+    int expectedCount = 1;
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(funcDefinitions);
+    List<FunctionDefinition> finalizeApplicationFunction =
+        WorkflowUtils.getFunctionDefinitionsForAction(workflow, actionLookUp);
+    assertEquals(expectedCount, finalizeApplicationFunction.size());
+  }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/funcdefinitiontest/functiondefinition.yml"})
+  public void testFunctionDefsForActionNotPresent(String funcDefinitions) {
+    String actionLookUp = "finalizeApplicationFunctionNotPresent";
+    int expectedCount = 0;
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(funcDefinitions);
+    List<FunctionDefinition> finalizeApplicationFunction =
+        WorkflowUtils.getFunctionDefinitionsForAction(workflow, actionLookUp);
+    assertEquals(expectedCount, finalizeApplicationFunction.size());
+  }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/funcdefinitiontest/functiondefinition.yml"})
+  public void testFunctionDefsForNullWorkflow(String funcDefinitions) {
+    assertNull(WorkflowUtils.getFunctionDefinitionsForAction(null, "TestAction"));
+  }
+}

utils/src/test/resources/funcdefinitiontest/functiondefinition.yml

@@ -0,0 +1,56 @@
+id: finalizeCollegeApplication
+name: Finalize College Application
+version: '1.0'
+specVersion: '0.7'
+events:
+  - name: ApplicationSubmitted
+    type: org.application.submitted
+    source: applicationsource
+    kind: produced
+    correlation:
+      - contextAttributeName: applicantId
+  - name: SATScoresReceived
+    type: org.application.satscores
+    source: applicationsource
+    correlation:
+      - contextAttributeName: applicantId
+  - name: RecommendationLetterReceived
+    type: org.application.recommendationLetter
+    source: applicationsource
+    correlation:
+      - contextAttributeName: applicantId
+functions:
+  - name: finalizeApplicationFunction
+    operation: http://myapis.org/collegeapplicationapi.json#finalize
+states:
+  - name: FinalizeApplication
+    type: event
+    exclusive: false
+    onEvents:
+      - eventRefs:
+          - ApplicationSubmitted
+          - SATScoresReceived
+          - RecommendationLetterReceived
+        actions:
+          - functionRef:
+              refName: finalizeApplicationFunction
+              arguments:
+                student: "${ .applicantId }"
+    end:
+      terminate: true
+
+  - name: CancelApplication
+    type: event
+    exclusive: false
+    onEvents:
+      - eventRefs:
+          - ApplicationSubmitted
+          - SATScoresReceived
+          - RecommendationLetterReceived
+        actions:
+          - functionRef:
+              refName: finalizeApplicationFunction
+              arguments:
+                student: "${ .applicantId }"
+    end:
+      terminate: true