def test_has_role_user_with_role(self): role_ids = TestUserRolesDatabase.add_roles_to_db(3) user = User('username', 'password') user.set_roles(role_ids) db.session.commit() for role in role_ids: self.assertTrue(user.has_role(role))
def test_has_role_user_without_role(self): role_ids = TestUserRolesDatabase.add_roles_to_db(3) user = User('username', 'password') user.set_roles(role_ids) self.assertFalse(user.has_role('invalid'))
def test_has_role_user_with_no_roles(self): user = User('username', 'password') self.assertFalse(user.has_role(100))