def test_upload_file(self): """ Tests the upload of a file using moto. Ideally, creation of bucket should be moved to setUp(), but moto creates new s3 mock with every decorator """ self.connection = s3_tools.connect() self.test_bucket = self.connection.create_bucket('mybucket') self.test_file_location = 'test/assets/rob.png' s3_tools.upload_file_to_bucket(self.test_bucket,'foo', self.test_file_location) key = boto.s3.key.Key(self.test_bucket) key.key = 'foo' assert key.get_contents_as_string() == open(self.test_file_location).read()
def __init__(self, connection_params): self.logger = logging.getLogger(__name__) self.logger.debug("Got here") try: rmq_connection = pika.BlockingConnection(connection_params) self.logger.debug("Obtained connection to rmq") except Exception: self.logger.fatal("No connection to rmq available") self.rmq_channel = rmq_connection.channel() self.rmq_channel.basic_qos(prefetch_count=1) self.result_queue = self.rmq_channel.queue_declare(queue="user_images", durable=True) self.result_queue_name = self.result_queue.method.queue self.image_creation_queue = self.rmq_channel.queue_declare(queue="user_image_creation", durable=True) self.creation_queue_name = self.image_creation_queue.method.queue self.s3_connection = s3_tools.connect()