def test_ensure_user_info(self):
"""Test creation logic for the a models.UserInfo data model."""
user_1 = FakeUser("*****@*****.**")
user_2 = FakeUser("*****@*****.**")
user_info_1 = models.UserInfo()
user_info_1.email = user_1.email()
user_info_1.safe_email = util.get_safe_email(user_1)
user_info_1.is_reviewer = True
user_info_1.put()
ret_info_1 = account_facade.ensure_user_info(user_1)
ret_info_2 = account_facade.ensure_user_info(user_2)
self.assertEqual(ret_info_1.email, user_1.email())
self.assertTrue(ret_info_1.is_reviewer)
self.assertEqual(ret_info_2.email, user_2.email())
self.assertFalse(ret_info_2.is_reviewer)
def get(self):
"""
GET request handler that ensures a user has a models.UserInfo record.
GET request handler that checks that a user has a models.UserInfo
record, creating it if it does not already exit. All users are then
redirected to the page specified by util.get_user_home.
"""
cur_user = users.get_current_user()
target_email = cur_user.email()
if not util.check_email(target_email):
account_facade.set_flash_message(
target_email,
constants.FLASH_MSG_TYPE_ERR,
constants.FLASH_MSG_INVALID_EMAIL
)
self.redirect(constants.HOME_URL)
else:
account_facade.ensure_user_info(cur_user)
self.redirect(util.get_user_home(cur_user))
