Ticket #24396: 24396.patch

tests/migration_test_data_persistence/migrations/0002_add_book.py

@@ -5 +5 @@
 
 
 def add_book(apps, schema_editor):
-    apps.get_model("migration_test_data_persistence", "Book").objects.using(
-        schema_editor.connection.alias,
-    ).create(
+    if schema_editor.connection.alias != 'default':
+        return
+    apps.get_model("migration_test_data_persistence", "Book").objects.create(
         title="I Love Django",
     )
 

tests/routers.py

@@ +1 @@
+APPS_ON_OTHER_DATABASE = {
+    # Contrib apps
+    'auth',
+    'contenttypes',
+    'sites',
+    # Test apps
+    'multiple_database',
+    'prefetch_related',
+}
+
+class DjangoTestSuiteRouter(object):
+
+    def db_for_read(self, model, **hints):
+        return None
+
+    def db_for_write(self, model, **hints):
+        return None
+
+    def allow_relation(self, obj1, obj2, **hints):
+        return None
+
+    def allow_migrate(self, db, model):
+        """
+        Migrate only apps that have tests cases with multi_db = True
+        on the 'other' database.
+        """
+        if db == 'other':
+            return model._meta.app_label in APPS_ON_OTHER_DATABASE
+        return None

tests/runtests.py

@@ -118 +118 @@
         'STATIC_URL': settings.STATIC_URL,
         'STATIC_ROOT': settings.STATIC_ROOT,
         'MIDDLEWARE_CLASSES': settings.MIDDLEWARE_CLASSES,
+        'DATABASE_ROUTERS': settings.DATABASE_ROUTERS,
     }
 
     # Redirect some settings for the duration of these tests.
@@ -149 +150 @@
         'auth': 'django.contrib.auth.tests.migrations',
         'contenttypes': 'contenttypes_tests.migrations',
     }
+    settings.DATABASE_ROUTERS = ['routers.DjangoTestSuiteRouter']
 
     if verbosity > 0:
         # Ensure any warnings captured to logging are piped through a verbose