def test_generate_token(self):
user = User(
username='******',
email='*****@*****.**',
password='******',
role=1,
status=1
)
db.session.add(user)
db.session.commit()
with app.test_request_context():
token = user.generate_token()
self.assertTrue(user.verify_token(token) == user)
def test_verify_token(self):
user = User(
username='******',
email='*****@*****.**',
password='******',
role=1,
status=1
)
db.session.add(user)
db.session.commit()
usertest = User.query.filter(User.username == 'Testuser').first()
with app.test_request_context():
token = user.generate_token(expiration=10)
self.assertIs(user.verify_token(token), user)
self.assertIsNot(user, usertest.verify_token(token))
def signup():
form = SignupForm()
if ('token' in session) and (User.verify_token(session['token'])):
return redirect(url_for('auth.profile'))
if request.method == 'POST':
if form.validate() is False:
return render_template("authentication/signup.html", form=form)
else:
new_user = User(form.username.data,
form.email.data,
form.password.data,
READ_ROLE + COMMENT_ROLE + WRITE_ROLE, 1)
db.session.add(new_user)
db.session.commit()
session['user_id'] = new_user.id
session['token'] = new_user.generate_token()
session['email'] = new_user.email
session['user_name'] = new_user.username
return redirect(url_for('auth.profile'))
elif request.method == 'GET':
return render_template("authentication/signup.html", form=form)
