amqp 0.7.0 → 0.7.1

data/spec/integration/workload_distribution_spec.rb

@@ -0,0 +1,245 @@
+# -*- coding: utf-8 -*-
+require "spec_helper"
+
+describe "Workload distribution" do
+
+  #
+  # Environment
+  #
+
+  include AMQP::Spec
+  include AMQP::SpecHelper
+
+  em_before { AMQP.cleanup_state }
+  em_after  { AMQP.cleanup_state }
+
+  default_options AMQP_OPTS
+  default_timeout 5
+
+  amqp_before do
+    @channel   = AMQP::Channel.new
+    @channel.should be_open
+  end
+
+  after(:all) do
+    AMQP.cleanup_state
+    done
+  end
+
+
+  #
+  # Examples
+  #
+
+  context "that uses fanout exchange" do
+    amqp_before :each do
+      @exchange = @channel.fanout("amqpgem.integration.multicast.fanout", :auto_delete => true)
+    end
+
+    context "with three bound queues" do
+      amqp_before :each do
+        @queue1   = @channel.queue("amqpgem.integration.multicast.queue1",  :auto_delete => true)
+        @queue2   = @channel.queue("amqpgem.integration.multicast.queue2",  :auto_delete => true)
+        @queue3   = @channel.queue("amqpgem.integration.multicast.queue3",  :auto_delete => true)
+
+        @queues   = [@queue1, @queue2, @queue3]
+
+        @sent_values = Array.new
+
+        @queue1.bind(@exchange).subscribe do |payload|
+          @received_messages[@queue1.name].push(payload.to_i)
+        end # subscribe
+
+        @queue2.bind(@exchange).subscribe do |payload|
+          @received_messages[@queue2.name].push(payload.to_i)
+        end # subscribe
+
+        @queue3.bind(@exchange).subscribe do |payload|
+          @received_messages[@queue3.name].push(payload.to_i)
+        end # subscribe
+      end
+
+      amqp_before :each do
+        @received_messages = {
+          @queue1.name => [],
+          @queue2.name => [],
+          @queue3.name => []
+        }
+
+        @expected_number_of_messages = {
+          @queue1.name => 100,
+          @queue2.name => 100,
+          @queue3.name => 100
+        }
+      end
+
+      amqp_after :each do
+        @sent_values.clear
+      end
+
+
+      context "and messages are published as non-mandatory" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :mandatory => false)
+          end
+
+          # for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+
+      context "and messages are published as mandatory" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :mandatory => true)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+
+      context "and messages are published as non-persistent" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :persistent => false)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+
+      context "and messages are published as persistent" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :persistent => true)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+      context "and messages are published as non-immediate" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :immediate => false)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+
+      context "and messages are published as immediate" do
+        it "may get a Basic.Return back"
+      end # context
+
+
+
+      context "and messages are published WITHOUT routing key" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+
+
+      context "and messages are published WITH routing key that matches name of one of the queues" do
+        it "routes all messages to all bound queues" do
+          100.times do
+            dispatched_data             = rand(5_000_000)
+            @sent_values.push(dispatched_data)
+
+            @exchange.publish(dispatched_data, :routing_key => @queues.sample.name)
+          end
+
+          # 6 seconds are for Rubinius, it is surprisingly slow on this workload
+          done(1.5) {
+            [@queue1, @queue2, @queue3].each do |q|
+              @received_messages[q.name].size.should == @expected_number_of_messages[q.name]
+
+              # this one is ordering assertion
+              @received_messages[q.name].should == @sent_values
+            end
+          }
+        end # it
+      end # context
+    end # context
+  end # context
+end # describe

data/spec/spec_helper.rb

@@ -1,4 +1,4 @@
-# encoding: utf-8
+# -*- coding: utf-8 -*-
 
 $LOAD_PATH.unshift File.expand_path("../../lib", __FILE__)
 
@@ -6,7 +6,8 @@ require 'bundler'
 Bundler.setup
 Bundler.require :default, :test
 
