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