Example #1
0
 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
Example #2
0
 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
Example #3
0
 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
Example #4
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())
Example #5
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())
Example #6
0
 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
Example #7
0
 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
Example #8
0
 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