Sha256: 582a47aa4d7ef985a51cdfa18b985b54c4f87418e6dda3504e835b039c730019

Contents?: true

Size: 1.65 KB

Versions: 10

Compression:

Stored size: 1.65 KB

Contents

# -*- coding: utf-8 -*-
require "spec_helper"

unless ENV["CI"]
  describe "Concurrent consumers sharing a connection" do
    before :all do
      @connection = Bunny.new(:user => "bunny_gem", :password => "bunny_password", :vhost => "bunny_testbed",
                    :automatic_recovery => false, :continuation_timeout => 6000)
      @connection.start
    end

    after :all do
      @connection.close
    end

    def any_not_drained?(qs)
      qs.any? { |q| !q.message_count.zero? }
    end

    context "when publishing thousands of messages over 128K in size" do
      let(:colors) { ["red", "blue", "white"] }

      let(:n) { 32 }
      let(:m) { 1000 }

      it "successfully drain all queues" do
        ch   = @connection.create_channel
        body = "абвг"
        x    = ch.topic("bunny.stress.concurrent.consumers.topic", :durable => true)

        chs  = {}
        n.times do |i|
          chs[i] = @connection.create_channel
        end
        qs   = []

        n.times do |i|
          t = Thread.new do
            cht = chs[i]

            q = cht.queue("", :exclusive => true)
            q.bind(x.name, :routing_key => colors.sample).subscribe do |delivery_info, meta, payload|
              # no-op
            end
            qs << q
          end
          t.abort_on_exception = true
        end

        sleep 1.0

        5.times do |i|
          m.times do
            x.publish(body, :routing_key => colors.sample)
          end
          puts "Published #{(i + 1) * m} messages..."
        end

        while any_not_drained?(qs)
          sleep 1.0
        end
        puts "Drained all the queues..."

        ch.close
      end
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version Path
bunny-2.3.1 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.3.0 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.2.2 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.2.1 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.2.0 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.1.0 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.0.1 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.0.0 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.0.0.rc2 spec/stress/concurrent_consumers_stress_spec.rb
bunny-2.0.0.rc1 spec/stress/concurrent_consumers_stress_spec.rb