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))
