def test_user_save(self):
user = User()
user.username = '******'
user.set_password('password')
db.session.add(user)
db.session.commit()
queried_user = User.query.filter_by(username="******").one()
assert queried_user is not None
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User()
user.username = form.username.data
user.password = form.username.data
db.session.add(user)
db.session.commit()
flash("User has been created please login!!", category="")
return redirect(url_for(".login"))
return render_template("register.html", form=form)
def test_login(self):
test_user = User()
test_user.username = "******"
test_user.set_password("password")
db.session.add(test_user)
db.session.commit()
result = self.client.post('/login',
data=dict(username='******',
password="******"),
follow_redirects=True)
assert b'Logged in successfully.' in result.get_data()
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User()
print(form.username.data)
user.username = form.username.data
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash("Your user has be created please login",category='success')
return redirect(request.args.get("next") or url_for("main.home"))
return render_template("register.html",form=form)
def test_role_relationship(self):
admin = Role()
admin.name = 'admin'
admin.description = 'admin'
admin_user = User()
admin_user.username = '******'
admin_user.set_password('password')
admin_user.roles.append(admin)
db.session.add(admin_user)
db.session.add(admin)
db.session.commit()
queried_user = Role.query.filter_by(name="admin").one().users.one()
assert queried_user == admin_user
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User()
user.username = form.username.data
user.set_password(form.password.data)
user.active = True
default_role = Role.query.filter_by(name="default").first()
user.roles.append(default_role)
db.session.add(user)
db.session.commit()
flash("Your user has been created, please login.", category="success")
return redirect(url_for(".login"))
return render_template("register.html", form=form)
def test_admin_logged_in(self):
test_user = User()
test_user.username = "******"
test_user.set_password("password")
admin = Role()
admin.name = 'admin'
test_user.roles.append(admin)
db.session.add(test_user)
db.session.commit()
result = self.client.post('/login',
data=dict(username='******',
password="******"),
follow_redirects=True)
result = self.client.get('/admin/user/')
assert result.status_code == 200
def test_user_password(self):
user = User()
user.username = '******'
user.set_password('password')
db.session.add(user)
db.session.commit()
assert user.check_password('password')
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User()
user.username = form.username.data
user.set_password(form.password.data)
user.active = True
default_role = Role.query.filter_by(name="default").first()
user.roles.append(default_role)
db.session.add(user)
db.session.commit()
flash("Your user has been created, please login.", category="success")
return redirect(url_for(".login"))
return render_template("register.html", form=form)