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