class MyTestCase(unittest.TestCase): def setUp(self): self.user = User("test_account", "*****@*****.**", "password1234") self.user.addUser() def test_login(self): assert(self.user.findUser() == True) def tearDown(self): self.user.deleteUser("test_account") self.user.db_close()
class MyTestCase(unittest.TestCase): #user = 0 def setUp(self): self.user = User("test", "test", "test") def test_connection(self): res = self.user.addUser() self.assertTrue(res) def tearDown(self): self.user.db_close()
def registerPage(): if (current_user.is_authenticated): return redirect("/") name = request.form.get('name') email = request.form.get('email') password = request.form.get('password') if (email is not None): if (not re.match(r'[^@]+@[^@]+\.[^@]+', email)): flash("Email is not in the right format!") return render_template("register.html") user = User() if (user.checkIfEmailExists(email)): flash("Email already used! Try another one!") return render_template("register.html") else: user.addUser(name, email, password) login_user(user.getUserByEmail(email)) return redirect("/") return render_template("register.html")