def testDownloadCollection(self):
self.client.login(username=self.user)
pk1 = self.Collection1.pk
request = self.factory.get("/collections/%s/download" % pk1)
request.user = self.user
response = download_collection(request, str(pk1))
self.assertTrue(len(response.getvalue())) # If there is something in the response, the file was generated.
self.assertEqual(response.status_code, 200)
def testDownloadCollection(self):
self.client.login(username=self.user)
pk1 = self.Collection1.pk
request = self.factory.get('/collections/%s/download' % pk1)
request.user = self.user
response = download_collection(request, str(pk1))
self.assertTrue(len(response.getvalue(
))) # If there is something in the response, the file was generated.
self.assertEqual(response.status_code, 200)
def testDownloadCollection(self):
self.client.login(username=self.user)
pk1 = self.Collection1.pk
request = self.factory.get('/collections/%s/download' %pk1)
request.user = self.user
response = download_collection(request, str(pk1))
self.assertTrue(response.streaming_content)
self.assertEqual(response.status_code, 200)
zf = zipfile.ZipFile(io.BytesIO(''.join(response.streaming_content)))
self.assertEqual(len(zf.filelist), 1) # 1 Atlas
self.assertIsNone(zf.testzip())
self.assertIn("Collection1/VentralFrontal_thr75_summaryimage_2mm.nii.gz", zf.namelist())
def testDownloadCollection(self):
self.client.login(username=self.user)
pk1 = self.Collection1.pk
request = self.factory.get('/collections/%s/download' %pk1)
request.user = self.user
response = download_collection(request, str(pk1))
self.assertTrue(response.streaming_content)
self.assertEqual(response.status_code, 200)
zf = zipfile.ZipFile(io.BytesIO(''.join(response.streaming_content)))
self.assertEqual(len(zf.filelist), 1) # 1 Atlas
self.assertIsNone(zf.testzip())
self.assertIn("Collection1/VentralFrontal_thr75_summaryimage_2mm.nii.gz", zf.namelist())
def testDownloadCollection_NIDM_results(self):
collection = Collection(owner=self.user, name="Collection2")
collection.save()
# Upload NIMDResult zip file
zip_file = open(
os.path.join(TEST_PATH, 'test_data/nidm/auditory.nidm.zip'), 'rb')
post_dict = {
'name': 'auditory',
'description': '{0} upload test'.format('spm_auditory_v1.2.0'),
'collection': collection.pk,
}
fname = os.path.basename(
os.path.join(TEST_PATH, 'test_data/nidm/auditory.nidm.zip'))
file_dict = {'zip_file': SimpleUploadedFile(fname, zip_file.read())}
form = NIDMResultsForm(post_dict, file_dict)
form.save()
# Upload Statistic Map
image = save_statmap_form(image_path=os.path.join(
TEST_PATH, 'test_data/statmaps/all.nii.gz'),
collection=collection,
image_name="all.nii.gz")
factory = RequestFactory()
self.client.login(username=self.user)
request = factory.get('/collections/%s/download' % collection.pk,
{'format': 'img.zip'})
request.user = self.user
response = download_collection(request, str(collection.pk))
self.assertTrue(response.streaming_content)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.get('Content-Disposition'),
"attachment; filename=" + collection.name + ".zip")
zf = zipfile.ZipFile(io.BytesIO(''.join(response.streaming_content)))
self.assertEqual(len(zf.filelist), 2) # 1 NIDMResult, 1 Statmap
self.assertIsNone(zf.testzip())
self.assertIn("Collection2/all.nii.gz", zf.namelist())
self.assertIn("Collection2/auditory.nidm.zip", zf.namelist())
def testDownloadCollection_NIDM_results(self):
collection = Collection(owner=self.user, name="Collection2")
collection.save()
# Upload NIMDResult zip file
zip_file = open(os.path.join(TEST_PATH, 'test_data/nidm/auditory.nidm.zip'), 'rb')
post_dict = {
'name': 'auditory',
'description': '{0} upload test'.format('spm_auditory_v1.2.0'),
'collection': collection.pk,
}
fname = os.path.basename(os.path.join(TEST_PATH, 'test_data/nidm/auditory.nidm.zip'))
file_dict = {
'zip_file': SimpleUploadedFile(fname, zip_file.read())}
form = NIDMResultsForm(post_dict, file_dict)
form.save()
# Upload Statistic Map
image = save_statmap_form(image_path=os.path.join(TEST_PATH, 'test_data/statmaps/all.nii.gz'),
collection=collection,
image_name="all.nii.gz")
factory = RequestFactory()
self.client.login(username=self.user)
request = factory.get('/collections/%s/download' % collection.pk, {'format': 'img.zip'})
request.user = self.user
response = download_collection(request, str(collection.pk))
self.assertTrue(response.streaming_content)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.get('Content-Disposition'), "attachment; filename=" + collection.name + ".zip")
zf = zipfile.ZipFile(io.BytesIO(''.join(response.streaming_content)))
self.assertEqual(len(zf.filelist),2) # 1 NIDMResult, 1 Statmap
self.assertIsNone(zf.testzip())
self.assertIn("Collection2/all.nii.gz", zf.namelist())
self.assertIn("Collection2/auditory.nidm.zip", zf.namelist())