sidestate 0.2.2 → 0.2.3

data/lib/sidestate/hook.rb

@@ -4,7 +4,8 @@ module Sidestate::Hook
       extend Sidestate::Hook::ClassMethods
       has_many aasm_state_collection
       
-      after_save :save_aasm_state_object
+      after_create :save_aasm_state_object_for_new_record
+      after_update :save_aasm_state_object
       
       define_method(aasm_state_object) do
         aasm_state_collection.last
@@ -47,6 +48,10 @@ module Sidestate::Hook
     aasm_state_collection.create! :name => new_state.to_s, self.class.name.underscore.to_sym => self
   end
   
+  def save_aasm_state_object_for_new_record
+    aasm_state_collection.create! :name => aasm_current_state.to_s, self.class.name.underscore.to_sym => self
+  end
+  
   def aasm_state_collection
     send(self.class.aasm_state_collection)
   end

data/spec/sidestate/hook_spec.rb

@@ -7,6 +7,11 @@ describe Sidestate::Hook do
       article.article_state.should_not be_nil
     end
     
+    it "should only create one state object when the model is created" do
+      article = Article.create
+      article.article_states.length.should == 1
+    end
+    
     it "should set the state accordingly" do
       Article.create.article_state.name.should == 'draft'
     end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 2
-  version: 0.2.2
+  - 3
+  version: 0.2.3
 platform: ruby
 authors: 
 - Pat Allan
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-30 00:00:00 +11:00
+date: 2010-04-29 00:00:00 +10:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency