Fixes MySQL build.

This commit is contained in:
Bertrand Bordage 2014-10-04 12:13:04 +02:00
parent e8652451fe
commit bc27896eb9

View file

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