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)
