fixing issue #66

lib/aasm/aasm.rb

@@ -83,7 +83,10 @@ def aasm
 
   # may be overwritten by persistence mixins
   def aasm_read_state
-    aasm.enter_initial_state
+    # all the following lines behave like @current_state ||= aasm.enter_initial_state
+    current = aasm.instance_variable_get("@current_state")
+    return current if current
+    aasm.instance_variable_set("@current_state", aasm.enter_initial_state)
   end
 
   # may be overwritten by persistence mixins

lib/aasm/instance_base.rb

@@ -6,7 +6,7 @@ def initialize(instance)
     end
 
     def current_state
-      @current_state ||= @instance.aasm_read_state
+      @instance.aasm_read_state
     end
 
     def current_state=(state)

spec/unit/persistence/active_record_persistence_spec.rb

@@ -122,6 +122,13 @@
 
 describe 'transitions with persistence' do
 
+  it "should work for valid models" do
+    valid_object = Validator.create(:name => 'name')
+    valid_object.should be_sleeping
+    valid_object.status = :running
+    valid_object.should be_running
+  end
+
   it 'should not store states for invalid models' do
     validator = Validator.create(:name => 'name')
     validator.should be_valid