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_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 )