Added GetNumOfEndStates util method #127

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

Author: manick02

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

@@ -290,6 +290,12 @@ public static long getNumOfStates(Workflow workflow, DefaultState.Type type) {
         : DEFAULT_STATE_COUNT;
   }
 
+  public static long getNumOfEndStates(Workflow workflow) {
+    return hasStates(workflow)
+        ? workflow.getStates().stream().filter(state -> state.getEnd() != null).count()
+        : DEFAULT_STATE_COUNT;
+  }
+
   private static List<Action> getActionsWhichUsesFunctionDefinition(
       Workflow workflow, String functionDefinitionName) {
     List<Action> actions = new ArrayList<>();

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

@@ -50,4 +50,12 @@ public void testGetNumStatesOfEventType(String workflowWithStates) {
     assertEquals(
         expectedStatesCount, WorkflowUtils.getNumOfStates(workflow, DefaultState.Type.EVENT));
   }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/events/workflowwithevents.yml"})
+  public void testGetNumEndStates(String workflowWithStates) {
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowWithStates);
+    int expectedEndStatesCount = 2;
+    assertEquals(expectedEndStatesCount, WorkflowUtils.getNumOfEndStates(workflow));
+  }
 }