amqp 0.8.0.rc13 → 0.8.0.rc14

data/spec/integration/ordering_of_published_messages_spec.rb

@@ -0,0 +1,96 @@
+# encoding: utf-8
+
+require 'spec_helper'
+
+describe "100 AMQP messages" do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+  default_options AMQP_OPTS
+  default_timeout 5
+
+
+  before :all do
+    @list = Range.new(0, 100, true).to_a
+  end
+
+
+  context "published and received on the same channel" do
+    amqp_before do
+      @channel   = AMQP::Channel.new
+      @channel.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue = @channel.queue("amqpgem.tests.integration.queue1", :auto_delete => true)
+    end
+
+    #
+    # Examples
+    #
+
+    it "are received on the same channel in the order of publishing" do
+      received = []
+
+      @queue.subscribe do |metadata, payload|
+        received << payload.to_i
+      end
+
+      EventMachine.add_timer(0.3) do
+        @list.each { |i| @channel.default_exchange.publish(i.to_s, :routing_key => @queue.name) }
+      end
+
+      done(3.5) {
+        received.size.should == 100
+        received.first.should == 0
+        received.last.should == 99
+
+        received.should == @list
+      }
+    end
+  end
+
+
+  context "published on two different channels" do
+    amqp_before do
+      @channel1   = AMQP::Channel.new
+      @channel2   = AMQP::Channel.new
+
+      @channel1.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+      @channel2.on_error do |ch, close|
+        raise "Channel-level error!: #{close.inspect}"
+      end
+
+      @queue = @channel1.queue("amqpgem.tests.integration.queue1", :auto_delete => true)
+    end
+
+    #
+    # Examples
+    #
+
+    it "are received on the same channel in the order of publishing" do
+      received = []
+
+      @queue.subscribe do |metadata, payload|
+        received << payload.to_i
+      end
+
+      EventMachine.add_timer(0.3) do
+        @list.each { |i| @channel2.default_exchange.publish(i.to_s, :routing_key => @queue.name) }
+      end
+
+      done(3.5) {
+        received.size.should == 100
+        received.first.should == 0
+        received.last.should == 99
+
+        received.should == @list
+      }
+    end
+  end
+end # describe

data/spec/integration/queue_declaration_spec.rb

@@ -18,7 +18,7 @@ describe AMQP do
 
   after(:all) do
     AMQP.cleanup_state
-    done(0.3)
+    done
   end
 
   #
@@ -32,13 +32,13 @@ describe AMQP do
       it "declares a new queue with that name" do
         queue = @channel.queue(name)
         queue.name.should == name
-        done(0.3)
+        done
       end
 
       it "caches that queue" do
         queue = @channel.queue(name)
         @channel.queue(name).object_id.should == queue.object_id
-        done(0.3)
+        done
       end
     end # context
 
@@ -48,7 +48,7 @@ describe AMQP do
           @channel.queue("") do |queue, *args|
             queue.name.should_not be_empty
             queue.delete
-            done(0.3)
+            done
           end
         end
       end
@@ -93,7 +93,7 @@ describe AMQP do
             @channel.queue(name, different_options)
           }.to raise_error(AMQP::IncompatibleOptionsError)
           @queue.delete
-          done(0.3)
+          done
         end
       end
     end
@@ -108,7 +108,7 @@ describe AMQP do
 
           queue.should == original_queue
 
-          done(0.3)
+          done
         end # it
       end
 
@@ -120,7 +120,7 @@ describe AMQP do
             exchange = @channel.queue("queue declared at #{Time.now.to_i}", :passive => true)
           }.to raise_error
 
-          done(0.3)
+          done
         end # it
       end # context
     end # context
@@ -136,7 +136,7 @@ describe AMQP do
           @channel.queue("previously.declared.durable.queue", :durable => false)
         }.to raise_error(AMQP::IncompatibleOptionsError)
 
-        done(0.3)
+        done
       end # it
     end # context
   end # describe

data/spec/integration/queue_status_spec.rb

