def test_remove_permission(test_client, test_db):
r = Role(name="Test Role", permissions=0)
r.add_permission(Permission.SUBMIT_SURVEY)
r.add_permission(Permission.READ_SURVEY)
assert r.has_permission(Permission.SUBMIT_SURVEY)
r.remove_permission(Permission.SUBMIT_SURVEY)
assert not r.has_permission(Permission.SUBMIT_SURVEY)
def test_tur001_perm_funcs(self):
r = Role(name='User')
# add two perms
r.add_permission(Permission.FOLLOW)
r.add_permission(Permission.REVIEW)
# assert has one of those
assert r.has_permission(Permission.FOLLOW)
# assert it doesn't have another
assert not r.has_permission(Permission.PUBLISH)
r.remove_permission(Permission.FOLLOW)
# assert now it doesn't have removed one
assert not r.has_permission(Permission.FOLLOW)
# erase all perms
r.reset_permissions()
assert not r.has_permission(Permission.REVIEW)
def test_has_permission(self):
"""测试三个关于权限的方法是否正确"""
r = Role(name='User')
r.add_permission(Permissions.FOLLOW)
r.add_permission(Permissions.ADMIN)
r.add_permission(Permissions.MODERATE)
self.assertTrue(r.has_permission(Permissions.FOLLOW))
self.assertTrue(r.has_permission(Permissions.ADMIN))
self.assertTrue(r.has_permission(Permissions.MODERATE))
self.assertFalse(r.has_permission(Permissions.COMMIT))
self.assertFalse(r.has_permission(Permissions.WRITE))
r.remove_permission(Permissions.ADMIN)
self.assertFalse(r.has_permission(Permissions.ADMIN))
r.reset_permission()
self.assertFalse(r.has_permission(Permissions.FOLLOW))
self.assertFalse(r.has_permission(Permissions.MODERATE))
def test_reset_permission_sets_permission_to_zero(test_client, test_db):
r = Role(name="Test Role", permissions=0)
r.add_permission(Permission.SUBMIT_SURVEY)
assert not r.permissions == 0
r.reset_permissions()
assert r.permissions == 0