-require "mq"
+require "amqp-spec"
+require "amqp"
 
 amqp_config = File.dirname(__FILE__) + '/amqp.yml'
 
@@ -26,8 +27,12 @@ else
   AMQP_OPTS = {:host => 'localhost', :port => 5672}
 end
 
+#
+# Ruby version-specific
+#
 
-if RUBY_VERSION == "1.8.7"
+case RUBY_VERSION
+when "1.8.7" then
   module ArrayExtensions
     def sample
       self.choice
@@ -37,18 +42,17 @@ if RUBY_VERSION == "1.8.7"
   class Array
     include ArrayExtensions
   end
+when "1.8.6" then
+  raise "Ruby 1.8.6 is not supported. Sorry, pal. Time to move on beyond One True Ruby. Yes, time flies by."
+when /^1.9/ then
+  puts "Encoding.default_internal was #{Encoding.default_internal || 'not set'}, switching to #{Encoding::UTF_8}"
+  Encoding.default_internal = Encoding::UTF_8
+
+  puts "Encoding.default_external was #{Encoding.default_internal || 'not set'}, switching to #{Encoding::UTF_8}"
+  Encoding.default_external = Encoding::UTF_8
 end
 
 
-
-# Shorthand for mocking subject's instance variable
-def subject_mock(name, as_null = false)
-  mock = mock(name)
-  mock.as_null_object if as_null
-  subject.instance_variable_set(name.to_sym, mock)
-  mock
-end
-
 # Returns Header that should be correctly parsed
 def basic_header(opts = {})
   AMQP::Frame::Header.new(
@@ -74,23 +78,3 @@ def test_method_deliver opts = {}
       AMQP::Protocol::Basic::Deliver.new(
           :consumer_tag => opts[:consumer_tag] || 'test_consumer'))
 end
-
-require "stringio"
-
-def capture_stdout(&block)
-  $stdout = StringIO.new
-  block.call
-  $stdout.rewind
-  result = $stdout.read
-  $stdout = STDOUT
-  return result
-end
-
-def capture_stderr(&block)
-  $stderr = StringIO.new
-  block.call
-  $stderr.rewind
-  result = $stderr.read
-  $stderr = STDOUT
-  return result
-end

data/spec/unit/{mq/mq_basic_spec.rb → amqp/basic_spec.rb}

@@ -2,7 +2,7 @@
 
 require 'spec_helper'
 
-describe MQ do
+describe AMQP do
 
   #
   # Environment
@@ -13,7 +13,7 @@ describe MQ do
   default_timeout 5
 
   amqp_before do
-    @channel = MQ.new
+    @channel = AMQP::Channel.new
   end
 
 
@@ -25,7 +25,7 @@ describe MQ do
   describe ".channel" do
     it 'gives each thread a separate channel' do
       pending 'This is not implemented in current lib'
-      class MQ
+      module AMQP
         @@cur_channel = 0
       end
 
@@ -36,4 +36,4 @@ describe MQ do
       done
     end
   end
-end # describe MQ
+end # describe AMQP

data/spec/unit/{mq → amqp}/collection_spec.rb

@@ -1,14 +1,14 @@
 # encoding: utf-8
 
 require "spec_helper"
-require "mq/collection"
+require "amqp/collection"
 
 Item = Struct.new(:name)
 
-describe MQ::Collection do
+describe AMQP::Collection do
   before do
     @items = 3.times.map { |int| Item.new("name-#{int}") }
-    @collection = MQ::Collection.new(@items)
+    @collection = AMQP::Collection.new(@items)
   end
 
   it "provides access to items by name" do
@@ -23,7 +23,7 @@ describe MQ::Collection do
 
   describe "#<<" do
     it "should raise IncompatibleItemError if the argument doesn't have method :name" do
