def testForStudent(self):
"""Tests that a student profile cannot create a connection."""
# make the profile a student
self.profile.student_data = profile_utils.seedStudentData()
result = connection_logic.canCreateConnection(self.profile, self.org.key)
self.assertFalse(result)
self.assertEqual(
result.extra,
connection_logic._PROFILE_IS_STUDENT % self.profile.profile_id)
def testForStudent(self):
"""Tests that a student profile cannot create a connection."""
# make the profile a student
self.profile.student_data = profile_utils.seedStudentData()
result = connection_logic.canCreateConnection(self.profile,
self.org.key)
self.assertFalse(result)
self.assertEqual(
result.extra,
connection_logic._PROFILE_IS_STUDENT % self.profile.profile_id)
def createUser(self,
roles=None,
mentor_organizations=None,
admin_organizations=None,
winning_organization=None,
return_key=False,
email=None):
"""Creates a dummy user with a GCIProfile.
See ConversationHelper.createUser for full specification.
"""
program = db.get(ndb.Key.to_old_key(self.program_key))
roles = set(roles) if roles else set()
profile = profile_utils.seedNDBProfile(program.key())
winner_for = None
if profile is None:
raise Exception('profile is none')
if email:
profile.email = email
if mentor_organizations:
roles.update([MENTOR])
profile.mentor_for = mentor_organizations
if admin_organizations:
roles.update([ADMIN])
profile.admin_for = admin_organizations
if winning_organization:
roles.update([WINNER])
winner_for = winning_organization
if winner_for or WINNER in roles or STUDENT in roles:
profile.student_data = profile_utils.seedStudentData(
winner_for=winner_for)
profile.put()
return profile.key.parent() if return_key else profile.key.parent(
).get()
def createUser(
self, roles=None, mentor_organizations=None, admin_organizations=None,
winning_organization=None, return_key=False, email=None):
"""Creates a dummy user with a GCIProfile.
See ConversationHelper.createUser for full specification.
"""
program = db.get(ndb.Key.to_old_key(self.program_key))
roles = set(roles) if roles else set()
profile = profile_utils.seedNDBProfile(program.key())
winner_for = None
if profile is None:
raise Exception('profile is none')
if email:
profile.email = email
if mentor_organizations:
roles.update([MENTOR])
profile.mentor_for = mentor_organizations
if admin_organizations:
roles.update([ADMIN])
profile.admin_for = admin_organizations
if winning_organization:
roles.update([WINNER])
winner_for = winning_organization
if winner_for or WINNER in roles or STUDENT in roles:
profile.student_data = profile_utils.seedStudentData(
winner_for=winner_for)
profile.put()
return profile.key.parent() if return_key else profile.key.parent().get()
