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