def test_requeue_message_no_previous_retries(self): mock_channel = Mock() message = {} message['court'] = 'nysd' message['casenum'] = '1234' success = ia_uploader._requeue_message(mock_channel, message) self.assertTrue(success) self.assertEquals(1, message['attempt_num'])
def test_requeue_message(self): mock_channel = Mock() # First make sure that it doesn't requeue after too many attempts message = {} message['court'] = 'nysd' message['casenum'] = '1234' message['attempt_num'] = 15 message['next_attempt_time'] = 200 success = ia_uploader._requeue_message(mock_channel, message) self.assertFalse(success) self.assertEquals(0, len(mock_channel.method_calls)) message['attempt_num'] = 1 success = ia_uploader._requeue_message(mock_channel, message) self.assertTrue(success) self.assertEquals(1, len(mock_channel.method_calls))