test/unit/daemon_test.rb in refinery-0.9.5 vs test/unit/daemon_test.rb in refinery-0.9.6

@@ -2,28 +2,30 @@
 class DaemonTest < Test::Unit::TestCase
   context "a daemon" do
     setup do
       @server = stub('Server')
       @waiting_queue = stub('Queue(waiting)')
-      @waiting_queue.stubs(:receive)
       @error_queue = stub('Queue(error)')
       @done_queue = stub('Queue(done)')
     end
     should "be startable" do
+      @waiting_queue.stubs(:receive)
       assert_nothing_raised do
         daemon = Refinery::Daemon.new(@server, 'sample', @waiting_queue, @error_queue, @done_queue)
       end
     end
     should "have logging" do
+      @waiting_queue.stubs(:receive)
       daemon = Refinery::Daemon.new(@server, 'sample', @waiting_queue, @error_queue, @done_queue)
       assert_not_nil daemon.logger
     end
     should "allow visibility setting" do
-      @waiting_queue.stubs(:receive).with({'visibility' => 600})
+      @waiting_queue.expects(:receive).with(600)
       daemon = Refinery::Daemon.new(@server, 'sample', @waiting_queue, @error_queue, @done_queue, {'visibility' => 600})
     end
     context "that is started" do
       setup do
+        @waiting_queue.stubs(:receive)
         @daemon = Refinery::Daemon.new(@server, 'sample', @waiting_queue, @error_queue, @done_queue)
       end
       should "have a state of running" do
         assert @daemon.running?
       end
\ No newline at end of file