Merge pull request #547 from markcf/master

#544 get_waiting_message_count()

pika/adapters/blocking_connection.py

@@ -733,6 +733,13 @@ def consume(self, queue, no_ack=False, exclusive=False):
                 yield self._generator_messages.pop(0)
             self.connection.process_data_events()
 
+    def get_waiting_message_count(self):
+        """Returns the amount of messages waiting in the generator messages list.
+
+        :rtype: int
+        """
+        return len(self._generator_messages)
+
     def force_data_events(self, enable):
         """Turn on and off forcing the blocking adapter to stop and look to see
         if there are any frames from RabbitMQ in the read buffer. By default

tests/unit/blocking_channel_tests.py

@@ -53,3 +53,18 @@ def test_init_initial_value_wait(self):
     def test_init_open_called(self):
         self._open.assert_called_once_with()
 
+
+    def test_init_create_generator(self):
+        self.obj.consume("queue")
+        self.assertEquals(self.obj.get_waiting_message_count(), 0)
+
+
+    def test_init_value_waiting_message_count(self):
+        self.obj.consume("queue")
+        self.obj._generator_messages.append("test")
+        self.assertEquals(self.obj.get_waiting_message_count(), 1)
+        self.obj._generator_messages.append("test")
+        self.assertEquals(self.obj.get_waiting_message_count(), 2)
+        self.obj._generator_messages.pop(0)
+        self.assertEquals(self.obj.get_waiting_message_count(), 1)
+