@@ -0,0 +1,66 @@
+# encoding: utf-8
+
+require 'spec_helper'
+
+describe AMQP::Queue do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+
+  default_timeout 5
+
+
+
+  #
+  # Examples
+  #
+
+  describe "#status" do
+    it "yields # of messages & consumers to the callback" do
+      events  = []
+      channel = AMQP::Channel.new
+
+      queue1 = channel.queue("", :auto_delete => true)
+      queue2 = channel.queue("amqpgem.tests.a.named.queue", :auto_delete => true)
+
+      EventMachine.add_timer(1.0) do
+        queue1.status do |m, c|
+          events << :queue1_declare_ok
+        end
+        queue2.status do |m, c|
+          events << :queue2_declare_ok
+        end        
+      end
+
+      done(2.0) {
+        events.should include(:queue1_declare_ok)
+        events.should include(:queue2_declare_ok)
+      }
+    end
+
+
+    it "yields # of messages & consumers to the callback in pseudo-synchronous code" do
+      events  = []
+      channel = AMQP::Channel.new
+
+      queue1 = channel.queue("", :auto_delete => true)
+      queue2 = channel.queue("amqpgem.tests.a.named.queue", :auto_delete => true)
+
+      queue1.status do |m, c|
+        events << :queue1_declare_ok
+      end
+      queue2.status do |m, c|
+        events << :queue2_declare_ok
+      end
+
+      done(2.0) {
+        queue1.name.should =~ /^amq\..+/
+        events.should include(:queue1_declare_ok)
+        events.should include(:queue2_declare_ok)
+      }
+    end
+  end
+end

data/spec/integration/recovery/per_channel_automatic_recovery_on_graceful_broker_shutdown_spec.rb

@@ -0,0 +1,76 @@
+# encoding: utf-8
+require 'spec_helper'
+
+unless ENV["CI"]
+  describe "Per-channel automatic channel recovery" do
+
+    #
+    # Environment
+    #
+
+    include RabbitMQ::Control
+
+    include EventedSpec::AMQPSpec
+    default_timeout 20.0
+
+    amqp_before do
+      @channel = AMQP::Channel.new(AMQP.connection, :auto_recovery => true)
+    end
+
+
+    after :all do
+      start_rabbitmq unless rabbitmq_pid
+    end
+
+    # ...
+
+
+    #
+    # Examples
+    #
+
+    it "can be used when broker is shut down gracefully" do
+      AMQP.connection.on_error do |conn, connection_close|
+        puts "[connection.close] Reply code = #{connection_close.reply_code}, reply text = #{connection_close.reply_text}"
+        if connection_close.reply_code == 320
+          puts "[connection.close] Setting up a periodic reconnection timer..."
+          conn.periodically_reconnect(1)
+        end
+      end
+
+      pid = rabbitmq_pid
+      puts "rabbitmq pid = #{pid.inspect}"
+
+      stop_rabbitmq
+      rabbitmq_pid.should be_nil
+
+      # 2 seconds later, check that we are reconnecting
+      EventMachine.add_timer(2.0) do
+        AMQP.connection.should_not be_connected
+        AMQP.connection.should be_reconnecting
+      end
+
+
+      # 4 seconds later, start RabbitMQ
+      EventMachine.add_timer(4.0) do
+        start_rabbitmq
+        rabbitmq_pid.should_not be_nil
+      end
+
+      # 10 seconds later, use the (now recovered) connection. Note that depending
+      # on # of plugins used it may take 5-10 seconds to start up RabbitMQ and initialize it,
+      # then open a new AMQP connection. That's why we wait. MK.
+      EventMachine.add_timer(10.0) do
+        AMQP.connection.should be_connected
+        AMQP.connection.should_not be_reconnecting
+
+        @channel.queue("amqpgem.tests.a.queue", :auto_delete => true).subscribe do |metadata, payload|
+          puts "Got a message: #{payload.inspect}"
+          done
+        end
+
+        EventMachine.add_timer(1.5) { @channel.default_exchange.publish("", :routing_key => "amqpgem.tests.a.queue") }
+      end
+    end
+  end
+end

data/spec/integration/recovery/per_channel_automatic_recovery_spec.rb

