Exemplo n.º 1
0
    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
Exemplo n.º 2
0
    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()
Exemplo n.º 3
0
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()