class TestUser(unittest.TestCase):
def setUp(self):
self.user = User(
'testuser',
'password'
)
self.user.set_new_user()
def tearDown(self):
self.db = Database()
self.db.make_query(
'''
DELETE FROM user WHERE username = "******"
'''.format('testuser')
)
def test_check_username(self):
self.assertTrue(
self.user.check_username()
)
def test_check_password(self):
self.assertTrue(
self.user.check_password()
)
def login_post():
form = LoginForm()
if form.validate_on_submit():
user = User(form.username.data)
password = form.password.data
if user is None or user.check_password(password) == False:
flash('Invalid username or password')
return redirect(url_for('login_get'))
login_user(user)
return redirect(url_for('message_get'))
else:
flash(parse_login_error(form.errors))
return redirect(url_for('login_get'))
