async def test_process_success_from_reject_to_requeue(self):
message = RabbitMQMessage(body={},
delivery_tag=10,
on_success=Actions.REJECT)
message.reject(requeue=True)
await message.process_success(self.queue_mock)
self.queue_mock.reject.assert_awaited_with(delivery_tag=10,
requeue=True)
self.queue_mock.ack.assert_not_awaited()
async def test_process_exception_from_requeue_to_reject(self):
message = RabbitMQMessage(body={},
delivery_tag=10,
on_exception=Actions.REQUEUE)
message.reject(requeue=False)
await message.process_exception(self.queue_mock)
self.queue_mock.reject.assert_awaited_with(delivery_tag=10,
requeue=False)
self.queue_mock.ack.assert_not_awaited()
async def test_process_success_from_reject_to_requeue(self):
message = RabbitMQMessage(
delivery_tag=10,
on_success=Actions.REJECT,
amqp_message=self.amqp_message,
)
message.reject(requeue=True)
await message.process_success()
self.amqp_message.reject.assert_awaited_with(requeue=True)
self.amqp_message.ack.assert_not_awaited()
async def test_process_exception_from_requeue_to_reject(self):
message = RabbitMQMessage(
delivery_tag=10,
on_exception=Actions.REQUEUE,
amqp_message=self.amqp_message,
)
message.reject(requeue=False)
await message.process_exception()
self.amqp_message.reject.assert_awaited_with(requeue=False)
self.amqp_message.ack.assert_not_awaited()
async def test_process_message_to_be_rejected(self):
"""
Sempre fazemos reject com requeue.
"""
expected_body = {"key": "value"}
message = RabbitMQMessage(body=expected_body, delivery_tag=42)
message.reject()
queue_mock = CoroutineMock(ack=CoroutineMock(), reject=CoroutineMock())
await message.process(queue_mock)
queue_mock.reject.assert_awaited_once_with(delivery_tag=42, requeue=True)
self.assertEqual(0, queue_mock.ack.await_count)
def test_mark_message_to_be_rejected(self):
message = RabbitMQMessage({}, 10)
message.reject()
self.assertFalse(message._do_ack)