async def test_process_success_from_requeue_to_ack(self):
message = RabbitMQMessage(body={},
delivery_tag=10,
on_success=Actions.REQUEUE)
message.accept()
await message.process_success(self.queue_mock)
self.queue_mock.ack.assert_awaited_with(delivery_tag=10)
self.queue_mock.reject.assert_not_awaited()
async def test_process_success_from_reject_to_ack(self):
message = RabbitMQMessage(
delivery_tag=10,
on_success=Actions.REJECT,
amqp_message=self.amqp_message,
)
message.accept()
await message.process_success()
self.amqp_message.ack.assert_awaited_with()
self.amqp_message.reject.assert_not_awaited()
async def test_process_exception_from_requeue_to_ack(self):
message = RabbitMQMessage(
delivery_tag=10,
on_exception=Actions.REQUEUE,
amqp_message=self.amqp_message,
)
message.accept()
await message.process_exception()
self.amqp_message.ack.assert_awaited_with()
self.amqp_message.reject.assert_not_awaited()
def test_mark_message_to_be_acked(self):
message = RabbitMQMessage({}, 10)
message._do_ack = False
message.accept()
self.assertTrue(message._do_ack)