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, )