lib/aasm/rspec/allow_transition_to.rb in aasm-4.12.3 vs lib/aasm/rspec/allow_transition_to.rb in aasm-5.0.0

@@ -1,12 +1,16 @@
 RSpec::Matchers.define :allow_transition_to do |state|
   match do |obj|
     @state_machine_name ||= :default
-    obj.aasm(@state_machine_name).states(:permitted => true).include?(state)
+    obj.aasm(@state_machine_name).states({:permitted => true}, *@args).include?(state)
   end
 
   chain :on do |state_machine_name|
     @state_machine_name = state_machine_name
+  end
+
+  chain :with do |*args|
+    @args = args
   end
 
   description do
     "allow transition to #{expected} (on :#{@state_machine_name})"
   end