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)