示例#1
0
 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)
示例#2
0
    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')
示例#3
0
    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': []}
        ])
示例#4
0
    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)
示例#5
0
 def setUp(self):
     super().setUp()
     self.amazon = AmazonMediaStorage(self.app)
示例#6
0
 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)
示例#7
0
 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')
示例#8
0
 def setUp(self):
     self.amazon = AmazonMediaStorage(self.app)