Avoids extra invalidations during write operations.

noripyt · Oct 28, 2015 · 7177361 · 7177361
1 parent dcfffbc
commit 7177361
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cachalot/monkey_patch.py b/cachalot/monkey_patch.py
@@ -88,6 +88,7 @@ def inner(compiler, *args, **kwargs):
 
 def _patch_write_compiler(original):
     @wraps(original)
+    @_unset_raw_connection
     def inner(write_compiler, *args, **kwargs):
         db_alias = write_compiler.using
         table = write_compiler.query.get_meta().db_table

diff --git a/cachalot/tests/write.py b/cachalot/tests/write.py
@@ -586,7 +586,7 @@ def test_invalidate_prefetch_related(self):
             permissions = list(Permission.objects.all()[:5])
             group.permissions.add(*permissions)
             u.groups.add(group)
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(2):
             data4 = list(Test.objects.select_related('owner')
                          .prefetch_related('owner__groups__permissions'))
             self.assertListEqual(data4, [t1])