def test_queue_errors(self): missing_settings = {} with self.assertRaises(SQSException) as context: sqs = SQSQueue(missing_settings) self.assertIn( 'Invalid SQS configuration', str(context.exception)) invalid_settings = { 'QUEUE_ARN': 'abc', 'KEY_ID': '', 'KEY': '', } with self.assertRaises(SQSException) as context: sqs = SQSQueue(invalid_settings) self.assertIn('Invalid ARN', str(context.exception))
def __init__(self, processor=None, exception=None, sqs_settings=None): """ :param processor: A sub-class object of MessageBodyProcessor """ if not processor: raise GatherException('missing event processor') self._processor = processor self._settings = sqs_settings if ( sqs_settings) else self._processor.get_queue_settings() self._queue = SQSQueue(self._settings)
def setUp(self): self._mock_settings = { 'QUEUE_ARN': 'arn:aws:sqs:xx-mock-999:000000000000:ww-wwww-1', 'KEY_ID': 'XXXXXXXXXXXXXXXX', 'KEY': 'YYYYYYYYYYYYYYYYYYYYYYYY', 'WAIT_TIME': 10, 'VISIBILITY_TIMEOUT': 10, 'MESSAGE_GATHER_SIZE': 10, 'POLL_COUNT': 10, } self.sqs = SQSQueue(self._mock_settings) self.sqs._queue = MockQueue()