def get_SQSqueue():
sqsregioninfo = SQSRegionInfo(name='localhost_region', endpoint='localhost')
connection = sqsregioninfo.connect(
port=8001,
aws_access_key_id='id',
aws_secret_access_key='secret',
is_secure=False
)
return connection.get_queue('test_queue')
def setUp(self):
sqsregioninfo = SQSRegionInfo(name='localhost_region', endpoint='localhost')
self.connection = sqsregioninfo.connect(
port=8001,
aws_access_key_id='id',
aws_secret_access_key='secret',
is_secure=False
)
self.queue = self.connection.create_queue('test_queue')
client = StatsClient(host='localhost', port=8125, prefix=None, maxudpsize=512)
self.statsd = client.pipeline()
def wait_sqs(host="localhost", port=9324, **kwargs):
""""Wait for sqs to be up.
Args:
host (Optional[str]): Host name. Default to "localhost".
port (Optional[int]): sqs port. Defaults to 8000.
Keyword Args:
fake_region (str): The fake region for you local sqs
region (str): The region for the aws sqs
secure (bool): The communication is secure, default value False
aws_access_key_id (str): Access key for aws (boto), default value foo
aws_secret_access_key (str): Secret key for aws (boto), default value bar
Returns:
bool: True if it is ready, False if something went wrong
"""
if "fake_region" in kwargs:
fake_region = SQSRegionInfo(name=kwargs.get("fake_region", "loktar"), endpoint=host)
client = fake_region.connect(port=port,
is_secure=kwargs.get("secure", False),
aws_access_key_id=kwargs.get("aws_access_key_id", "foo"),
aws_secret_access_key=kwargs.get("aws_secret_access_key", "bar"))
else:
client = sqs_connect_to_region(kwargs.get("region", "eu-west-1"),
port=port,
is_secure=kwargs.get("secure", False),
aws_access_key_id=kwargs.get("aws_access_key_id", "foo"),
aws_secret_access_key=kwargs.get("aws_secret_access_key", "bar"))
try:
logger.info("Waiting sqs (5min max internal boto retry)")
client.get_all_queues()
except socket.gaierror:
logger.error("Cannot connect to sqs.")
logger.error("Aborting")
return False
logger.info("sqs is ready")
return True
