Contents

require "spec_helper"
require "baton"
require "baton/consumer"
require "baton/consumer_manager"
require "baton/server"
require "ostruct"

describe Baton::ConsumerManager do

  before :each do
    Baton::Server.any_instance.stub(:facts).and_return({
      "fqdn" => "camac.dsci.it",
      "chef_environment" => "production"
    })
    server = Baton::Server.new
    @consumer = Baton::Consumer.new("camac", server)
  end

  subject {
    Baton::ConsumerManager.new(@consumer, nil, mock_exchange({:direct => true}), mock_exchange({:direct => true}))
  }

  let(:metadata) do
    obj = OpenStruct.new
    obj.content_type = "application/json"
    obj
  end

  let(:payload) do
    JSON({"type" => "message type" })
  end

  describe "#start" do
    it "will subscribe to a queue using the correct routing key" do
      subject.exchange_in.stub(:name)
      allow_message_expectations_on_nil
      queue = mock("queue")
      queue.should_receive(:bind).with(subject.exchange_in, routing_key: "camac.production")
      queue.should_receive(:subscribe)
      subject.channel.stub(:queue).and_return(queue)
      subject.start
    end
  end

  describe "#handle_message" do
    include FakeFS::SpecHelpers

    context "given a message" do
      it "should forward the payload to the consumer" do
        subject.consumer.should_receive(:handle_message).with(payload)
        subject.handle_message(metadata, payload)
      end

      it "should call process_message on the consumer" do
        subject.consumer.should_receive(:process_message)
        subject.handle_message(metadata, payload)
      end
    end
  end

  describe "#update" do
    context "given a message is sent to the consumer and the consumer notifies" do
      it "should trigger update with a message" do
        @consumer.stub(:process_message) do |message|
          @consumer.notify("message from consumer")
        end
        subject.should_receive(:update)
        subject.handle_message(metadata, payload)
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
baton-0.4.5	spec/baton/consumer_manager_spec.rb
baton-0.4.4	spec/baton/consumer_manager_spec.rb
baton-0.4.3	spec/baton/consumer_manager_spec.rb
baton-0.4.2	spec/baton/consumer_manager_spec.rb
baton-0.4.1	spec/baton/consumer_manager_spec.rb
baton-0.3.7	spec/baton/consumer_manager_spec.rb
baton-0.3.6	spec/baton/consumer_manager_spec.rb