Merge pull request #152 from tsurdilo/updateutil2

adding getStateWithName util method

Author: tsurdilo

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

@@ -293,6 +293,28 @@ public static long getNumOfStates(Workflow workflow, DefaultState.Type type) {
         : DEFAULT_STATE_COUNT;
   }
 
+  /**
+   * Returns workflow state from provided name, or null if not found.
+   *
+   * @param workflow
+   * @param name
+   * @return
+   */
+  public static State getStateWithName(Workflow workflow, String name) {
+    if (!hasStates(workflow)) {
+      return null;
+    }
+
+    Optional<State> state =
+        workflow.getStates().stream().filter(s -> s.getName().equals(name)).findFirst();
+
+    if (state.isPresent()) {
+      return state.get();
+    } else {
+      return null;
+    }
+  }
+
   public static long getNumOfEndStates(Workflow workflow) {
     if (hasStates(workflow)) {
       long count = workflow.getStates().stream().filter(state -> state.getEnd() != null).count();

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

@@ -21,6 +21,7 @@
 import io.serverlessworkflow.api.Workflow;
 import io.serverlessworkflow.api.interfaces.State;
 import io.serverlessworkflow.api.states.DefaultState;
+import io.serverlessworkflow.api.states.EventState;
 import io.serverlessworkflow.util.testutil.TestUtils;
 import io.serverlessworkflow.utils.WorkflowUtils;
 import java.util.List;
@@ -42,6 +43,21 @@ public void testGetStatesByDefaultState(String workflowWithState) {
     assertEquals(notMatchingEvents, notMatchingStates.size());
   }
 
+  @ParameterizedTest
+  @ValueSource(strings = {"/getStates/workflowwithstates.yml"})
+  public void testGetStateByName(String workflowWithState) {
+    Workflow workflow = TestUtils.createWorkflowFromTestResource(workflowWithState);
+
+    State finalizeApplicationState =
+        WorkflowUtils.getStateWithName(workflow, "FinalizeApplication");
+    assertNotNull(finalizeApplicationState);
+    assertTrue(finalizeApplicationState instanceof EventState);
+
+    State cancelApplicationState = WorkflowUtils.getStateWithName(workflow, "CancelApplication");
+    assertNotNull(cancelApplicationState);
+    assertTrue(cancelApplicationState instanceof EventState);
+  }
+
   @Test
   public void testGetsStatesForNullWorkflow() {
     assertNull(WorkflowUtils.getStates(null, DefaultState.Type.EVENT));