made sure, old state callbacks are still working with the new dsl

spec/models/process_with_new_dsl.rb

@@ -5,9 +5,11 @@ def self.state(*args)
     raise "wrong state method"
   end
 
+  attr_accessor :flagged
+
   aasm do
     state :sleeping, :initial => true
-    state :running
+    state :running, :after_enter => :flag
     state :suspended
 
     event :start do
@@ -18,6 +20,10 @@ def self.state(*args)
     end
   end
 
+  def flag
+    self.flagged = true
+  end
+
   def self.event(*args)
     raise "wrong event method"
   end

spec/unit/new_dsl_spec.rb

@@ -12,8 +12,10 @@
   end
 
   it 'should have states and transitions' do
+    @process.flagged.should be_nil
     @process.start!
     @process.should be_running
+    @process.flagged.should be_true
     @process.stop!
     @process.should be_suspended
   end