fix the timesheet controller unit tests

1 files changed, 7 insertions, 7 deletions

diff --git a/spec/unit/controllers/timesheet.rb b/spec/unit/controllers/timesheet.rb
index 3189069..82ba6c9 100644
--- a/spec/unit/controllers/timesheet.rb
+++ b/spec/unit/controllers/timesheet.rb
@@ -1,13 +1,14 @@
 require 'developwithpassion_fakes'
 
 class TimesheetController
-  def initialize(blah)
-    
+  def initialize(repository)
+    @repository = repository
   end
   def index
-    []
+    @repository.find_all Timesheet
   end
-  
+end
+class Timesheet
 end
 describe TimesheetController do
   before do
@@ -16,15 +17,14 @@ describe TimesheetController do
   end
   describe "when loading all timesheets" do
     it "should return all the timesheets" do
-      @results.should[0].equal(@timesheet)
+      @results[0].must_equal(@timesheet)
     end
     before do
       timesheets = []
       @timesheet = fake
       timesheets << @timesheet
-      @repository.stub(:find_all).ignore_arg.and_return(timesheets)
+      @repository.stub(:find_all).with(Timesheet).and_return(timesheets)
       @results = @sut.index
     end
   end
-
 end