diff --git a/cachalot/tests/multi_db.py b/cachalot/tests/multi_db.py index a9833b8..c189456 100644 --- a/cachalot/tests/multi_db.py +++ b/cachalot/tests/multi_db.py @@ -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 can’t change the DB used since there’s 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):