def test_register_duplicate_user_is_rejected(self): user = '******' password = '******' self.assertTrue(Password.register_user(user, password, self.db)) self.assertFalse(Password.register_user(user, password, self.db)) result = self.db.user.count({'username': user}) self.assertEqual(result, 1)
def test_register_valid_user(self): user = '******' password = '******' Password.register_user(user, password, self.db) result = self.db.user.find_one({'username': user}) self.assertIsNotNone(result) self.assertEqual(result[u'username'], user) self.assertIsNotNone(result[u'password'])
def register(): error = None if request.method == 'POST': collection = db.user.find({'username': request.form['username']}) if len(request.form['password']) <= 0: error = 'Password cannot be empty' elif collection.retrieved > 0: error = 'Username taken' else: Password.register_user(request.form['username'], request.form['password'], db) session['session_id'] = user_session.create_session(request.form['username']) return redirect(url_for('show_entries')) return render_template('register.html', error=error)
def login(): error = None if request.method == 'POST': app.logger.info('POST /login') result = Password.check_password(request.form['username'], request.form['password'], db=db) if result: session['session_id'] = user_session.create_session(request.form['username']) flash('You were logged in') return redirect(url_for('show_entries')) error = 'Invalid login' app.logger.info('GET /login') return render_template('login.html', error=error)
def test_check_password_valid_accepted_unicode(self): user = u'Max' password = u'1234567890asdfghjkl' Password.register_user(user, password, self.db) result = Password.check_password(user, password, self.db) self.assertTrue(result)
def test_check_password_invalid_rejected_unicode(self): user = u'Max' password = u'1234567890asdfghjkl' Password.register_user(user, password, self.db) result = Password.check_password(user, password[:-1], self.db) self.assertFalse(result)
def test_register_empty_username_or_password_rejected(self): self.assertFalse(Password.register_user('', '1234', self.db)) self.assertFalse(Password.register_user('Max', '', self.db)) self.assertEqual(self.db.user.count({'username': '******'}), 0)
def test_register_and_check_password(self): user = '******' password = '******' Password.register_user(user, password, self.db) result = Password.check_password(user, password, self.db) self.assertTrue(result)
def test_register_with_none(self): self.assertFalse(Password.register_user(None, 'qwerty', self.db)) self.assertFalse(Password.register_user('ralph', None, self.db)) self.assertFalse(Password.register_user('ralph', 'qwerty', None))