From bc27896eb9d9eb187bab4bc9c60a77a36c8df925 Mon Sep 17 00:00:00 2001 From: Bertrand Bordage Date: Sat, 4 Oct 2014 12:13:04 +0200 Subject: [PATCH] Fixes MySQL build. --- cachalot/tests.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/cachalot/tests.py b/cachalot/tests.py index 5d4d79c..4253cad 100644 --- a/cachalot/tests.py +++ b/cachalot/tests.py @@ -900,6 +900,8 @@ class WriteTestCase(TestCase): self.assertEqual(data4[1].owner, u1) def test_invalidate_prefetch_related(self): + is_mysql = connection.vendor == 'mysql' + with self.assertNumQueries(1): data1 = list(Test.objects.select_related('owner') .prefetch_related('owner__groups__permissions')) @@ -942,7 +944,7 @@ class WriteTestCase(TestCase): with self.assertNumQueries(1): t2 = Test.objects.create(name='test2') - with self.assertNumQueries(1): + with self.assertNumQueries(3 if is_mysql else 1): data5 = list(Test.objects.select_related('owner') .prefetch_related('owner__groups__permissions')) self.assertListEqual(data5, [t1, t2]) @@ -956,7 +958,7 @@ class WriteTestCase(TestCase): with self.assertNumQueries(1): permissions[0].save() - with self.assertNumQueries(1): + with self.assertNumQueries(2 if is_mysql else 1): list(Test.objects.select_related('owner') .prefetch_related('owner__groups__permissions')) @@ -972,7 +974,7 @@ class WriteTestCase(TestCase): with self.assertNumQueries(1): User.objects.update(username='modified_user') - with self.assertNumQueries(2): + with self.assertNumQueries(3 if is_mysql else 2): data7 = list(Test.objects.select_related('owner') .prefetch_related('owner__groups__permissions')) self.assertEqual(data7[0].owner.username, 'modified_user')