class WhenProcessingMessage(_BaseTestCase):
def configure(self):
self.consumer = MagicMock()
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
def execute(self):
self.result = self.agent._process(self.message)
def should_call_consumer_process(self):
self.consumer.process.assert_called_once_with(self.message)
def should_return_true(self):
self.assertTrue(self.result)
class WhenProcessingMessage(_BaseTestCase):
def configure(self):
self.consumer = MagicMock()
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
def execute(self):
self.result = self.agent._process(self.message)
def should_call_consumer_process(self):
self.consumer.process.assert_called_once_with(self.message)
def should_return_true(self):
self.assertTrue(self.result)
class _BaseProcessingErrorTestCase(_BaseTestCase):
def configure(self):
self.consumer = MagicMock()
self.consumer.process.side_effect = self.exc
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
self.agent.reject = MagicMock()
self.agent.stop = MagicMock()
self.agent._record_exception = MagicMock()
self.agent.processing_failure = MagicMock()
self.agent.reconnect = MagicMock()
def execute(self):
self.result = self.agent._process(self.message)
def should_call_consumer_process(self):
self.consumer.process.assert_called_once_with(self.message)
def should_return_false(self):
self.assertFalse(self.result)
class _BaseProcessingErrorTestCase(_BaseTestCase):
def configure(self):
self.consumer = MagicMock()
self.consumer.process.side_effect = self.exc
self.message = NonCallableMagicMock()
self.agent = ConsumerAgent(self.consumer, sentinel.broker,
sentinel.bindings)
self.agent.reject = MagicMock()
self.agent.stop = MagicMock()
self.agent._record_exception = MagicMock()
self.agent.processing_failure = MagicMock()
self.agent.reconnect = MagicMock()
def execute(self):
self.result = self.agent._process(self.message)
def should_call_consumer_process(self):
self.consumer.process.assert_called_once_with(self.message)
def should_return_false(self):
self.assertFalse(self.result)
