From 4a9ed8aa9d32db0000be324c0e1aea29c600fe48 Mon Sep 17 00:00:00 2001
From: Jason Ward <jason.ward@policystat.com>
Date: Mon, 18 Mar 2013 15:46:34 -0400
Subject: [PATCH] refs #11: added tests for group caching

---
 authority/tests.py  | 66 +++++++++++++++++++++++++++++++++++++++------
 example/settings.py |  6 +++++
 2 files changed, 64 insertions(+), 8 deletions(-)

diff --git a/authority/tests.py b/authority/tests.py
index 79d8d46..12288f2 100644
--- a/authority/tests.py
+++ b/authority/tests.py
@@ -210,9 +210,24 @@ class PerformanceTest(SmartCachingTestCase):
         # Regardless of how many times has_user_perms is called, the number of
         # queries is the same.
         with self.assertNumQueries(1):
-            self.user_check.has_user_perms('foo', self.user, True, False)
-            self.user_check.has_user_perms('foo', self.user, True, False)
-            self.user_check.has_user_perms('foo', self.user, True, False)
+            for _ in range(5):
+                # Need to assert it so the query actually gets executed.
+                assert not self.user_check.has_user_perms(
+                    'foo',
+                    self.user,
+                    True,
+                    False,
+                )
+
+    def test_group_has_perms(self):
+        with self.assertNumQueries(1):
+            for _ in range(5):
+                # Need to assert it so the query actually gets executed.
+                assert not self.group_check.has_group_perms(
+                    'foo',
+                    self.group,
+                    True,
+                )
 
     def test_has_user_perms_check_group(self):
         # Regardless of the number groups permissions, it should only take one
@@ -225,18 +240,53 @@ class PerformanceTest(SmartCachingTestCase):
                 check_groups=True,
             )
 
-    def test_invalidate_permissions_cache(self):
+    def test_invalidate_user_permissions_cache(self):
         # Show that calling invalidate_permissions_cache will cause extra
         # queries.
         with self.assertNumQueries(2):
-            self.user_check.has_user_perms('foo', self.user, True, False)
+            for _ in range(5):
+                assert not self.user_check.has_user_perms(
+                    'foo',
+                    self.user,
+                    True,
+                    False,
+                )
 
             # Invalidate the cache to show that a query will be generated when
             # checking perms again.
             self.user_check.invalidate_permissions_cache()
 
             # One query to re generate the cache.
-            self.user_check.has_user_perms('foo', self.user, True, False)
+            for _ in range(5):
+                assert not self.user_check.has_user_perms(
+                    'foo',
+                    self.user,
+                    True,
+                    False,
+                )
+
+    def test_invalidate_group_permissions_cache(self):
+        # Show that calling invalidate_permissions_cache will cause extra
+        # queries.
+        with self.assertNumQueries(2):
+            for _ in range(5):
+                assert not self.group_check.has_group_perms(
+                    'foo',
+                    self.group,
+                    True,
+                )
+
+            # Invalidate the cache to show that a query will be generated when
+            # checking perms again.
+            self.group_check.invalidate_permissions_cache()
+
+            # One query to re generate the cache.
+            for _ in range(5):
+                assert not self.group_check.has_group_perms(
+                    'foo',
+                    self.group,
+                    True,
+                )
 
     def test_has_user_perms_check_group_multiple(self):
         # Create a permission with just a group.
@@ -250,7 +300,7 @@ class PerformanceTest(SmartCachingTestCase):
 
         # Check the number of queries.
         with self.assertNumQueries(1):
-            self.user_check.has_user_perms('foo', self.user, True, True)
+            assert self.user_check.has_user_perms('foo', self.user, True, True)
 
         # Create a second group.
         new_group = Group.objects.create(name='new_group')
@@ -269,7 +319,7 @@ class PerformanceTest(SmartCachingTestCase):
 
         # Make sure it is the same number of queries.
         with self.assertNumQueries(1):
-            self.user_check.has_user_perms('foo', self.user, True, True)
+            assert self.user_check.has_user_perms('foo', self.user, True, True)
 
 
 class GroupPermissionCacheTestCase(SmartCachingTestCase):
diff --git a/example/settings.py b/example/settings.py
index 80946c3..a9f382f 100644
--- a/example/settings.py
+++ b/example/settings.py
@@ -19,6 +19,12 @@ DATABASES = {
     }
 }
 
+CACHES = {
+    'default': {
+        'BACKEND': 'django.core.cache.backends.dummy.DummyCache',
+    }
+}
+
 TIME_ZONE = 'America/Chicago'
 
 LANGUAGE_CODE = 'en-us'