def test_save(self):
project = Project()
project.name = 'abc'
project.creator = self.ionadmin
project.save()
self.assertIsNotNone(project.id, 'Project id is None')
return project
def bulk_get_or_create_create_multi_projects(self):
name = [' a ', 'b', '', None, ' ', ' a b ']
projects = Project.bulk_get_or_create(name)
self.assertEqual(len(projects), 3, 'Incorrect number of projects created')
Project.objects.get(name='a')
Project.objects.get(name='b')
Project.objects.get(name='a_b')
return projects
def bulk_get_or_create_create_multi_projects(self):
name = [' a ', 'b', '', None, ' ', ' a b ']
projects = Project.bulk_get_or_create(name)
self.assertEqual(len(projects), 3, 'Incorrect number of projects created')
Project.objects.get(name='a')
Project.objects.get(name='b')
Project.objects.get(name='a_b')
return projects
def getProjectObjectList(self, step_helper, username):
retval = []
if step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.PROJECTS]:
for project_id in step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.PROJECTS]:
retval.append(Project.objects.get(id=int(project_id)))
newProjectNames = step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.NEW_PROJECTS]
if newProjectNames:
newProjectNames = newProjectNames.split(',')
retval.extend(Project.bulk_get_or_create(newProjectNames, User.objects.get(username=username)))
return retval
def getProjectObjectList(self, step_helper, username):
retval = []
if step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.PROJECTS]:
for project_id in step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.PROJECTS]:
retval.append(Project.objects.get(id=int(project_id)))
newProjectNames = step_helper.steps[StepNames.OUTPUT].savedFields[OutputFieldNames.NEW_PROJECTS]
if newProjectNames:
newProjectNames = newProjectNames.split(',')
retval.extend(Project.bulk_get_or_create(newProjectNames, User.objects.get(username=username)))
return retval
def get_projects(username, json_data):
# selected projects, projectIdAndNameList is a string if 1 entry; list otherwise
projectIdAndNameList = json_data.get('projects', '')
projectObjList = []
if projectIdAndNameList:
projectObjList, missings = get_projects_helper(projectIdAndNameList)
for missing in missings:
logger.debug("views.editplannedexperiment project= %s is no longer in db" % missing)
# new projects added
newProjectNames = json_data.get('newProjects', '')
if newProjectNames:
newProjectNames = newProjectNames.split(',')
projectObjList.extend(Project.bulk_get_or_create(newProjectNames, User.objects.get(username=username)))
return projectObjList
def get_projects(username, json_data):
# selected projects, projectIdAndNameList is a string if 1 entry; list otherwise
projectIdAndNameList = json_data.get('projects', '')
projectObjList = []
if projectIdAndNameList:
projectObjList, missings = get_projects_helper(projectIdAndNameList)
for missing in missings:
logger.debug("views.editplannedexperiment project= %s is no longer in db" % missing)
# new projects added
newProjectNames = json_data.get('newProjects', '')
if newProjectNames:
newProjectNames = newProjectNames.split(',')
projectObjList.extend(Project.bulk_get_or_create(newProjectNames, User.objects.get(username=username)))
return projectObjList
def test_save(self):
project = Project()
project.name = 'abc'
project.creator = self.ionadmin
project.save()
self.assertIsNotNone(project.id, 'Project id is None')
return project
def _test_bulk_get_or_create_create_single_without_user(self, name='a'):
projects = Project.bulk_get_or_create(name)
self.assertIsNotNone(projects[0].id, 'Project id is None')
user = User.objects.order_by('pk')[0]
self.assertEqual(projects[0].creator, user, 'Default user not same')
print 'created %s ' % projects[0].id
def _test_bulk_get_or_create_create_single_without_user(self, name = 'a'):
projects = Project.bulk_get_or_create(name)
self.assertIsNotNone(projects[0].id, 'Project id is None')
user = User.objects.order_by('pk')[0]
self.assertEqual(projects[0].creator, user, 'Default user not same')
print 'created %s ' % projects[0].id
