require 'fake_smith'

class MyAgent < Smith::Agent
  def run
    receiver("ack_twice_queue", :auto_ack => false).subscribe(&method(:ack_twice))
    receiver("ack_once_queue", :auto_ack => false).subscribe(&method(:ack_once))
    receiver("auto_ack_and_ack_queue", :auto_ack => true).subscribe(&method(:auto_ack_and_ack))
    receiver("send_and_expect_reply", :auto_ack => true).subscribe(&method(:send_and_expect_reply))
  end

  def ack_twice(payload, receiver)
    receiver.ack
    receiver.ack
  end

  def ack_once(payload, receiver)
    receiver.ack
  end

  def auto_ack_and_ack(payload, receiver)
    receiver.ack
  end

  def send_and_expect_reply(payload, receiver)
    sender("replying_queue") do |q|
      q.on_reply(:reply_queue_name => "somereplyqueuename") { |response| @replying_worked = true if response == {:correct => "response"} }
      q.on_timeout { raise "replying epic fail" }
      q.publish({:correct => "message"})
    end
  end

  def replying_worked?
    @replying_worked
  end
end

describe FakeSmith do
  let(:receiver) { double(:receiver, :ack => true) }
  let(:message) { {} }
  let(:agent) { MyAgent.new }
  before(:each) do
    agent.run
  end

  describe 'acking messages twice' do
    it 'raises errors' do
      expect { FakeSmith.send_message("ack_twice_queue", message, receiver) }.to raise_error(FakeSmith::MessageAckedTwiceError)
    end

    it 'raises errors when acking once and auto_ack' do
      expect { FakeSmith.send_message("auto_ack_and_ack_queue", message, receiver) }.to raise_error(FakeSmith::MessageAckedTwiceError)
    end
  end

  describe 'acking' do
    it 'acks the passed in receiver' do
      expect(receiver).to receive(:ack)
      FakeSmith.send_message("ack_once_queue", message, receiver)
    end
  end

  describe 'replying' do
    it 'allows setting reply handlers' do
      FakeSmith.set_reply_handler("replying_queue") do |send_message|
        {:correct => "response"} if send_message == {:correct => "message"}
      end
      FakeSmith.send_message("send_and_expect_reply", message, receiver)
    end
  end
end