Add method getPossibleStates

winzou · Dec 23, 2015 · 97f8f74 · 97f8f74
1 parent f302ac9
commit 97f8f74
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/spec/SM/StateMachine/StateMachineSpec.php b/spec/SM/StateMachine/StateMachineSpec.php
@@ -159,4 +159,11 @@ function it_returns_possible_transitions($object)
 
         $this->getPossibleTransitions()->shouldReturn(array('create', 'confirm'));
     }
+
+    function it_returns_possible_states($object)
+    {
+        $object->getState()->willReturn('checkout');
+
+        $this->getPossibleStates()->shouldReturn(array('pending', 'confirmed'));
+    }
 }
diff --git a/src/SM/StateMachine/StateMachine.php b/src/SM/StateMachine/StateMachine.php
@@ -186,6 +186,23 @@ public function getPossibleTransitions()
         );
     }
 
+    /**
+     * Returns the possible states
+     *
+     * @return array
+     */
+    public function getPossibleStates()
+    {
+        $states = array();
+        $transitions = $this->getPossibleTransitions();
+
+        foreach ($transitions as $transition) {
+            $states[] = $this->config['transitions'][$transition]['to'];
+        }
+
+        return $states;
+    }
+
     /**
      * Set a new state to the underlying object
      *

diff --git a/src/SM/StateMachine/StateMachineInterface.php b/src/SM/StateMachine/StateMachineInterface.php
@@ -65,4 +65,11 @@ public function getGraph();
      * @return array
      */
     public function getPossibleTransitions();
+
+    /**
+     * Returns the possible states
+     *
+     * @return array
+     */
+    public function getPossibleStates();
 }