amqp 0.8.0.rc13 → 0.8.0.rc14

data/spec/integration/extensions/rabbitmq/publisher_confirmations_spec.rb

@@ -0,0 +1,180 @@
+# encoding: utf-8
+
+require "spec_helper"
+
+require "amqp/extensions/rabbitmq"
+
+
+describe "confirm.select" do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+  include EventedSpec::SpecHelper
+
+  default_options AMQP_OPTS
+  default_timeout 3
+
+  amqp_before do
+    @channel = AMQP::Channel.new
+  end
+
+
+  context "with :nowait attribute off" do
+    it "results in a confirm.select-ok response" do
+      @channel.confirm_select do |select_ok|
+        done
+      end
+    end
+  end
+
+
+  context "with :nowait attribute set" do
+    it "results in NOT confirm.select-ok response" do
+      lambda do
+        @channel.confirm_select(:nowait => true) do
+          fail "Should never be called"
+        end
+      end.should raise_error(ArgumentError, /makes no sense/)
+
+      @channel.confirm_select(:nowait => true)
+
+      done(0.5)
+    end
+  end
+end
+
+
+
+
+describe "Publisher confirmation(s)" do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+  include EventedSpec::SpecHelper
+
+  default_options AMQP_OPTS
+  default_timeout 3
+
+
+  amqp_before do
+    @channel1 = AMQP::Channel.new
+    @channel2 = AMQP::Channel.new
+  end
+
+
+  context "when messages are transient" do
+    context "and routable" do
+      it "are confirmed as soon as they arrive on all the queues they were routed to" do
+        events   = Array.new
+
+        exchange = @channel2.fanout("amqpgem.tests.fanout1", :auto_delete => true)
+        queue    = @channel1.queue("", :auto_delete => true).bind(exchange).subscribe do |metadata, payload|
+          events << :basic_delivery
+        end
+
+        @channel2.confirm_select
+        @channel2.on_ack do |basic_ack|
+          events << :basic_ack
+        end
+        exchange.on_return do |basic_return, metadata, payload|
+          fail "Should never happen"
+        end
+
+        EventMachine.add_timer(0.5) do
+          exchange.publish("Hi", :persistent => false, :mandatory => true)
+        end
+
+        done(2.0) do
+          events.should include(:basic_ack)
+          events.should include(:basic_delivery)
+        end
+      end
+    end
+
+
+    context "and can be delivered immediately" do
+      it "are confirmed as soon as they arrive on all the queues they were routed to" do
+        events   = Array.new
+
+        exchange = @channel2.fanout("amqpgem.tests.fanout2", :auto_delete => true)
+        queue    = @channel1.queue("", :auto_delete => true).bind(exchange).subscribe do |metadata, payload|
+          events << :basic_delivery
+        end
+
+        @channel2.confirm_select
+        @channel2.on_ack do |basic_ack|
+          events << :basic_ack
+        end
+        exchange.on_return do |basic_return, metadata, payload|
+          fail "Should never happen"
+        end
+
+        EventMachine.add_timer(0.5) do
+          exchange.publish("Hi", :persistent => false, :immediately => true)
+        end
+
+        done(2.0) do
+          events.should include(:basic_ack)
+          events.should include(:basic_delivery)
+        end
+      end
+    end
+
+
+
+    context "and NOT routable" do
+      it "are delivered immediately after basic.return" do
+        events   = Array.new
+
+        queue    = @channel1.queue("", :auto_delete => true)
+        exchange = @channel2.fanout("amqpgem.tests.fanout3", :auto_delete => true)
+
+        @channel2.confirm_select
+        @channel2.on_ack do |basic_ack|
+          events << :basic_ack
+        end
+        exchange.on_return do |basic_return, metadata, payload|
+          events << :basic_return
+        end
+
+        EventMachine.add_timer(0.5) do
+          exchange.publish("Hi", :persistent => false, :mandatory => true)
+        end
+
+        done(2.0) { events.should == [:basic_return, :basic_ack] }
+      end
+    end
+
+
+
+    context "and CAN NOT be delivered immediately" do
+      it "are delivered immediately after basic.return" do
+        events   = Array.new
+
+        queue    = @channel1.queue("", :auto_delete => true)
+        exchange = @channel2.fanout("amqpgem.tests.fanout4", :auto_delete => true)
+
+        @channel2.confirm_select
+        @channel2.on_ack do |basic_ack|
+          events << :basic_ack
+        end
+        exchange.on_return do |basic_return, metadata, payload|
+          events << :basic_return
+        end
+
+        EventMachine.add_timer(0.5) do
+          exchange.publish("Hi", :persistent => false, :mandatory => true)
+        end
+
+        done(2.0) { events.should == [:basic_return, :basic_ack] }
+      end
+    end
+
+  end
+end

data/spec/integration/{workload_distribution_spec.rb → fanout_exchange_routing_spec.rb}

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# encoding: utf-8
+
 require "spec_helper"
 
 describe "Workload distribution" do
