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