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