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