@@ -0,0 +1,72 @@
+# encoding: utf-8
+require 'spec_helper'
+
+unless ENV["CI"]
+  describe "Per-channel automatic channel recovery" do
+
+    #
+    # Environment
+    #
+
+    include RabbitMQ::Control
+
+    include EventedSpec::AMQPSpec
+    default_timeout 20.0
+
+    amqp_before do
+      @channel = AMQP::Channel.new(AMQP.connection, :auto_recovery => true)
+    end
+
+    after :all do
+      start_rabbitmq unless rabbitmq_pid
+    end
+
+    # ...
+
+
+    #
+    # Examples
+    #
+
+    it "kicks in when broker goes down" do
+      AMQP.connection.on_tcp_connection_loss do |conn, settings|
+        puts "[network failure] Trying to reconnect..."
+        conn.reconnect(false, 1)
+      end
+
+      pid = rabbitmq_pid
+      puts "rabbitmq pid = #{pid.inspect}"
+
+      kill_rabbitmq
+      rabbitmq_pid.should be_nil
+
+      # 2 seconds later, check that we are reconnecting
+      EventMachine.add_timer(2.0) do
+        AMQP.connection.should_not be_connected
+        AMQP.connection.should be_reconnecting
+      end
+
+
+      # 4 seconds later, start RabbitMQ
+      EventMachine.add_timer(4.0) do
+        start_rabbitmq
+        rabbitmq_pid.should_not be_nil
+      end
+
+      # 12 seconds later, use the (now recovered) connection. Note that depending
+      # on # of plugins used it may take 5-10 seconds to start up RabbitMQ and initialize it,
+      # then open a new AMQP connection. That's why we wait. MK.
+      EventMachine.add_timer(12.0) do
+        AMQP.connection.should be_connected
+        AMQP.connection.should_not be_reconnecting
+
+        @channel.queue("amqpgem.tests.a.queue", :auto_delete => true).subscribe do |metadata, payload|
+          puts "Got a message"
+          done
+        end
+
+        EventMachine.add_timer(1.5) { @channel.default_exchange.publish("Hi", :routing_key => "amqpgem.tests.a.queue") }
+      end
+    end
+  end
+end

data/spec/integration/redelivery_of_unacknowledged_messages_spec.rb

@@ -0,0 +1,96 @@
+# encoding: utf-8
+
+require 'spec_helper'
+
+describe "Unacknowledged messages" do
+
+  #
+  # Environment
+  #
+
+  include EventedSpec::AMQPSpec
+
+  default_timeout 5
+
+  amqp_before do
+    @connection1 = AMQP.connect
+    @connection2 = AMQP.connect
+    @connection3 = AMQP.connect
+
+    @channel1    = AMQP::Channel.new(@connection1)
+    @channel2    = AMQP::Channel.new(@connection2)
+    @channel3    = AMQP::Channel.new(@connection3)
+
+    [@channel1, @channel2, @channel3].each { |ch| ch.on_error { fail } }
+
+    @channel1.prefetch(3)
+    @channel2.prefetch(1)
+  end
+
+  after(:all) do
+    AMQP.cleanup_state
+    done
+  end
+
+
+
+  #
+  # Examples
+  #
+
+  # this is a spec example based on one of the Working With Queues doc guides.
+  # It is somewhat hairy since it imitates 3 apps in a single process
+  # but demonstrates redeliveries pretty well. MK.
+
+  it "are redelivered to alternate consumers when the 'primary' one disconnects" do
+    number_of_messages_app2_received = 0
+    expected_number_of_deliveries    = 21
+    redelivery_values                = Array.new
+
+    exchange = @channel3.direct("amq.direct")
+
+    queue1    = @channel1.queue("amqpgem.examples.acknowledgements.explicit", :auto_delete => false)
+    # purge the queue so that we don't get any redeliveries from previous runs
+    queue1.purge
+    queue1.bind(exchange).subscribe(:ack => true) do |metadata, payload|
+      # acknowledge some messages, they will be removed from the queue
+      if metadata.headers["i"] < 10
+        @channel1.acknowledge(metadata.delivery_tag, false)
+      else
+        # some messages are not ack-ed and will remain in the queue for redelivery
+        # when app #1 connection is closed (either properly or due to a crash)
+      end
+    end
+
+    queue2    = @channel2.queue!("amqpgem.examples.acknowledgements.explicit", :auto_delete => false)
+    queue2.subscribe(:ack => true) do |metadata, payload|
+      redelivery_values << metadata.redelivered?
+
+      # app 2 always acks messages
+      metadata.ack
+
+      number_of_messages_app2_received += 1
+    end
+
+    EventMachine.add_timer(2.0) {
+      # app1 quits/crashes
+      @connection1.close
+    }
+
+
+    # 0.5 seconds later, publish a bunch of messages
+    EventMachine.add_timer(0.5) {
+      30.times do |i|
+        exchange.publish("Message ##{i}", :headers => { :i => i })
+        i += 1
+      end
+    }
+
+
+    done(4.8) {
+      number_of_messages_app2_received.should be >= expected_number_of_deliveries
+      # 3 last messages are redeliveries
+      redelivery_values.last(7).should == [false, false, false, false, true, true, true]
+    }
+  end
+end