def create_and_log_in_user(testapp=None,
department=None,
rolename=None,
username="******"):
# set up a user
user = User.create(username=username,
email="{}@example.com".format(username),
password="******")
# associate a department if a department is passed
if department:
user.departments.append(department)
user.active = True
user.save()
# associate a role if a name is passed
if rolename:
user_role = Role(name=rolename)
user_role.save()
user.roles.append(user_role)
# login
response = testapp.get("/login/")
form = response.forms['loginForm']
form['username'] = user.username
form['password'] = '******'
response = form.submit().follow()
return user
def test_role(self):
role = Role(name='admin')
role.save()
u = UserFactory()
u.roles.append(role)
u.save()
assert role in u.roles
def make_admin_user():
password = prompt_pass("Password")
user = User.create(username="******", email="*****@*****.**", password=password, active=True)
admin_role = Role(name='admin')
admin_role.save()
user.roles.append(admin_role)
user.save()
def test_role(self):
role = Role(name='admin')
role.save()
u = UserFactory()
u.roles.append(role)
u.save()
assert role in u.roles
def test_is_admin(self):
admin_role = Role(name='admin')
admin_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.save()
assert u.is_admin()
def test_is_admin(self):
admin_role = Role(name='admin')
admin_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.save()
assert u.is_admin()
def make_admin_user():
password = prompt_pass("Password")
user = User.create(username="******",
email="*****@*****.**",
password=password,
active=True)
admin_role = Role(name='admin')
admin_role.save()
user.roles.append(admin_role)
user.save()
def test_many_roles(self):
admin_role = Role(name='admin')
admin_role.save()
other_role = Role(name='other')
other_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.roles.append(other_role)
u.save()
assert admin_role in u.roles
assert other_role in u.roles
def test_many_roles(self):
admin_role = Role(name='admin')
admin_role.save()
other_role = Role(name='other')
other_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.roles.append(other_role)
u.save()
assert admin_role in u.roles
assert other_role in u.roles
def test_many_users(self):
admin_role = Role(name='admin')
admin_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.save()
admin_role_two = Role(name='admin')
admin_role_two.save()
u_two = UserFactory()
u_two.roles.append(admin_role_two)
u_two.save()
assert admin_role in u.roles
assert admin_role_two in u_two.roles
def test_many_users(self):
admin_role = Role(name='admin')
admin_role.save()
u = UserFactory()
u.roles.append(admin_role)
u.save()
admin_role_two = Role(name='admin')
admin_role_two.save()
u_two = UserFactory()
u_two.roles.append(admin_role_two)
u_two.save()
assert admin_role in u.roles
assert admin_role_two in u_two.roles
def create_and_log_in_user(testapp=None, department=None, rolename=None, username="******"):
# set up a user
user = User.create(username=username, email="{}@example.com".format(username), password="******")
# associate a department if a department is passed
if department:
user.departments.append(department)
user.active = True
user.save()
# associate a role if a name is passed
if rolename:
user_role = Role(name=rolename)
user_role.save()
user.roles.append(user_role)
# login
response = testapp.get("/login/")
form = response.forms['loginForm']
form['username'] = user.username
form['password'] = '******'
response = form.submit().follow()
return user