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)
