From 3e4bc2525f069e43f93f92588f4260a485b78765 Mon Sep 17 00:00:00 2001
From: Michael Klishin <mklishin@gopivotal.com>
Date: Wed, 29 Jan 2014 17:08:27 +0400
Subject: [PATCH] Tests for #186

---
 .../integration/basic_consume_spec.rb         | 119 ++++++++++++++++++
 1 file changed, 119 insertions(+)

diff --git a/spec/higher_level_api/integration/basic_consume_spec.rb b/spec/higher_level_api/integration/basic_consume_spec.rb
index c3fe938e7..544df0452 100644
--- a/spec/higher_level_api/integration/basic_consume_spec.rb
+++ b/spec/higher_level_api/integration/basic_consume_spec.rb
@@ -222,4 +222,123 @@
     end
   end
 
+
+  context "with uncaught exceptions in delivery handler" do
+    context "and defined exception handler" do
+      let(:queue_name) { "bunny.basic_consume#{rand}" }
+
+      it "uses exception handler" do
+        caught = nil
+        t = Thread.new do
+          ch = connection.create_channel
+          q  = ch.queue(queue_name, :auto_delete => true, :durable => false)
+
+          ch.on_uncaught_exception do |e, consumer|
+            caught = e
+          end
+
+          q.subscribe(:exclusive => false, :manual_ack => false) do |delivery_info, properties, payload|
+            raise RuntimeError.new(queue_name)
+          end
+        end
+        t.abort_on_exception = true
+        sleep 0.5
+
+        ch     = connection.create_channel
+        x  = ch.default_exchange
+        x.publish("hello", :routing_key => queue_name)
+        sleep 0.5
+
+        caught.message.should == queue_name
+
+        ch.close
+      end
+    end
+
+
+    context "and default exception handler" do
+      let(:queue_name) { "bunny.basic_consume#{rand}" }
+
+      it "uses exception handler" do
+        t = Thread.new do
+          ch = connection.create_channel
+          q  = ch.queue(queue_name, :auto_delete => true, :durable => false)
+
+          q.subscribe(:exclusive => false, :manual_ack => false) do |delivery_info, properties, payload|
+            raise RuntimeError.new(queue_name)
+          end
+        end
+        t.abort_on_exception = true
+        sleep 0.5
+
+        ch     = connection.create_channel
+        x  = ch.default_exchange
+        5.times { x.publish("hello", :routing_key => queue_name) }
+        sleep 1.5
+
+        ch.close
+      end
+    end
+
+
+    context "with a single consumer" do
+      let(:queue_name) { "bunny.basic_consume#{rand}" }
+
+      it "provides delivery tag access" do
+        delivery_tags = SortedSet.new
+
+        cch = connection.create_channel
+        q = cch.queue(queue_name, :auto_delete => true, :durable => false)
+        q.subscribe(:exclusive => false, :manual_ack => false) do |delivery_info, properties, payload|
+          delivery_tags << delivery_info.delivery_tag
+        end
+        sleep 0.5
+
+        ch = connection.create_channel
+        x  = ch.default_exchange
+        100.times do
+          x.publish("hello", :routing_key => queue_name)
+        end
+
+        sleep 1.0
+        delivery_tags.should == SortedSet.new(Range.new(1, 100).to_a)
+
+        ch.queue(queue_name, :auto_delete => true, :durable => false).message_count.should == 0
+
+        ch.close
+      end
+    end
+
+
+    context "with multiple consumers on the same channel" do
+      let(:queue_name) { "bunny.basic_consume#{rand}" }
+
+      it "provides delivery tag access" do
+        delivery_tags = SortedSet.new
+
+        cch = connection.create_channel
+        q   = cch.queue(queue_name, :auto_delete => true, :durable => false)
+
+        7.times do
+          q.subscribe(:exclusive => false, :manual_ack => false) do |delivery_info, properties, payload|
+            delivery_tags << delivery_info.delivery_tag
+          end
+        end
+        sleep 1.0
+
+        ch = connection.create_channel
+        x  = ch.default_exchange
+        100.times do
+          x.publish("hello", :routing_key => queue_name)
+        end
+
+        sleep 1.5
+        delivery_tags.should == SortedSet.new(Range.new(1, 100).to_a)
+
+        ch.queue(queue_name, :auto_delete => true, :durable => false).message_count.should == 0
+
+        ch.close
+      end
+    end
+  end
 end # describe