def test_sqs_queue_timeout(bare_sqs_queue, handler): q = SQSQueue(bare_sqs_queue.name, handler) start = time.time() q.drain(timeout=1) duration = time.time() - start error = abs(duration - 1.0) assert error < 0.1 assert handler.messages == []
def test_sqs_queue_drain(bare_sqs_queue, handler): q = SQSQueue(bare_sqs_queue.name, handler) expected_msg = {'foo': 'bar'} msg = bare_sqs_queue.new_message(json.dumps(expected_msg)) bare_sqs_queue.write(msg) q.drain(timeout=1) assert handler.messages == [expected_msg] handler.messages = [] q.drain(timeout=1) assert handler.messages == []
def maker(queue_name): q = bare_sqs_queue_maker(queue_name) return SQSQueue(q.name)
def __init__(self): self.messages = [] q = bare_sqs_queue_maker('reporter-queue') self._queue = SQSQueue(q.name, self.handler) sns_connection.subscribe_sqs_queue(sns_topic_arn, q)