@@ -14,7 +15,7 @@ describe "Workload distribution" do
   em_after  { AMQP.cleanup_state }
 
   default_options AMQP_OPTS
-  default_timeout 5
+  default_timeout 6
 
   amqp_before do
     @channel   = AMQP::Channel.new
@@ -88,7 +89,7 @@ describe "Workload distribution" do
           end
 
           # for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -111,7 +112,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -134,7 +135,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -157,7 +158,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -179,7 +180,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -208,7 +209,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 
@@ -230,7 +231,7 @@ describe "Workload distribution" do
           end
 
           # 6 seconds are for Rubinius, it is surprisingly slow on this workload
-          done(1.5) {
+          done(3.5) {
             [@queue1, @queue2, @queue3].each do |q|
               @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
 

data/spec/integration/headers_exchange_routing_spec.rb

@@ -0,0 +1,269 @@
+# encoding: utf-8
+
+require 'spec_helper'
+
+describe "Headers exchange" do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+
+  default_timeout 5
+
+  amqp_before do
+    @connection = AMQP.connect
+    @channel    = AMQP::Channel.new(@connection)
+
+    @channel.on_error do |ch, channel_close|
+      fail "A channel-level exception: #{channel_close.inspect}"
+    end
+  end
+
+  after(:all) do
+    AMQP.cleanup_state
+    done
+  end
+
+
+  #
+  # Examples
+  #
+
+  # it would be following good practices to split this into
+  # 2 separate examples but I think this particular example
+  # is complete because it demonstrates routing in cases when
+  # different queues are bound with x-match = any AND x-match = all. MK.
+  it "can route messages based on any or all of N headers" do
+    exchange = @channel.headers("amq.match", :durable => true)
+
+    linux_and_ia64_messages = []
+    @channel.queue("", :auto_delete => true).bind(exchange, :arguments => { 'x-match' => 'all', :arch => "IA64", :os => 'linux' }).subscribe do |metadata, payload|
+      linux_and_ia64_messages << [metadata, payload]
+    end
+
+    linux_and_x86_messages = []
+    @channel.queue("", :auto_delete => true).bind(exchange, :arguments => { 'x-match' => 'all', :arch => "x86", :os => 'linux' }).subscribe do |metadata, payload|
+      linux_and_x86_messages << [metadata, payload]
+    end
+
+    any_linux_messages = []
+    @channel.queue("", :auto_delete => true).bind(exchange, :arguments => { :os => 'linux' }).subscribe do |metadata, payload|
+      any_linux_messages << [metadata, payload]
+    end
+
+    osx_or_octocore_messages = []
+    @channel.queue("", :auto_delete => true).bind(exchange, :arguments => { 'x-match' => 'any', :os => 'macosx', :cores => 8 }).subscribe do |metadata, payload|
+      osx_or_octocore_messages << [metadata, payload]
+    end
+
+    riak_messages = []
+    @channel.queue("", :auto_delete => true).bind(exchange, :arguments => { :package => { :name => 'riak', :version => '0.14.2' } }).subscribe do |metadata, payload|
+      riak_messages << [metadata, payload]
+    end
+
+
+    EventMachine.add_timer(0.5) do
+      exchange.publish "For linux/IA64",   :headers => { :arch => "IA64", :os => 'linux' }
+      exchange.publish "For linux/x86",    :headers => { :arch => "x86",  :os => 'linux' }
+      exchange.publish "For any linux",    :headers => { :os => 'linux' }
+      exchange.publish "For OS X",         :headers => { :os => 'macosx' }
+      exchange.publish "For solaris/IA64", :headers => { :os => 'solaris', :arch => 'IA64' }
+      exchange.publish "For ocotocore",    :headers => { :cores => 8 }
+
+      exchange.publish "For nodes with Riak 0.14.2", :headers => { :package => { :name => 'riak', :version => '0.14.2' } }
+    end
+
+    done(4.5) {
+      linux_and_ia64_messages.size.should == 1
+      linux_and_x86_messages.size.should == 1
+      any_linux_messages.size.should == 3
+      osx_or_octocore_messages.size.should == 2
+
+      riak_messages.size.should == 1
+    }
+  end
+end
+
+
+
+
+
+describe "Multiple consumers" do
+  include EventedSpec::AMQPSpec
+  default_options AMQP_OPTS
+  default_timeout 5
+
+  describe "bound to a queue with the same single header" do
+
+    #
+    # Environment
+    #
+
+    amqp_before do
+      @channel   = AMQP::Channel.new
+      @channel.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue    = @channel.queue("", :auto_delete => true)
+      @exchange = @channel.headers("amqpgem.tests.integration.headers.exchange1", :auto_delete => true)
+
+      @queue.bind(@exchange, :arguments => { :slug => "all" })
+    end
+
+
+
+    it "get messages distributed to them in a round-robin manner" do
+      mailbox1  = Array.new
+      mailbox2  = Array.new
+
+      consumer1 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox1 << payload }
+      consumer2 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox2 << payload }
+
+
+      EventMachine.add_timer(0.5) do
+        12.times { @exchange.publish(".", :headers => { :slug => "all" }) }
+        12.times { @exchange.publish(".", :headers => { :slug => "rspec" }) }
+      end
+
+      done(3.5) {
+        mailbox1.size.should == 6
+        mailbox2.size.should == 6
+      }
+    end
+  end
+
+
+
+  describe "bound to a queue with the same two header & x-match = all" do
+
+    #
+    # Environment
+    #
+
+    amqp_before do
+      @channel   = AMQP::Channel.new
+      @channel.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue    = @channel.queue("", :auto_delete => true)
+      @exchange = @channel.headers("amqpgem.tests.integration.headers.exchange1", :auto_delete => true)
+
+      @queue.bind(@exchange, :arguments => { :slug => "all", :arch => "ia64", 'x-match' => 'all' })
+    end
+
+
+    it "get messages distributed to them in a round-robin manner" do
+      mailbox1  = Array.new
+      mailbox2  = Array.new
+
+      consumer1 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox1 << payload }
+      consumer2 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox2 << payload }
+
+      EventMachine.add_timer(0.5) do
+        12.times { @exchange.publish(".", :headers => { :slug => "all",   :arch => "ia64" }) }
+        12.times { @exchange.publish(".", :headers => { :slug => "rspec", :arch => "ia64" }) }
+      end
+
+      done(3.5) {
+        mailbox1.size.should == 6
+        mailbox2.size.should == 6
+      }
+    end
+  end
+
+
+
+  describe "bound to 2 queues with the same two header & x-match = all" do
+
+    #
+    # Environment
+    #
+
+    amqp_before do
+      @channel   = AMQP::Channel.new
+      @channel.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue1   = @channel.queue("", :auto_delete => true)
+      @queue2   = @channel.queue("", :auto_delete => true)
+      @exchange = @channel.headers("amqpgem.tests.integration.headers.exchange1", :auto_delete => true)
+
+      args      = { :slug => "all", :arch => "ia64", 'x-match' => 'all' }
+      @queue1.bind(@exchange, :arguments => args)
+      @queue2.bind(@exchange, :arguments => args)
+    end
+
+    it "get messages distributed to both queues, and in a round-robin manner between consumers on one queue" do
+      mailbox1  = Array.new
+      mailbox2  = Array.new
+      mailbox3  = Array.new
+      mailbox4  = Array.new
+
+      consumer1 = AMQP::Consumer.new(@channel, @queue1).consume.on_delivery { |metadata, payload| mailbox1 << payload }
+      consumer2 = AMQP::Consumer.new(@channel, @queue1).consume.on_delivery { |metadata, payload| mailbox2 << payload }
+      consumer3 = AMQP::Consumer.new(@channel, @queue2).consume.on_delivery { |metadata, payload| mailbox3 << payload }
+      consumer4 = AMQP::Consumer.new(@channel, @queue2).consume.on_delivery { |metadata, payload| mailbox4 << payload }
+
+      EventMachine.add_timer(0.5) do
+        12.times { |i| @exchange.publish("all-#{i}",   :headers => { :slug => "all",   :arch => "ia64" }) }
+        16.times { |i| @exchange.publish("rspec-#{i}", :headers => { :slug => "rspec", :arch => "ia64" }) }
+      end
+
+      done(3.5) {
+        mailbox1.size.should == 6
+        mailbox1.should == ["all-0", "all-2", "all-4", "all-6", "all-8", "all-10"]
+
+        mailbox2.size.should == 6
+        mailbox2.should == ["all-1", "all-3", "all-5", "all-7", "all-9", "all-11"]
+
+        mailbox3.size.should == 6
+        mailbox4.size.should == 6
+      }
+    end
+  end
+
+
+
+
+  describe "bound to a queue with the same two header & x-match = any" do
+
+    #
+    # Environment
+    #
+
+    amqp_before do
+      @channel   = AMQP::Channel.new
+      @channel.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue    = @channel.queue("", :auto_delete => true)
+      @exchange = @channel.headers("amqpgem.tests.integration.headers.exchange1", :auto_delete => true)
+
+      @queue.bind(@exchange, :arguments => { :slug => "all", :arch => "ia64", 'x-match' => 'any' })
+    end
+
+    it "get messages distributed to them in a round-robin manner" do
+      mailbox1  = Array.new
+      mailbox2  = Array.new
+
+      consumer1 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox1 << payload }
+      consumer2 = AMQP::Consumer.new(@channel, @queue).consume.on_delivery { |metadata, payload| mailbox2 << payload }
+
+      EventMachine.add_timer(0.5) do
+        12.times { @exchange.publish(".", :headers => { :slug => "all",   :arch => "ia64" }) }
+        4.times  { @exchange.publish(".", :headers => { :slug => "rspec", :arch => "ia64" }) }
+      end
+
+      done(3.5) {
+        mailbox1.size.should == 8
+        mailbox2.size.should == 8
+      }
+    end
+  end
+end