Refs #31117 -- Made various tests properly handle unexpected databases aliases.

- Used selected "databases" instead of django.db.connections.
- Made routers in tests.migrations skip migrations on unexpected
  databases.
- Added DiscoverRunnerGetDatabasesTests.assertSkippedDatabases() hook
  which properly asserts messages about skipped databases.

tests/admin_views/test_multidb.py

@@ -2,7 +2,6 @@ from unittest import mock
 
 from django.contrib import admin
 from django.contrib.auth.models import User
-from django.db import connections
 from django.test import TestCase, override_settings
 from django.urls import path, reverse
 
@@ -34,7 +33,7 @@ class MultiDatabaseTests(TestCase):
     def setUpTestData(cls):
         cls.superusers = {}
         cls.test_book_ids = {}
-        for db in connections:
+        for db in cls.databases:
             Router.target_db = db
             cls.superusers[db] = User.objects.create_superuser(
                 username='admin', password='something', email='test@test.org',
@@ -45,7 +44,7 @@ class MultiDatabaseTests(TestCase):
 
     @mock.patch('django.contrib.admin.options.transaction')
     def test_add_view(self, mock):
-        for db in connections:
+        for db in self.databases:
             with self.subTest(db=db):
                 Router.target_db = db
                 self.client.force_login(self.superusers[db])
@@ -57,7 +56,7 @@ class MultiDatabaseTests(TestCase):
 
     @mock.patch('django.contrib.admin.options.transaction')
     def test_change_view(self, mock):
-        for db in connections:
+        for db in self.databases:
             with self.subTest(db=db):
                 Router.target_db = db
                 self.client.force_login(self.superusers[db])
@@ -69,7 +68,7 @@ class MultiDatabaseTests(TestCase):
 
     @mock.patch('django.contrib.admin.options.transaction')
     def test_delete_view(self, mock):
-        for db in connections:
+        for db in self.databases:
             with self.subTest(db=db):
                 Router.target_db = db
                 self.client.force_login(self.superusers[db])

tests/auth_tests/test_admin_multidb.py

@@ -3,7 +3,6 @@ from unittest import mock
 from django.contrib import admin
 from django.contrib.auth.admin import UserAdmin
 from django.contrib.auth.models import User
-from django.db import connections
 from django.test import TestCase, override_settings
 from django.urls import path, reverse
 
@@ -32,7 +31,7 @@ class MultiDatabaseTests(TestCase):
     @classmethod
     def setUpTestData(cls):
         cls.superusers = {}
-        for db in connections:
+        for db in cls.databases:
             Router.target_db = db
             cls.superusers[db] = User.objects.create_superuser(
                 username='admin', password='something', email='test@test.org',
@@ -40,7 +39,7 @@ class MultiDatabaseTests(TestCase):
 
     @mock.patch('django.contrib.auth.admin.transaction')
     def test_add_view(self, mock):
-        for db in connections:
+        for db in self.databases:
             with self.subTest(db_connection=db):
                 Router.target_db = db
                 self.client.force_login(self.superusers[db])

tests/backends/tests.py

@@ -687,11 +687,15 @@ class ThreadTests(TransactionTestCase):
                 conn.inc_thread_sharing()
                 connections_dict[id(conn)] = conn
         try:
-            for x in range(2):
+            num_new_threads = 2
+            for x in range(num_new_threads):
                 t = threading.Thread(target=runner)
                 t.start()
                 t.join()
-            self.assertEqual(len(connections_dict), 6)
+            self.assertEqual(
+                len(connections_dict),
+                len(connections.all()) * (num_new_threads + 1),
+            )
         finally:
             # Finish by closing the connections opened by the other threads
             # (the connection opened in the main thread will automatically be

tests/migrations/routers.py

@@ -1,5 +1,6 @@
-class EmptyRouter:
+class DefaultOtherRouter:
-    pass
+    def allow_migrate(self, db, app_label, model_name=None, **hints):
+        return db in {'default', 'other'}
 
 
 class TestRouter:
@@ -9,5 +10,5 @@ class TestRouter:
         """
         if model_name == 'tribble':
             return db == 'other'
-        elif db == 'other':
+        elif db != 'default':
             return False

tests/migrations/test_base.py

@@ -24,7 +24,7 @@ class MigrationTestBase(TransactionTestCase):
 
     def tearDown(self):
         # Reset applied-migrations state.
-        for db in connections:
+        for db in self.databases:
             recorder = MigrationRecorder(connections[db])
             recorder.migration_qs.filter(app='migrations').delete()
 

tests/migrations/test_commands.py

@@ -129,7 +129,7 @@ class MigrateTests(MigrationTestBase):
         that check.
         """
         # Make sure no tables are created
-        for db in connections:
+        for db in self.databases:
             self.assertTableNotExists("migrations_author", using=db)
             self.assertTableNotExists("migrations_tribble", using=db)
         # Run the migrations to 0001 only
@@ -192,7 +192,7 @@ class MigrateTests(MigrationTestBase):
         call_command("migrate", "migrations", "zero", verbosity=0)
         call_command("migrate", "migrations", "zero", verbosity=0, database="other")
         # Make sure it's all gone
-        for db in connections:
+        for db in self.databases:
             self.assertTableNotExists("migrations_author", using=db)
             self.assertTableNotExists("migrations_tribble", using=db)
             self.assertTableNotExists("migrations_book", using=db)
@@ -931,7 +931,7 @@ class MakeMigrationsTests(MigrationTestBase):
 
                 # With a router that doesn't prohibit migrating 'other',
                 # consistency is checked.
-                with self.settings(DATABASE_ROUTERS=['migrations.routers.EmptyRouter']):
+                with self.settings(DATABASE_ROUTERS=['migrations.routers.DefaultOtherRouter']):
                     with self.assertRaisesMessage(Exception, 'Other connection'):
                         call_command('makemigrations', 'migrations', verbosity=0)
                 self.assertEqual(has_table.call_count, 4)  # 'default' and 'other'
@@ -944,12 +944,14 @@ class MakeMigrationsTests(MigrationTestBase):
                 allow_migrate.assert_any_call('other', 'migrations', model_name='UnicodeModel')
                 # allow_migrate() is called with the correct arguments.
                 self.assertGreater(len(allow_migrate.mock_calls), 0)
+                called_aliases = set()
                 for mock_call in allow_migrate.mock_calls:
                     _, call_args, call_kwargs = mock_call
                     connection_alias, app_name = call_args
-                    self.assertIn(connection_alias, ['default', 'other'])
+                    called_aliases.add(connection_alias)
                     # Raises an error if invalid app_name/model_name occurs.
                     apps.get_app_config(app_name).get_model(call_kwargs['model_name'])
+                self.assertEqual(called_aliases, set(connections))
                 self.assertEqual(has_table.call_count, 4)
 
     def test_failing_migration(self):

tests/multiple_database/tests.py

@@ -7,7 +7,7 @@ from unittest.mock import Mock
 from django.contrib.auth.models import User
 from django.contrib.contenttypes.models import ContentType
 from django.core import management
-from django.db import DEFAULT_DB_ALIAS, connections, router, transaction
+from django.db import DEFAULT_DB_ALIAS, router, transaction
 from django.db.models import signals
 from django.db.utils import ConnectionRouter
 from django.test import SimpleTestCase, TestCase, override_settings
@@ -1632,7 +1632,7 @@ class PickleQuerySetTestCase(TestCase):
     databases = {'default', 'other'}
 
     def test_pickling(self):
-        for db in connections:
+        for db in self.databases:
             Book.objects.using(db).create(title='Dive into Python', published=datetime.date(2009, 5, 4))
             qs = Book.objects.all()
             self.assertEqual(qs.db, pickle.loads(pickle.dumps(qs)).db)

tests/test_runner/test_discover_runner.py

@@ -308,6 +308,15 @@ class DiscoverRunnerGetDatabasesTests(SimpleTestCase):
             databases = self.runner.get_databases(suite)
         return databases, stdout.getvalue()
 
+    def assertSkippedDatabases(self, test_labels, expected_databases):
+        databases, output = self.get_databases(test_labels)
+        self.assertEqual(databases, expected_databases)
+        skipped_databases = set(connections) - expected_databases
+        if skipped_databases:
+            self.assertIn(self.skip_msg + ', '.join(sorted(skipped_databases)), output)
+        else:
+            self.assertNotIn(self.skip_msg, output)
+
     def test_mixed(self):
         databases, output = self.get_databases(['test_runner_apps.databases.tests'])
         self.assertEqual(databases, set(connections))
@@ -319,24 +328,22 @@ class DiscoverRunnerGetDatabasesTests(SimpleTestCase):
         self.assertNotIn(self.skip_msg, output)
 
     def test_default_and_other(self):
-        databases, output = self.get_databases([
+        self.assertSkippedDatabases([
             'test_runner_apps.databases.tests.DefaultDatabaseTests',
             'test_runner_apps.databases.tests.OtherDatabaseTests',
-        ])
+        ], {'default', 'other'})
-        self.assertEqual(databases, set(connections))
-        self.assertNotIn(self.skip_msg, output)
 
     def test_default_only(self):
-        databases, output = self.get_databases(['test_runner_apps.databases.tests.DefaultDatabaseTests'])
+        self.assertSkippedDatabases([
-        self.assertEqual(databases, {'default'})
+            'test_runner_apps.databases.tests.DefaultDatabaseTests',
-        self.assertIn(self.skip_msg + 'other', output)
+        ], {'default'})
 
     def test_other_only(self):
-        databases, output = self.get_databases(['test_runner_apps.databases.tests.OtherDatabaseTests'])
+        self.assertSkippedDatabases([
-        self.assertEqual(databases, {'other'})
+            'test_runner_apps.databases.tests.OtherDatabaseTests'
-        self.assertIn(self.skip_msg + 'default', output)
+        ], {'other'})
 
     def test_no_databases_required(self):
-        databases, output = self.get_databases(['test_runner_apps.databases.tests.NoDatabaseTests'])
+        self.assertSkippedDatabases([
-        self.assertEqual(databases, set())
+            'test_runner_apps.databases.tests.NoDatabaseTests'
-        self.assertIn(self.skip_msg + 'default, other', output)
+        ], set())

tests/test_utils/test_transactiontestcase.py

@@ -44,7 +44,7 @@ class TransactionTestCaseDatabasesTests(TestCase):
         so that it's less likely to overflow. An overflow causes
         assertNumQueries() to fail.
         """
-        for alias in connections:
+        for alias in self.databases:
             self.assertEqual(len(connections[alias].queries_log), 0, 'Failed for alias %s' % alias)