def test_remove_permission(self): from olp.utils import remove_all_permissions apple = Apple(name="test") apple.save() self.user.assign_perm("tests.can_be_awesome", apple) remove_all_permissions(self.user) self.assertEqual(ObjectPermission.objects.count(), 0)
def test_permission_removed_only_from_specified_object(self): from olp.utils import remove_all_permissions from django.contrib.auth.models import User self.other_user = User.objects.create_user( "other_test", "*****@*****.**", "other_test" ) self.other_user.save() apple = Apple(name="test") apple.save() self.user.assign_perm("tests.can_be_awesome", apple) self.other_user.assign_perm("tests.can_be_awesome", apple) remove_all_permissions(self.user) self.assertEqual(ObjectPermission.objects.count(), 1) self.assertFalse(self.user.has_perm("tests.can_be_awesome", apple)) self.assertTrue(self.other_user.has_perm("tests.can_be_awesome", apple))