test limiting after consumer subscription

Author: Matthias Radestock

test/src/com/rabbitmq/client/test/functional/QosTests.java

@@ -159,6 +159,19 @@ public void testFairness()
 
     }
 
+    public void testSetLimitAfterConsume()
+        throws IOException
+    {
+        QueueingConsumer c = new QueueingConsumer(channel);
+        String queue = declareBindConsume(c);
+        channel.basicQos(1);
+        fill(2);
+        //We actually only guarantee that the limit takes effect
+        //*eventually*, so this can in fact fail. It's pretty unlikely
+        //though.
+        drain(c, 1);
+    }
+
     protected void runLimitTests(int limit,
                                  boolean multiAck,
                                  boolean txMode,
@@ -249,11 +262,7 @@ protected List<String> configure(QueueingConsumer c,
         //declare/bind/consume-from queues
         List <String> queues = new ArrayList<String>();
         for (int i = 0; i < queueCount; i++) {
-            AMQP.Queue.DeclareOk ok = channel.queueDeclare();
-            String queue = ok.getQueue();
-            queues.add(queue);
-            channel.queueBind(queue, "amq.fanout", "");
-            channel.basicConsume(queue, false, c);
+            queues.add(declareBindConsume(c));
         }
 
         //publish
@@ -262,6 +271,16 @@ protected List<String> configure(QueueingConsumer c,
         return queues;
     }
 
+    protected String declareBindConsume(QueueingConsumer c)
+        throws IOException
+    {
+        AMQP.Queue.DeclareOk ok = channel.queueDeclare();
+        String queue = ok.getQueue();
+        channel.queueBind(queue, "amq.fanout", "");
+        channel.basicConsume(queue, false, c);
+        return queue;
+    }
+
     protected void ackDelivery(Delivery d, boolean multiple)
         throws IOException
     {