Fixes tests for Django 1.7 & MySQL.

noripyt · Oct 21, 2014 · 5a4f0a4 · 5a4f0a4
1 parent e276d39
commit 5a4f0a4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cachalot/tests/write.py b/cachalot/tests/write.py
@@ -554,6 +554,11 @@ def test_flush(self):
 
         call_command('flush', verbosity=0, interactive=False)
 
+        if django_version >= (1, 7) and connection.vendor == 'mysql':
+            # We need to reopen the connection or Django
+            # will execute an extra SQL request below.
+            connection.cursor()
+
         with self.assertNumQueries(1):
             self.assertListEqual(list(Test.objects.all()), [])
 
@@ -564,6 +569,11 @@ def test_loaddata(self):
         call_command('loaddata', 'cachalot/tests/loaddata_fixture.json',
                      verbosity=0, interactive=False)
 
+        if django_version >= (1, 7) and connection.vendor == 'mysql':
+            # We need to reopen the connection or Django
+            # will execute an extra SQL request below.
+            connection.cursor()
+
         with self.assertNumQueries(1):
             self.assertListEqual([t.name for t in Test.objects.all()],
                                  ['test1', 'test2'])