no more prizes to hand out and picking winners require prizes to stil…

…l exist.

app/controllers/entries_controller.rb

@@ -23,10 +23,14 @@ def show
   end
 
   def pick
-    winner = Entry.pick_winner
-    winner.won = true
-    winner.save
-    redirect_to(root_path, :notice => "Winner is #{winner.name}")
+    prizes = Prize.remaining
+    if prizes.count > 0
+      prize = prizes.first
+      winner = prize.pick_winner
+      winner.won = true
+      winner.save
+      redirect_to(root_path, :notice => "Winner is #{winner.name} - #{prize.name}")
+    end
   end
 
 end

app/models/prize.rb

@@ -4,4 +4,13 @@ class Prize < ActiveRecord::Base
 
   scope :remaining, where("amount > 0")
 
+  def pick_winner
+    if self.amount > 0
+      winner = Entry.pick_winner
+      self.amount -= 1
+      self.save
+      winner
+    end
+  end
+
 end

features/contest_over.feature

@@ -4,9 +4,19 @@ Feature: Contest over
   I want to pick the winners for the prizes
 
   Scenario: Pick the winners
-  Given I am a logged in admin user
-  And I have entries
-  And I have prizes
-  When I go to the home page
-  And I press "Pick a winner"
-  Then I should see "Winner is"
+    Given I am a logged in admin user
+    And I have entries
+    And I have prizes
+    When I go to the home page
+    And I press "Pick a winner"
+    Then I should see "Winner is"
+
+  Scenario: No More Prizes
+    Given I am a logged in admin user
+    And I have entries
+    And I have no more prizes
+    When I go to the home page
+    Then I should see "Sorry, no more prizes to hand out."
+
+
+

features/step_definitions/entry_steps.rb

@@ -9,6 +9,10 @@
   end
 end
 
+Given /^I have no more prizes$/ do
+  Prize.remaining.count.should eql(0)
+end
+
 Given /^I have prizes$/ do
   @prize = Factory.create(:prize)
   Prize.count.should > 0

spec/models/prize_spec.rb

@@ -23,4 +23,13 @@
     prize.should_not be_valid
   end
 
+  it "should have one less amount after it is handed out" do
+    [1,2,3].each do |i|
+      entry = Factory.create(:entry, :name => "john #{i}")
+    end
+    @prize.amount.should eql(1)
+    @prize.pick_winner
+    @prize.amount.should eql(0)
+  end
+
 end