def test_user_role(self): u = User(email='*****@*****.**', password='******') self.assertTrue(u.can(Permission.FOLLOW)) self.assertTrue(u.can(Permission.COMMENT)) self.assertTrue(u.can(Permission.WRITE_ARTICLES)) self.assertFalse(u.can(Permission.MODERATE_COMMENTS)) self.assertFalse(u.can(Permission.ADMINISTER))
def test_administrator_role(self): r = Role.query.filter_by(name='Administrator').first() u = User(email='*****@*****.**', password='******', role=r) self.assertTrue(u.can(Permission.FOLLOW)) self.assertTrue(u.can(Permission.COMMENT)) self.assertTrue(u.can(Permission.WRITE_ARTICLES)) self.assertTrue(u.can(Permission.MODERATE_COMMENTS)) self.assertTrue(u.can(Permission.ADMINISTER))