diff options
author | rick <technoweenie@gmail.com> | 2008-06-28 11:33:50 -0700 |
---|---|---|
committer | rick <technoweenie@gmail.com> | 2008-06-28 11:33:50 -0700 |
commit | c9e366e997c6f3a383cfaa6351fa847e92de7fe4 (patch) | |
tree | 1fce341c1b769bd78494bfb6a99a314b57d6a3ed /activemodel/lib/active_model/state_machine.rb | |
parent | a9d9ca16c739ec39a192d29c62f760e51040fc6e (diff) | |
download | rails-c9e366e997c6f3a383cfaa6351fa847e92de7fe4.tar.gz rails-c9e366e997c6f3a383cfaa6351fa847e92de7fe4.tar.bz2 rails-c9e366e997c6f3a383cfaa6351fa847e92de7fe4.zip |
all aasm tests without activerecord moved over and passing
Diffstat (limited to 'activemodel/lib/active_model/state_machine.rb')
-rw-r--r-- | activemodel/lib/active_model/state_machine.rb | 22 |
1 files changed, 19 insertions, 3 deletions
diff --git a/activemodel/lib/active_model/state_machine.rb b/activemodel/lib/active_model/state_machine.rb index 2a5ac95a3e..96df6539ae 100644 --- a/activemodel/lib/active_model/state_machine.rb +++ b/activemodel/lib/active_model/state_machine.rb @@ -37,13 +37,29 @@ module ActiveModel end end - def current_state(name = nil, new_state = nil) + def current_state(name = nil, new_state = nil, persist = false) sm = self.class.state_machine(name) - ivar = "@#{sm.name}_current_state" + ivar = sm.current_state_variable if name && new_state + if persist && respond_to?(:write_state) + write_state(sm, new_state) + end + + if respond_to?(:write_state_without_persistence) + write_state_without_persistence(sm, new_state) + end + instance_variable_set(ivar, new_state) else - instance_variable_get(ivar) || instance_variable_set(ivar, sm.initial_state) + instance_variable_set(ivar, nil) unless instance_variable_defined?(ivar) + value = instance_variable_get(ivar) + return value if value + + if respond_to?(:read_state) + value = instance_variable_set(ivar, read_state(sm)) + end + + value || sm.initial_state end end end |