-      lambda { @collection << nil }.should raise_error(MQ::Collection::IncompatibleItemError)
+      lambda { @collection << nil }.should raise_error(AMQP::Collection::IncompatibleItemError)
     end
 
     it "should add an item into the collection" do
@@ -88,7 +88,7 @@ describe MQ::Collection do
   describe "#add!" do
     context "when the argument doesn't respond to :name" do
       it "should raise IncompatibleItemError " do
-        lambda { @collection.add!(nil) }.should raise_error(MQ::Collection::IncompatibleItemError)
+        lambda { @collection.add!(nil) }.should raise_error(AMQP::Collection::IncompatibleItemError)
       end # it
     end
 
@@ -114,7 +114,7 @@ describe MQ::Collection do
   describe "#<<" do
     context "when the argument doesn't respond to :name" do
       it "should raise IncompatibleItemError " do
-        lambda { @collection << nil }.should raise_error(MQ::Collection::IncompatibleItemError)
+        lambda { @collection << nil }.should raise_error(AMQP::Collection::IncompatibleItemError)
       end # it
     end # context
 
@@ -126,4 +126,19 @@ describe MQ::Collection do
       end
     end # context
   end # describe
-end # describe MQ::Collection
+
+
+
+
+
+  describe "#delete" do
+    it "should remove item with the given name from the collection" do
+      items = [Item.new("test-0"), Item.new("test-1")]
+      collection = AMQP::Collection.new(items)
+      lambda {
+        collection.delete("test-1")
+      }.should change(collection, :length).by(-1)
+      collection.should_not include("test-1")
+    end
+  end # describe
+end # describe AMQP::Collection

data/spec/unit/amqp/connection_spec.rb

@@ -0,0 +1,116 @@
+require 'spec_helper'
+require 'amqp'
+
+describe AMQP, 'class object' do
+
+  #
+  # Environment
+  #
+
+  subject { AMQP }
+
+  #
+  # Examples
+  #
+
+  its(:settings) do
+    should == {
+      :host    => "127.0.0.1",
+      :port    => 5672,
+      :user    => "guest",
+      :pass    => "guest",
+      :vhost   => "/",
+      :timeout => nil,
+      :logging => false,
+      :ssl     => false
+    }
+  end
+
+  its(:client) { should == AMQP::BasicClient }
+
+
+
+
+  describe 'logging' do
+    after(:all) do
+      AMQP.logging = false
+    end
+
+    it 'is silent by default' do
+      AMQP.logging.should be_false
+    end
+  end # .logging=
+
+
+
+
+  describe '.start' do
+
+    #
+    # Environment
+    #
+
+    include AMQP::SpecHelper
+
+    em_before { AMQP.cleanup_state }
+    em_after  { AMQP.cleanup_state }
+
+    #
+    # Examples
+    #
+
+    it 'yields to given block AFTER connection is established' do
+      em do
+        AMQP.start AMQP_OPTS do
+          @block_fired = true
+
+          AMQP.connection.should be_connected
+        end
+        done(0.1) { @block_fired.should be_true }
+      end
+    end
+  end # .start
+
+
+
+
+  describe '.stop' do
+    context "when connection is not established" do
+      it 'is a no-op' do
+        expect { @res = AMQP.stop }.to_not raise_error
+        @res.should be_nil
+      end # it
+    end # context
+
+
+    context 'with established AMQP connection' do
+
+      #
+      #
+      #
+
+      include AMQP::Spec
+      after           { AMQP.cleanup_state; done }
+      default_options AMQP_OPTS
+
+      #
+      # Examples
+      #
+
+      it 'properly closes AMQP broker connection and fires a callback. Mind the delay!' do
+        AMQP.start(AMQP_OPTS)
+        AMQP.connection.should be_connected
+
+        @block_has_fired = false
+
+        AMQP.stop do
+          @block_has_fired = true
+        end
+        AMQP.connection.should_not be_nil
+        done(0.1) do
+          @block_has_fired.should be_true
+        end
+      end # it
+    end # context
+  end # describe
+end # describe AMQP