class DescribeStartConsuming(_BaseTestCase):
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.add_on_cancel_callback = MagicMock()
self.agent.channel = MagicMock()
self.agent.channel.basic_consume.return_value = sentinel.consumer_tag
def execute(self):
self.agent.start_consuming(sentinel.queue)
def should_add_on_cancel_callback(self):
self.agent.add_on_cancel_callback.assert_called_once_with()
def should_call_basic_consume(self):
self.agent.channel.basic_consume.assert_called_once_with(
consumer_callback=self.agent.process,
queue=sentinel.queue,
no_ack=not self.agent._ack)
def should_set_consumer_tag(self):
self.assertIs(self.agent._consumer_tags[sentinel.queue],
sentinel.consumer_tag)
class DescribeStartConsuming(_BaseTestCase):
def configure(self):
self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker,
sentinel.bindings)
self.agent.add_on_cancel_callback = MagicMock()
self.agent.channel = MagicMock()
self.agent.channel.basic_consume.return_value = sentinel.consumer_tag
def execute(self):
self.agent.start_consuming(sentinel.queue)
def should_add_on_cancel_callback(self):
self.agent.add_on_cancel_callback.assert_called_once_with()
def should_call_basic_consume(self):
self.agent.channel.basic_consume.assert_called_once_with(
consumer_callback=self.agent.process,
queue=sentinel.queue,
no_ack=not self.agent._ack)
def should_set_consumer_tag(self):
self.assertIs(self.agent._consumer_tags[sentinel.queue],
sentinel.consumer_tag)
