def test_update(self): user = User(nickname='kxxoling', openid='', google_account='*****@*****.**') user.register() user.update() assert User.query.filter_by(nickname='kxxoling')[0] is not None
def test_update_role(self): user = User(nickname='kxxoling', openid='', google_account='*****@*****.**') user.register() user.update_role(3) role2 = User.query.filter_by(nickname='kxxoling')[0].role assert role2 == 3
def test_delete(self): user = User(nickname='kxxoling', openid='', google_account='*****@*****.**') user.register() for user in User.query.all(): user.delete() assert User.query.count() == 0
def create_or_login(resp): user = User.query.filter_by(google_account=resp.email).first() if user is None: user = User(nickname=resp.nickname or resp.fullname, google_account=resp.email, openid=resp.identity_url) if user.register(): flash(u"You've successfully registered and signed in!") else: flash(u'Register failed!') return redirect(oid.get_next_url()) else: flash("You've successfully logged in!") session['user_id'] = user.id g.user = user return redirect(oid.get_next_url())