def __register_user(self, google_user, permission=UserPermission.normal):
"""
Register the user.
:param google_user: The google.appengine.api.users.get_current_user().
:param permission: The user's permission.
:return: The datastore.user_model.UserModel object.
"""
user = User()
user.email = google_user.email().lower()
user.name = google_user.nickname()
user.permission = permission
user.put()
user.get(user.key())
return user
def test_user_model_with_data(self):
fake_key = MagicMock()
fake_key.id.return_value = 100
user = User()
user.name = 'name'
user.email = '*****@*****.**'
user.key = MagicMock(return_value=fake_key)
model = UserModel(user)
self.assertEqual(model.id, 100)
self.assertEqual(model.permission, UserPermission.normal)
self.assertEqual(model.name, 'name')
self.assertEqual(model.email, '*****@*****.**')
self.assertEqual(model.logout_url, 'logout_url')
