require File.dirname(__FILE__) + '/../../spec_helper'
require File.dirname(__FILE__) + '/shared_adapter_behavior'
require 'rosetta_queue/adapters/beanstalk'

module RosettaQueue
  module Gateway

    describe BeanstalkAdapter do

      before(:each) do
        @msg = "Hello World!"
        @handler = mock('handler', :on_message => "", :destination => :foo)
        @msg_obj = mock("message", :body => @msg, :delete => true)
        @conn = mock("Beanstalk::Pool", :put => true, :reserve => @msg_obj)
        ::Beanstalk::Pool.stub!(:new).and_return(@conn)
        @adapter = BeanstalkAdapter.new({:host => "host", :port => "port"})
        @adapter.stub!(:running).and_yield
      end

      it_should_behave_like "an adapter"

      describe "#receive_once" do
        def do_receiving_once
          @adapter.receive_once
        end

        it "should delete messages once received" do
          when_receiving_once {
            @msg_obj.should_receive(:delete)
          }
        end
      end

      describe "#receive" do
        def do_receiving
          @adapter.receive
        end

        it "should delete message during receive" do
          when_receiving {
            @msg_obj.should_receive(:delete)
          }
        end
      end
    end
  end
end