Added GetNumState WorkflowUtils #127

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

Author: manick02

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

@@ -31,6 +31,7 @@
 /** Provides common utility methods to provide most often needed answers from a workflow */
 public final class WorkflowUtils {
   private static final int DEFAULT_STARTING_STATE_POSITION = 0;
+  private static final int DEFAULT_STATE_COUNT = 0;
 
   /**
    * Gets State matching Start state. If start is not present returns first state. If start is
@@ -265,6 +266,25 @@ public static List<Action> getActionsForFunctionDefinition(
     return getActionsWhichUsesFunctionDefinition(workflow, functionDefinitionName);
   }
 
+  /**
+   * Gets Num of State in the workflow does not consider child workflow
+   * @param workflow
+   * @return
+   */
+  public static int getNumOfStates(Workflow workflow) {
+    return hasStates(workflow) ? workflow.getStates().size() : DEFAULT_STATE_COUNT;
+  }
+
+  /**
+   *
+   * @param workflow
+   * @param type
+   * @return
+   */
+  public static int getNumOfStates(Workflow workflow,DefaultState.Type type) {
+    return hasStates(workflow) ? workflow.getStates().size() : 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

@@ -0,0 +1,43 @@
+/*
+ * 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 io.serverlessworkflow.api.Workflow;
+import io.serverlessworkflow.util.testutil.TestUtils;
+import io.serverlessworkflow.utils.WorkflowUtils;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.ValueSource;
+
+public class GetNumTests {
+  @ParameterizedTest
+  @ValueSource(strings = {"/getStates/workflowwithstates.yml"})
+  public void testGetNumStates(String workflowWithStates) {
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowWithStates);
+    int expectedStatesCount = 2;
+    assertEquals(expectedStatesCount, WorkflowUtils.getNumOfStates(workflow));
+  }
+
+  @ParameterizedTest
+  @ValueSource(strings = {"/start/workflowwithnostate.yml"})
+  public void testGetNumStatesForNoStateInWorkflow(String workflowWithStates) {
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowWithStates);
+    int expectedStatesCount = 0;
+    assertEquals(expectedStatesCount, WorkflowUtils.getNumOfStates(workflow));
+  }
+}