def setUp(self): self.app.config['AMAZON_CONTAINER_NAME'] = 'AMAZON_CONTAINER_NAME' self.app.config['AMAZON_REGION'] = 'us-east-1' self.app.config['AMAZON_S3_USE_HTTPS'] = True self.app.config['AMAZON_SERVE_DIRECT_LINKS'] = True self.app.config['AMAZON_SERVER'] = 'amazonaws.com' self.amazon = AmazonMediaStorage(self.app)
def test_url_for_media(self): app = MagicMock() app.config = self.config url = AmazonMediaStorage(app).url_for_media('media_id.jpg') self.assertEqual( url, 'http://test_container.s3-test_region.amazonaws.com/media_id.jpg') app.config['AMAZON_S3_USE_HTTPS'] = True url = AmazonMediaStorage(app).url_for_media('media_id.jpg') self.assertEqual( url, 'https://test_container.s3-test_region.amazonaws.com/media_id.jpg') app.config['AMAZON_URL_GENERATOR'] = 'partial' url = AmazonMediaStorage(app).url_for_media('media_id.jpg') self.assertEqual( url, 'https://test_container.s3-test_region.amazonaws.com/media_id.jpg')
def setUp(self): self.amazon = AmazonMediaStorage(self.app) self.amazon.client = Mock() # Mock getting list of files from Amazon, first request returns a file, second request returns empty list self.amazon.client.list_objects = Mock(side_effect=[ { 'Contents': [{'Key': 'gridtest/abcd1234', 'LastModified': utcnow() - timedelta(minutes=30), 'Size': 500, 'ETag': 'abcd1234'}] }, {'Contents': []} ])
def setUp(self): self.amazon = AmazonMediaStorage(self.app) # Patch config with defaults p = patch.dict(self.app.config, { 'AMAZON_CONTAINER_NAME': 'acname', 'AMAZON_REGION': 'us-east-1', 'AMAZON_S3_USE_HTTPS': True, 'AMAZON_SERVE_DIRECT_LINKS': True, 'AMAZON_SERVER': 'amazonaws.com', 'AMAZON_S3_SUBFOLDER': '', }) p.start() self.addCleanup(p.stop)
def setUp(self): super().setUp() self.amazon = AmazonMediaStorage(self.app)
def setUp(self): super().setUp() self.app.config['AMAZON_CONTAINER_NAME'] = 'AMAZON_CONTAINER_NAME' self.app.config['AMAZON_S3_USE_HTTPS'] = True self.app.config['AMAZON_SERVE_DIRECT_LINKS'] = True self.amazon = AmazonMediaStorage(self.app)
def test_media_id(self): app = MagicMock() app.config = self.config media_id = AmazonMediaStorage(app).media_id('media_id', 'image/jpeg') self.assertEqual(media_id.split('/')[1], 'media_id.jpg')
def setUp(self): self.amazon = AmazonMediaStorage(self.app)