def test_user_in_users(self):
self.request.user = User.objects.create(
username='******')
self.crit.users.add(self.request.user)
self.mock.StubOutWithMock(utils, 'cache_criteria')
utils.cache_criteria(instance=self.crit) # criteria
utils.cache_criteria(instance=self.crit) # users
self.mock.ReplayAll()
self.assertIs(
meets_criteria(self.request, 'test_crit'), True)
self.mock.VerifyAll()
def test_cache(self):
cache.add('criteria:test_crit', self.crit)
cache.add('criteria:test_crit:flags',
mox.SameElementsAs(self.crit.flags.all()))
cache.add('criteria:test_crit:users',
mox.SameElementsAs(self.crit.users.all()))
cache.add('criteria:test_crit:groups',
mox.SameElementsAs(self.crit.groups.all()))
self.mock.ReplayAll()
cache_criteria(instance=self.crit)
self.mock.VerifyAll()
def test_criteria_cached(self):
self.mock.StubOutWithMock(cache, 'get')
cache.get('criteria:test_crit').AndReturn(self.crit)
cache.get('criteria:test_crit:users')
cache.get('criteria:test_crit:groups')
self.mock.StubOutWithMock(utils, 'cache_criteria')
utils.cache_criteria(instance=self.crit) # users
utils.cache_criteria(instance=self.crit) # groups
self.mock.ReplayAll()
meets_criteria(self.request, 'test_crit')
self.mock.VerifyAll()
def test_user_in_cached_users(self):
self.request.user = User.objects.create(
username='******')
self.crit.users.add(self.request.user)
self.mock.StubOutWithMock(cache, 'get')
cache.get('criteria:test_crit')
cache.get('criteria:test_crit:users').AndReturn([self.request.user])
self.mock.StubOutWithMock(utils, 'cache_criteria')
utils.cache_criteria(instance=self.crit) # criteria
self.mock.ReplayAll()
self.assertIs(
meets_criteria(self.request, 'test_crit'), True)
self.mock.VerifyAll()
def test_meets_nothing(self):
self.mock.StubOutWithMock(cache, 'get')
cache.get('criteria:test_crit')
cache.get('criteria:test_crit:users')
cache.get('criteria:test_crit:groups')
self.mock.StubOutWithMock(utils, 'cache_criteria')
utils.cache_criteria(instance=self.crit) # criteria
utils.cache_criteria(instance=self.crit) # users
utils.cache_criteria(instance=self.crit) # groups
self.mock.ReplayAll()
self.assertIs(
meets_criteria(self.request, 'test_crit'), False)
self.mock.VerifyAll()
def test_cache_w_other_action(self):
self.mock.ReplayAll()
cache_criteria(instance=self.crit, action='pre_add')
self.mock.VerifyAll()
