sidestate 0.1.0 → 0.2.0

data/lib/sidestate/hook.rb

@@ -8,20 +8,36 @@ module Sidestate::Hook
       
       define_method(aasm_state_object) do
         aasm_state_collection.last
-      end      
+      end
+    end
+    
+    base.aasm_state_class.instance_eval do
+      belongs_to base.name.underscore.to_sym
     end
   end
   
   module ClassMethods
+    def aasm_state_class
+      "#{name}State".constantize
+    end
+    
     def aasm_state_collection
-      "#{name}State".pluralize.underscore.to_sym
+      aasm_state_class.name.pluralize.underscore.to_sym
     end
     
     def aasm_state_object
-      "#{name}State".underscore.to_sym
+      aasm_state_class.name.underscore.to_sym
     end
   end
   
+  def next_states
+    self.class.aasm_events.collect { |name, event|
+      event.transitions_from_state(aasm_current_state)
+    }.flatten.collect { |transition|
+      Array(transition.to)
+    }.flatten
+  end
+  
   private
   
   def save_aasm_state_object

data/spec/sidestate/hook_spec.rb

@@ -1,7 +1,7 @@
 require 'spec/spec_helper'
 
 describe Sidestate::Hook do
-  context '#article_state' do
+  describe '#article_state' do
     it "should create a new state when a model object is created" do
       article = Article.create
       article.article_state.should_not be_nil
@@ -21,7 +21,7 @@ describe Sidestate::Hook do
     end
   end
   
-  context '#article_states' do
+  describe '#article_states' do
     it "should store all states for a given object" do
       Timecop.travel 2.days.ago
       article = Article.create
@@ -36,4 +36,23 @@ describe Sidestate::Hook do
       }.should == ['draft', 'published', 'flagged']
     end
   end
+  
+  describe '#next_states' do
+    it "should return the available states the object can move to" do
+      Timecop.travel 1.day.ago
+      article = Article.create
+      Timecop.return
+      
+      article.next_states.should =~ [:published, :deleted]
+      
+      article.publish!
+      article.next_states.should =~ [:flagged, :deleted]
+    end
+  end
+  
+  describe '#article' do
+    it "should add the association on the state object" do
+      ArticleState.reflect_on_association(:article).should_not be_nil
+    end
+  end
 end

data/spec/support/models.rb

@@ -1,3 +1,7 @@
+class ArticleState < ActiveRecord::Base
+  #
+end
+
 class Article < ActiveRecord::Base
   include AASM
   include Sidestate
@@ -25,7 +29,3 @@ class Article < ActiveRecord::Base
     transitions :from => [:draft, :published, :flagged], :to => :deleted
   end
 end
-
-class ArticleState < ActiveRecord::Base
-  #
-end

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Pat Allan
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-11 00:00:00 +11:00
+date: 2010-03-16 00:00:00 +11:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency