def test_view_creates_guid(self):
guid_fid = 'unique'
guid_content = 'snowflake'
self._upload_file(guid_fid, guid_content)
node_file = NodeFile.load(self.project.files_current[guid_fid])
guid_count = OsfGuidFile.find().count()
# View file for the first time
url = node_file.url(self.project)
res = self.app.get(
url,
auth=self.user.auth,
).follow(
auth=self.user.auth,
)
guid = OsfGuidFile.find_one(
Q('node', 'eq', self.project) &
Q('name', 'eq', guid_fid)
)
# GUID count has been incremented by one
assert_equal(
OsfGuidFile.find().count(),
guid_count + 1
)
# Client has been redirected to GUID
assert_equal(
res.request.path.strip('/'),
guid._id,
)
# View file for the second time
self.app.get(
url,
auth=self.user.auth,
).follow(
auth=self.user.auth,
)
# GUID count has not been incremented
assert_equal(
OsfGuidFile.find().count(),
guid_count + 1
)
def setUp(self):
super(TestMigrateGuids, self).setUp()
self.clear_guids()
self.project = ProjectFactory()
self.paths = ['peppers', 'sausage', 'pepperoni']
self.legacy_objs = [
OsfGuidFile(node=self.project, name=path) for path in self.paths
]
for obj in self.legacy_objs:
obj.save()
def test_view_creates_guid(self):
guid_fid = 'unique'
guid_content = 'snowflake'
self._upload_file(guid_fid, guid_content)
node_file = NodeFile.load(self.project.files_current[guid_fid])
guid_count = OsfGuidFile.find().count()
# View file for the first time
url = node_file.url(self.project)
res = self.app.get(
url,
auth=self.user.auth,
).follow(auth=self.user.auth, )
guid = OsfGuidFile.find_one(
Q('node', 'eq', self.project) & Q('name', 'eq', guid_fid))
# GUID count has been incremented by one
assert_equal(OsfGuidFile.find().count(), guid_count + 1)
# Client has been redirected to GUID
assert_equal(
res.request.path.strip('/'),
guid._id,
)
# View file for the second time
self.app.get(
url,
auth=self.user.auth,
).follow(auth=self.user.auth, )
# GUID count has not been incremented
assert_equal(OsfGuidFile.find().count(), guid_count + 1)
def clear_guids(self):
OsfGuidFile.remove()
OsfStorageGuidFile.remove()
def find_legacy_objs():
return OsfGuidFile.find()
def view_file(auth, **kwargs):
node_settings = kwargs['node_addon']
node = kwargs['node'] or kwargs['project']
file_name = kwargs['fid']
file_name_clean = file_name.replace('.', '_')
try:
guid = OsfGuidFile.find_one(
Q('node', 'eq', node) &
Q('name', 'eq', file_name)
)
except:
guid = OsfGuidFile(
node=node,
name=file_name,
)
guid.save()
redirect_url = check_file_guid(guid)
if redirect_url:
return redirect(redirect_url)
# Throw 404 and log error if file not found in files_versions
try:
file_id = node.files_versions[file_name_clean][-1]
except KeyError:
logger.error('File {} not found in files_versions of component {}.'.format(
file_name_clean, node._id
))
raise HTTPError(http.NOT_FOUND)
file_object = NodeFile.load(file_id)
# Ensure NodeFile is attached to Node; should be fixed by actions or
# improved data modeling in future
if not file_object.node:
file_object.node = node
file_object.save()
download_url = file_object.download_url(node)
render_url = file_object.render_url(node)
info_url = file_object.info_url(node)
file_path = os.path.join(
settings.UPLOADS_PATH,
node._primary_key,
file_name
)
# Throw 404 and log error if file not found on disk
if not os.path.isfile(file_path):
logger.error('File {} not found on disk.'.format(file_path))
raise HTTPError(http.NOT_FOUND)
_, file_ext = os.path.splitext(file_path.lower())
# Get or create rendered file
cache_file = get_cache_file(
file_object.filename,
file_object.latest_version_number(node)
)
rendered = get_cache_content(
node_settings, cache_file, start_render=True, file_path=file_path,
file_content=None, download_path=download_url,
)
ret = {
'file_name': file_name,
'render_url': render_url,
'rendered': rendered,
'info_url': info_url,
}
ret.update(_view_project(node, auth))
return ret
def clear_guids(self):
OsfGuidFile.remove()
OsfStorageGuidFile.remove()
def find_legacy_objs():
return OsfGuidFile.find()
def view_file(auth, **kwargs):
node_settings = kwargs['node_addon']
node = kwargs['node'] or kwargs['project']
file_name = kwargs['fid']
file_name_clean = file_name.replace('.', '_')
try:
guid = OsfGuidFile.find_one(
Q('node', 'eq', node) &
Q('name', 'eq', file_name)
)
except:
guid = OsfGuidFile(
node=node,
name=file_name,
)
guid.save()
redirect_url = check_file_guid(guid)
if redirect_url:
return redirect(redirect_url)
# Throw 404 and log error if file not found in files_versions
try:
file_id = node.files_versions[file_name_clean][-1]
except KeyError:
logger.error('File {} not found in files_versions of component {}.'.format(
file_name_clean, node._id
))
raise HTTPError(http.NOT_FOUND)
file_object = NodeFile.load(file_id)
# Ensure NodeFile is attached to Node; should be fixed by actions or
# improved data modeling in future
if not file_object.node:
file_object.node = node
file_object.save()
download_url = file_object.download_url(node)
render_url = file_object.render_url(node)
info_url = file_object.info_url(node)
file_path = os.path.join(
settings.UPLOADS_PATH,
node._primary_key,
file_name
)
# Throw 404 and log error if file not found on disk
if not os.path.isfile(file_path):
logger.error('File {} not found on disk.'.format(file_path))
raise HTTPError(http.NOT_FOUND)
_, file_ext = os.path.splitext(file_path.lower())
# Get or create rendered file
cache_file = get_cache_file(
file_object.filename,
file_object.latest_version_number(node)
)
rendered = get_cache_content(
node_settings, cache_file, start_render=True, file_path=file_path,
file_content=None, download_path=download_url,
)
rv = {
'file_name': file_name,
'render_url': render_url,
'rendered': rendered,
'info_url': info_url,
}
rv.update(_view_project(node, auth))
return rv
