def test_requeue_quarantined_messages(self):
msg = DummyMessage('Oops, my bad.')
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(msg, (None, None, None))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
self.assertEqual(len(queue), 0)
self.assertEqual(len(list(queue.get_quarantined_messages())), 3)
queue.requeue_quarantined_messages()
self.assertEqual(len(queue), 3)
self.assertEqual(len(list(queue.get_quarantined_messages())), 0)
def test_requeue_quarantined_messages(self):
msg = DummyMessage('Oops, my bad.')
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(msg, (None, None, None))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
self.assertEqual(len(queue), 0)
self.assertEqual(len(list(queue.get_quarantined_messages())), 3)
queue.requeue_quarantined_messages()
self.assertEqual(len(queue), 3)
self.assertEqual(len(list(queue.get_quarantined_messages())), 0)
def test_remove_from_quarantine(self):
msg = DummyMessage('Oops, my bad.')
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(msg, (None, None, None))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 3)
queue.remove_from_quarantine(msg)
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 2)
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Oh nos!')
self.assertEqual(error, ('OMG', 'WTH', '???'))
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Woopsy!')
self.assertEqual(error, ('IRCC', 'FWIW', 'ROTFLMAO'))
def test_remove_from_quarantine(self):
msg = DummyMessage('Oops, my bad.')
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(msg, (None, None, None))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 3)
queue.remove_from_quarantine(msg)
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 2)
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Oh nos!')
self.assertEqual(error, ('OMG', 'WTH', '???'))
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Woopsy!')
self.assertEqual(error, ('IRCC', 'FWIW', 'ROTFLMAO'))
def test_quarantine(self):
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 2)
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Oh nos!')
self.assertEqual(error, ('OMG', 'WTH', '???'))
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Woopsy!')
self.assertEqual(error, ('IRCC', 'FWIW', 'ROTFLMAO'))
self.assertEqual(queue.get_quarantined_message(msg['X-Postoffice-Id']),
msg)
def test_quarantine(self):
queue = self._make_one()
queue.quarantine(DummyMessage('Oh nos!'), ('OMG', 'WTH', '???'))
queue.quarantine(DummyMessage('Woopsy!'), ('IRCC', 'FWIW', 'ROTFLMAO'))
msgs = list(queue.get_quarantined_messages())
self.assertEqual(len(msgs), 2)
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Oh nos!')
self.assertEqual(error, ('OMG', 'WTH', '???'))
msg, error = msgs.pop(0)
self.assertEqual(msg, 'Woopsy!')
self.assertEqual(error, ('IRCC', 'FWIW', 'ROTFLMAO'))
self.assertEqual(
queue.get_quarantined_message(msg['X-Postoffice-Id']), msg
)
