Fixes multi-databases tests.

This commit is contained in:
Bertrand Bordage 2017-06-03 17:48:59 +02:00
parent 542a5b6097
commit d0ee580cf3
1 changed files with 5 additions and 4 deletions

View File

@ -8,17 +8,14 @@ from django.db import DEFAULT_DB_ALIAS, connections, transaction
from django.test import TransactionTestCase
from .models import Test
from .test_utils import TestUtilsMixin
@skipIf(len(settings.DATABASES) == 1,
'We cant change the DB used since theres only one configured')
class MultiDatabaseTestCase(TestUtilsMixin, TransactionTestCase):
class MultiDatabaseTestCase(TransactionTestCase):
multi_db = True
def setUp(self):
super(MultiDatabaseTestCase, self).setUp()
self.t1 = Test.objects.create(name='test1')
self.t2 = Test.objects.create(name='test2')
self.db_alias2 = next(alias for alias in settings.DATABASES
@ -26,6 +23,10 @@ class MultiDatabaseTestCase(TestUtilsMixin, TransactionTestCase):
connection2 = connections[self.db_alias2]
self.is_sqlite2 = connection2.vendor == 'sqlite'
self.is_mysql2 = connection2.vendor == 'mysql'
if connection2.vendor in ('mysql', 'postgresql'):
# We need to reopen the connection or Django
# will execute an extra SQL request below.
connection2.cursor()
def test_read(self):
with self.assertNumQueries(1):