@@ -5,4 +5,23 @@ class Vote < ActiveRecord::Base validates_presence_of :user_id validates_presence_of :pain_point_id validates_uniqueness_of :user_id, :scope => [:pain_point_id] + + acts_as_state_machine :initial => :neutral + state :neutral + state :up + state :down + + event :up_vote do + transitions :from => :neutral, :to => :up + transitions :from => :down, :to => :up + transitions :from => :up, :to => :neutral + end + alias_method :up_vote, :up_vote! + + event :down_vote do + transitions :from => :neutral, :to => :down + transitions :from => :up, :to => :down + transitions :from => :down, :to => :neutral + end + alias_method :down_vote, :down_vote! end app/models/vote.rb @@ -3,6 +3,7 @@ class CreateVotes < ActiveRecord::Migration create_table :votes do |t| t.integer :user_id t.integer :pain_point_id + t.column :state, :string, :null => :no, :default => 'neutral' t.timestamps end end db/migrate/003_create_votes.rb @@ -36,6 +36,7 @@ ActiveRecord::Schema.define(:version => 3) do create_table "votes", :force => true do |t| t.integer "user_id" t.integer "pain_point_id" + t.string "state", :default => "neutral" t.datetime "created_at" t.datetime "updated_at" end db/schema.rb @@ -32,4 +32,85 @@ describe Vote do end end end + + describe "State Transitions" do + attr_reader :vote + before do + @vote = votes(:quentin_slow_tests) + end + + describe "#up_vote" do + describe "when in neutral state" do + before do + vote.state.should == 'neutral' + end + + it "transitions to up state" do + vote.up_vote + vote.state.should == 'up' + end + end + + describe "when in down state" do + before do + vote.state = 'down' + vote.save! + end + + it "transitions to up state" do + vote.up_vote + vote.state.should == 'up' + end + end + + describe "when in up state" do + before do + vote.state = 'up' + vote.save! + end + + it "transitions to neutral state" do + vote.up_vote + vote.state.should == 'neutral' + end + end + end + + describe "#down_vote" do + describe "when in neutral state" do + before do + vote.state.should == 'neutral' + end + + it "transitions to down state" do + vote.down_vote + vote.state.should == 'down' + end + end + + describe "when in down state" do + before do + vote.state = 'down' + vote.save! + end + + it "transitions to neutral state" do + vote.down_vote + vote.state.should == 'neutral' + end + end + + describe "when in up state" do + before do + vote.state = 'up' + vote.save! + end + + it "transitions to down state" do + vote.down_vote + vote.state.should == 'down' + end + end + end + end end spec/models/vote_spec.rb 298d61ec249fabef62906ebb8b88e36f19e0dbcb Brian Takita brian.takita@gmail.com http://github.com/btakita/pain-point/commit/33d3ab1b9b874120d0ec81de515a041403524d32 33d3ab1b9b874120d0ec81de515a041403524d32 2008-04-20T23:54:14-07:00 2008-04-20T23:54:14-07:00 Vote has a state machine with up, down, and neutral states. Vote also has the up_vote and down_vote state transitions available. d5d07a704628071654ee6c87c0d333f5510b7852 Brian Takita brian.takita@gmail.com