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 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_delete_user(self):
"""Should remove a user when a username is given"""
User.remove(USER['username'])
self.assertEquals(User.get(USER['username']), {})
def load_user(user_id):
user = User.get(user_id)
if user:
return AuthUser(user['username'])
else:
return None
