def test_incrementing_counter_works(self):
obj = Mock()
first = get_counter(obj)
self.assertEqual(first, 0)
increment_counter(obj)
second = get_counter(obj)
self.assertNotEqual(second, first)
def test_incrementing_counter_works(self):
obj = Mock()
first = get_counter(obj)
self.assertEqual(first, 0)
increment_counter(obj)
second = get_counter(obj)
self.assertNotEqual(second, first)
def test_rulez_invalidate_works(self):
model = TestModel()
user = MockUser()
first = get_counter(model)
self.assertEqual(first, 0)
model.rulez_invalidate()
second = get_counter(model)
self.assertNotEqual(second, first)
def test_rulez_invalidate_works(self):
model = TestModel()
user = MockUser()
first = get_counter(model)
self.assertEqual(first, 0)
model.rulez_invalidate()
second = get_counter(model)
self.assertNotEqual(second, first)
def roles_key(user, obj):
if obj.__class__ in (User, AnonymousUser,):
obj_id = get_user_pk(obj)
else:
obj_id = obj.pk
obj_type = str(obj.__class__.__name__).lower()
obj_counter = get_counter(obj)
user_id = get_user_pk(user)
user_counter = get_counter(user)
return "%s-%s-%s-%s-%s" % (user_id, user_counter, obj_type, obj_id,
obj_counter)
def test_get_roles_for_None_raises(self):
with self.assertRaises(AttributeError):
res = get_counter(None)
self.assertEqual(res, None)
def test_get_roles_for_None_raises(self):
with self.assertRaises(AttributeError):
res = get_counter(None)
self.assertEqual(res, None)
