def test_download_url_filename(self):
obj = mock.Mock()
obj.key = 'key'
obj.bucket = 'bucket'
resource.get_download_url(obj, filename='data.zip')
assert obj.meta.client.generate_presigned_url.called_once_with(
'get_object',
Params={
'Key': 'key',
'Bucket': 'bucket',
'ResponseContentDisposition': 'filename=data.zip',
},
ExpiresIn=resource.URL_EXPIRY,
)
def test_download_url_filename(self):
obj = mock.Mock()
obj.key = 'key'
obj.bucket = 'bucket'
resource.get_download_url(obj, filename='data.zip')
assert obj.meta.client.generate_presigned_url.called_once_with(
'get_object',
Params={
'Key': 'key',
'Bucket': 'bucket',
'ResponseContentDisposition': 'filename=data.zip',
},
ExpiresIn=resource.URL_EXPIRY,
)
def test_download_url(self):
obj = mock.Mock()
obj.key = 'key'
obj.bucket = 'bucket'
obj.meta.client.generate_presigned_url.return_value = '/download'
url = resource.get_download_url(obj)
assert url == '/download'
assert obj.meta.client.generate_presigned_url.called_once_with(
'get_object',
Params={'Key': 'key', 'Bucket': 'bucket'},
ExpiresIn=resource.URL_EXPIRY,
)
def test_download_url(self):
obj = mock.Mock()
obj.key = 'key'
obj.bucket = 'bucket'
obj.meta.client.generate_presigned_url.return_value = '/download'
url = resource.get_download_url(obj)
assert url == '/download'
assert obj.meta.client.generate_presigned_url.called_once_with(
'get_object',
Params={'Key': 'key', 'Bucket': 'bucket'},
ExpiresIn=resource.URL_EXPIRY,
)
