def test_get_perms(self):
user = User.objects.get(name='admin user')
zoo = Zoo.objects.first()
self.assertEqual(0, get_perms().count())
self.assertEqual(3, get_perms(ANY, ANY, ANY).count())
self.assertEqual(1, get_perms(
role='zoo.admin',
agent=user,
target=zoo
).count())
def test_add_perm(self):
user = User.objects.get(name='visiting user')
zoo = Zoo.objects.first()
# Do it twice - we should get no exceptions
add_perm('zoo.admin', user, zoo)
add_perm('zoo.admin', user, zoo)
self.assertEqual(1, get_perms('zoo.admin', user, zoo).count())
def test_remove_perm(self):
with self.assertRaises(TypeError):
remove_perm('zoo.admin')
remove_perm('zoo.admin', ANY, ANY)
self.assertEqual(0, get_perms('zoo.admin', ANY, ANY).count())