def test_role(self):
role = Role(name='admin')
role.save()
u = UserFactory()
u.roles.append(role)
u.save()
assert role in u.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
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 from_department_and_password(department, password):
extractor = Extractor.create(username='******' % department.name.replace (" ", "_"), email='*****@*****.**', department_id=department.id, password=password)
extractor.roles.append(Role.create(name="extractor"))
extractor.save()
envs = extractor.generate_envs(password)
return (extractor,envs)
def test_access(self, user, testapp):
admin_role = Role.create(name="admin")
user.roles.append(admin_role)
user.save()
TestLoggingIn.test_can_log_in_returns_200(self, user=user, testapp=testapp)
res = testapp.get("/admin").follow()
assert res.status_code == 200
assert "Admin Dashboard" in res
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 from_department_and_password(department, password):
extractor = Extractor.create(username='******' % department.name.replace(
" ", "_"), email='*****@*****.**', password=password)
extractor.departments.append(department)
extractor.roles.append(Role.create(name="extractor"))
extractor.save()
envs = extractor.generate_envs(password)
return (extractor, envs)
def test_access(self, user, testapp):
admin_role = Role.create(name="admin")
user.roles.append(admin_role)
user.save()
TestLoggingIn.test_can_log_in_returns_200(self,
user=user,
testapp=testapp)
res = testapp.get("/admin").follow()
assert res.status_code == 200
assert "Admin Dashboard" in res
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_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_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
