def signup():
error = None
next = request.args.get('next')
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
if not User.get(username):
u = User(username, password)
u.save()
return redirect('/')
else:
error = "The user already exists"
return render_template('signup.html', login=True, next=next, error=error)
def test_create_user(self):
"""Should create a user with the correct data"""
expected_user_data = {
'username': USER['username'],
'password': crypt(USER['password'], SECRET_KEY),
'enabled': "True"
}
self.assertEqual(User.get(USER['username']), expected_user_data)
def login():
error = None
next = request.args.get('next')
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
remember = request.form.get('remember', False)
if User.check_user_password(username, password):
# Right password
user = AuthUser(username)
if login_user(user, remember=remember):
# Logged in correctly
return redirect('/')
error = "Username or password incorrect"
return render_template('login.html', login=True, next=next, error=error)
def test_delete_user(self):
"""Should remove a user when a username is given"""
User.remove(USER['username'])
self.assertEquals(User.get(USER['username']), {})
def test_check_password(self):
"""Should check that the user credentials are valid"""
self.assertTrue(User.check_user_password(USER['username'],
USER['password']))
def setUp(self):
super(UserTestCase, self).setUp()
u = User(USER['username'], USER['password'])
u.save()
def load_user(user_id):
user = User.get(user_id)
if user:
return AuthUser(user['username'])
else:
return None
