aasm_from_states_for_state now supports to filter for specific transi…

…tion

CHANGELOG.md

@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 3.0.12
+
+ * aasm_from_states_for_state now supports to filter for specific transition
+
 ## 3.0.11
 
  * added class method aasm_from_states_for_state to retrieve all from states (regarding transitions) for a given state

lib/aasm/aasm.rb

@@ -31,8 +31,12 @@ def aasm_initial_state(set_state=nil)
       end
     end
 
-    def aasm_from_states_for_state(state)
-      aasm.events.map {|k,v| v.transitions_to_state(state)}.flatten.map(&:from).flatten
+    def aasm_from_states_for_state(state, options={})
+      if options[:transition]
+        aasm.events[options[:transition]].transitions_to_state(state).flatten.map(&:from).flatten
+      else
+        aasm.events.map {|k,v| v.transitions_to_state(state)}.flatten.map(&:from).flatten
+      end
     end
 
     # deprecated

spec/spec_helpers/models_spec_helper.rb

@@ -109,7 +109,7 @@ class AuthMachine
     event :unpassify do
       transitions :from => :passive, :to => :active, :guard => Proc.new {|u| false }
     end
-  
+
     event :unsuspend do
       transitions :from => :suspended, :to => :active,  :guard => Proc.new {|u| u.has_activated? }
       transitions :from => :suspended, :to => :pending, :guard => Proc.new {|u| u.has_activation_code? }

spec/unit/aasm_spec.rb

@@ -49,7 +49,12 @@
 describe "aasm_from_states_for_state" do
   it "should return all from states for a state" do
     froms = AuthMachine.aasm_from_states_for_state(:active)
-    [:pending, :passive, :suspended].each {|from| froms.should include(:pending)}
+    [:pending, :passive, :suspended].each {|from| froms.should include(from)}
+  end
+
+  it "should return from states for a state for a particular transition only" do
+    froms = AuthMachine.aasm_from_states_for_state(:active, :transition => :unsuspend)
+    [:suspended].each {|from| froms.should include(from)}
   end
 end