def test_remove_from_quarantine_not_in_quarantine(self): msg = DummyMessage('Oops, my bad') queue = self._make_one() queue.quarantine(msg, (None, None, None)) self.assertEqual(queue.count_quarantined_messages(), 1) queue.remove_from_quarantine(msg) self.assertEqual(queue.count_quarantined_messages(), 0) self.assertRaises(ValueError, queue.remove_from_quarantine, msg)
def test_remove_from_quarantine_bad_id(self): msg = DummyMessage('Oops, my bad') queue = self._make_one() queue.quarantine(msg, (None, None, None)) self.assertEqual(queue.count_quarantined_messages(), 1) id = msg['X-Postoffice-Id'] queue.remove_from_quarantine(msg) msg['X-Postoffice-Id'] = id self.assertEqual(queue.count_quarantined_messages(), 0) self.assertRaises(ValueError, queue.remove_from_quarantine, msg)
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'))