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')