def test_rackspace_uploader_creates_container(self, mock, mock2):
"""Test RACKSPACE UPLOADER creates container works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.get_container.side_effect = NoSuchContainer
mycf.create_container.return_value = True
mycf.make_container_public.return_value = True
u = RackspaceUploader()
res = u.init_app(self.flask_app)
err_msg = "Init app should return the container."
assert res is True, err_msg
def test_rackspace_uploader_upload_wrong_file(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload wrong file extension works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.upload_file.return_value = True
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.docs')
err_msg = "Upload file should return False"
res = u.upload_file(file, container='user_3')
assert res is False, err_msg
def test_rackspace_uploader_upload_file_fails(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file fail works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
from pyrax.exceptions import UploadFailed
mycf.upload_file.side_effect = UploadFailed
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return False"
assert u.upload_file(file, container='user_3') is False, err_msg
def test_rackspace_uploader_upload_wrong_file(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload wrong file extension works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.upload_file.return_value = True
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.docs')
err_msg = "Upload file should return False"
res = u.upload_file(file, container='user_3')
assert res is False, err_msg
def test_rackspace_uploader_creates_container(self, mock, mock2):
"""Test RACKSPACE UPLOADER creates container works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.get_container.side_effect = NoSuchContainer
mycf.create_container.return_value = True
mycf.make_container_public.return_value = True
u = RackspaceUploader()
res = u.init_app(self.flask_app)
err_msg = "Init app should return the container."
assert res is True, err_msg
def test_rackspace_uploader_upload_file_fails(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file fail works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
from pyrax.exceptions import UploadFailed
mycf.upload_file.side_effect = UploadFailed
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return False"
assert u.upload_file(file, container='user_3') is False, err_msg
def test_rackspace_uploader_delete(self, mock1):
"""Test RACKSPACE UPLOADER delete method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
calls = [call.get_container('container'),
call.get_container().get_object('file'),
call.get_container().get_object().delete()
]
u = RackspaceUploader()
u.init_app(self.flask_app)
err_msg = "It should return True"
assert u.delete_file('file', 'container') is True, err_msg
mycf.assert_has_calls(calls, any_order=True)
def test_file_exists_for_missing_file(self, credentials):
"""Test RACKSPACE UPLOADER file_exists returns False if the file does not exist"""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
u = RackspaceUploader()
u.init_app(self.flask_app)
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
file_exists = u.file_exists('noexist.txt', 'mycontainer')
mycf.get_container.assert_called_with('mycontainer')
assert file_exists is False
def test_rackspace_uploader_upload_file_object_fails(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file object fail works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
from pyrax.exceptions import NoSuchObject
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return True"
assert u.upload_file(file, container='user_3') is True, err_msg
def test_file_exists_for_missing_file(self, credentials):
"""Test RACKSPACE UPLOADER file_exists returns False if the file does not exist"""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
u = RackspaceUploader()
u.init_app(self.flask_app)
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
file_exists = u.file_exists('noexist.txt', 'mycontainer')
mycf.get_container.assert_called_with('mycontainer')
assert file_exists is False
def test_rackspace_uploader_delete(self, mock1):
"""Test RACKSPACE UPLOADER delete method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
calls = [call.get_container('container'),
call.get_container().get_object('file'),
call.get_container().get_object().delete()
]
u = RackspaceUploader()
u.init_app(self.flask_app)
err_msg = "It should return True"
assert u.delete_file('file', 'container') is True, err_msg
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_upload_file_object_fails(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file object fail works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
from pyrax.exceptions import NoSuchObject
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return True"
assert u.upload_file(file, container='user_3') is True, err_msg
def test_rackspace_uploader_upload_correct_file(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.upload_file.return_value=True
mycf.get_object.side_effect = NoSuchObject
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return True"
assert u.upload_file(file, container='user_3') is True, err_msg
calls = [call.get_container('user_3'),
call.get_container().get_object('test.jpg')]
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_delete_fails(self, mock1):
"""Test RACKSPACE UPLOADER delete fails method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
calls = [call.get_container('container')]
u = RackspaceUploader()
u.init_app(self.flask_app)
err_msg = "It should return False"
assert u.delete_file('file', 'container') is False, err_msg
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_delete_fails(self, mock1):
"""Test RACKSPACE UPLOADER delete fails method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
container = MagicMock()
container.get_object.side_effect = NoSuchObject
mycf.get_container.return_value = container
calls = [call.get_container('container')]
u = RackspaceUploader()
u.init_app(self.flask_app)
err_msg = "It should return False"
assert u.delete_file('file', 'container') is False, err_msg
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_upload_correct_file(self, mock, mock2):
"""Test RACKSPACE UPLOADER upload file works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.upload_file.return_value=True
mycf.get_object.side_effect = NoSuchObject
u = RackspaceUploader()
u.init_app(self.flask_app)
file = FileStorage(filename='test.jpg')
err_msg = "Upload file should return True"
assert u.upload_file(file, container='user_3') is True, err_msg
calls = [call.get_container('user_3'),
call.get_container().get_object('test.jpg')]
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_lookup_url_none(self, mock1):
"""Test RACKSPACE UPLOADER lookup returns None for non enabled CDN."""
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
err_msg = "We should get the None"
assert res is None, err_msg
def test_file_exists_for_real_file(self, credentials):
"""Test RACKSPACE UPLOADER file_exists returns True if the file exists"""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
u = RackspaceUploader()
u.init_app(self.flask_app)
filename='test.jpg'
container = MagicMock()
container.get_object.return_value = "Real File"
mycf.get_container.return_value = container
file_exists = u.file_exists(filename, 'mycontainer')
mycf.get_container.assert_called_with('mycontainer')
container.get_object.assert_called_with(filename)
assert file_exists is True
def test_file_exists_for_real_file(self, credentials):
"""Test RACKSPACE UPLOADER file_exists returns True if the file exists"""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
u = RackspaceUploader()
u.init_app(self.flask_app)
filename='test.jpg'
container = MagicMock()
container.get_object.return_value = "Real File"
mycf.get_container.return_value = container
file_exists = u.file_exists(filename, 'mycontainer')
mycf.get_container.assert_called_with('mycontainer')
container.get_object.assert_called_with(filename)
assert file_exists is True
def test_rackspace_uploader_lookup_url_returns_failover_url_project(self, mock):
"""Test RACKSPACE UPLOADER lookup returns failover_url for project avatar."""
filename = 'project_32.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
fake_container.make_public.side_effect = NoSuchObject
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
failover_url = 'https://localhost/static/img/placeholder.project.png'
err_msg = "We should get the %s but we got %s " % (failover_url, res)
assert res == failover_url, err_msg
def test_rackspace_uploader_get_container(self, mock1):
"""Test RACKSPACE UPLOADER get_container method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.side_effect = NoSuchContainer
calls = [call.get_container('user_3'),
call.create_container('user_3'),
call.make_container_public('user_3')
]
u = RackspaceUploader()
u.init_app(self.flask_app)
assert u.get_container('user_3')
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_get_container(self, mock1):
"""Test RACKSPACE UPLOADER get_container method works."""
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.side_effect = NoSuchContainer
calls = [call.get_container('user_3'),
call.create_container('user_3'),
call.make_container_public('user_3')
]
u = RackspaceUploader()
u.init_app(self.flask_app)
assert u.get_container('user_3')
mycf.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_lookup_url_returns_failover_url_project(self, mock):
"""Test RACKSPACE UPLOADER lookup returns failover_url for project avatar."""
filename = 'project_32.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
fake_container.make_public.side_effect = NoSuchObject
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
failover_url = 'https://localhost/static/img/placeholder.project.png'
err_msg = "We should get the %s but we got %s " % (failover_url, res)
assert res == failover_url, err_msg
def test_rackspace_uploader_lookup_url(self, mock1, mock2):
"""Test RACKSPACE UPLOADER lookup returns a valid link."""
uri = 'https://rackspace.com'
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=True)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
expected_url = "%s/%s" % (uri, filename)
err_msg = "We should get the following URL: %s" % expected_url
assert res == expected_url, err_msg
def test_rackspace_uploader_lookup_url(self, mock1, mock2):
"""Test RACKSPACE UPLOADER lookup returns a valid link."""
uri = 'https://rackspace.com'
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=True)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
expected_url = "%s/%s" % (uri, filename)
err_msg = "We should get the following URL: %s" % expected_url
assert res == expected_url, err_msg
def test_rackspace_uploader_lookup_url_returns_failover_url_project_backwards_compat(self, mock):
"""Test RACKSPACE UPLOADER lookup returns failover_url for project
avatar for old project avatars named with 'app'."""
filename = 'app_32.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
fake_container.make_public.side_effect = NoSuchObject
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
failover_url = '{}/static/img/placeholder.project.png'.format(self.flask_app.config['SERVER_NAME'])
err_msg = "We should get the %s but we got %s " % (failover_url, res)
assert res.endswith(failover_url), err_msg
def test_rackspace_uploader_lookup_url_none(self, mock1):
"""Test RACKSPACE UPLOADER lookup returns None for non enabled CDN."""
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {
'filename': filename,
'container': 'user_3'
})
err_msg = "We should get the None"
assert res is None, err_msg
def test_rackspace_uploader_lookup_url_enable_cdn(self, mock1, mock2):
"""Test RACKSPACE UPLOADER lookup enables CDN for non enabled CDN."""
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
url = 'https://rackspace.com/test.jpg'
err_msg = "We should get the %s but we got %s " % (url, res)
assert res == url, err_msg
calls = [call.make_public()]
fake_container.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_lookup_url_enable_cdn(self, mock1, mock2):
"""Test RACKSPACE UPLOADER lookup enables CDN for non enabled CDN."""
filename = 'test.jpg'
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
cdn_enabled_mock = PropertyMock(return_value=False)
type(fake_container).cdn_enabled = cdn_enabled_mock
mycf.get_container.return_value = fake_container
u = RackspaceUploader()
u.init_app(self.flask_app)
res = u._lookup_url('rackspace', {'filename': filename,
'container': 'user_3'})
url = 'https://rackspace.com/test.jpg'
err_msg = "We should get the %s but we got %s " % (url, res)
assert res == url, err_msg
calls = [call.make_public()]
fake_container.assert_has_calls(calls, any_order=True)
def test_rackspace_uploader_init(self, Mock):
"""Test RACKSPACE UPLOADER init works."""
new_extensions = ['pdf', 'doe']
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles',
return_value=cloudfiles_mock):
with patch.dict(self.flask_app.config,
{'ALLOWED_EXTENSIONS': new_extensions}):
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.get_container.return_value = True
u = RackspaceUploader()
res = u.init_app(self.flask_app, cont_name='mycontainer')
err_msg = "It should return the container."
assert res is True, err_msg
err_msg = "The container name should be updated."
assert u.cont_name == 'mycontainer', err_msg
for ext in new_extensions:
err_msg = "The .%s extension should be allowed" % ext
assert ext in u.allowed_extensions, err_msg
def test_rackspace_uploader_init(self, Mock):
"""Test RACKSPACE UPLOADER init works."""
new_extensions = ['pdf', 'doe']
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles',
return_value=cloudfiles_mock):
with patch.dict(self.flask_app.config,
{'ALLOWED_EXTENSIONS': new_extensions}):
with patch('pybossa.uploader.rackspace.pyrax.cloudfiles') as mycf:
mycf.get_container.return_value = True
u = RackspaceUploader()
res = u.init_app(self.flask_app, cont_name='mycontainer')
err_msg = "It should return the container."
assert res is True, err_msg
err_msg = "The container name should be updated."
assert u.cont_name == 'mycontainer', err_msg
for ext in new_extensions:
err_msg = "The .%s extension should be allowed" % ext
assert ext in u.allowed_extensions, err_msg
def setup_uploader(app):
global uploader
if app.config.get('UPLOAD_METHOD') == 'local':
from pybossa.uploader.local import LocalUploader
uploader = LocalUploader()
uploader.init_app(app)
if app.config.get('UPLOAD_METHOD') == 'rackspace': # pragma: no cover
from pybossa.uploader.rackspace import RackspaceUploader
uploader = RackspaceUploader()
app.url_build_error_handlers.append(uploader.external_url_handler)
uploader.init_app(app)
def setup_uploader(app):
"""Setup uploader."""
global uploader
upload_method = app.config.get('UPLOAD_METHOD')
if upload_method == 'local':
from pybossa.uploader.local import LocalUploader
uploader = LocalUploader()
if upload_method == 'rackspace': # pragma: no cover
from pybossa.uploader.rackspace import RackspaceUploader
uploader = RackspaceUploader()
app.url_build_error_handlers.append(uploader.external_url_handler)
if upload_method == 'cloud': # pragma: no cover
from pybossa.uploader.cloud_store import CloudStoreUploader
uploader = CloudStoreUploader()
app.url_build_error_handlers.append(uploader.external_url_handler)
if upload_method == 'cloudproxy': #pragma: no cover
from pybossa.uploader.cloud_proxy import CloudProxyUploader
uploader = CloudProxyUploader()
uploader.init_app(app)
