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 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)