Fixed #25855 -- Enhanced the migration warning for runserver.

Added unapplied migration count and the list of unmigrated apps.

9 files changed, 93 insertions, 1 deletions

diff --git a/tests/admin_scripts/another_app_waiting_migration/__init__.py b/tests/admin_scripts/another_app_waiting_migration/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/admin_scripts/another_app_waiting_migration/__init__.py
diff --git a/tests/admin_scripts/another_app_waiting_migration/migrations/0001_initial.py b/tests/admin_scripts/another_app_waiting_migration/migrations/0001_initial.py
new file mode 100644
index 0000000000..1486231d6b
--- /dev/null
+++ b/tests/admin_scripts/another_app_waiting_migration/migrations/0001_initial.py
@@ -0,0 +1,21 @@
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Foo',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255)),
+            ],
+        ),
+    ]
diff --git a/tests/admin_scripts/another_app_waiting_migration/migrations/__init__.py b/tests/admin_scripts/another_app_waiting_migration/migrations/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/admin_scripts/another_app_waiting_migration/migrations/__init__.py
diff --git a/tests/admin_scripts/another_app_waiting_migration/models.py b/tests/admin_scripts/another_app_waiting_migration/models.py
new file mode 100644
index 0000000000..6c12c6ab5d
--- /dev/null
+++ b/tests/admin_scripts/another_app_waiting_migration/models.py
@@ -0,0 +1,10 @@
+from __future__ import unicode_literals
+
+from django.db import models
+
+
+class Foo(models.Model):
+    name = models.CharField(max_length=255)
+
+    class Meta:
+        app_label = 'another_app_waiting_migration'
diff --git a/tests/admin_scripts/app_waiting_migration/__init__.py b/tests/admin_scripts/app_waiting_migration/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/admin_scripts/app_waiting_migration/__init__.py
diff --git a/tests/admin_scripts/app_waiting_migration/migrations/0001_initial.py b/tests/admin_scripts/app_waiting_migration/migrations/0001_initial.py
new file mode 100644
index 0000000000..52d594a94a
--- /dev/null
+++ b/tests/admin_scripts/app_waiting_migration/migrations/0001_initial.py
@@ -0,0 +1,21 @@
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Bar',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255)),
+            ],
+        ),
+    ]
diff --git a/tests/admin_scripts/app_waiting_migration/migrations/__init__.py b/tests/admin_scripts/app_waiting_migration/migrations/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/admin_scripts/app_waiting_migration/migrations/__init__.py
diff --git a/tests/admin_scripts/app_waiting_migration/models.py b/tests/admin_scripts/app_waiting_migration/models.py
new file mode 100644
index 0000000000..5e9f0e3314
--- /dev/null
+++ b/tests/admin_scripts/app_waiting_migration/models.py
@@ -0,0 +1,10 @@
+from __future__ import unicode_literals
+
+from django.db import models
+
+
+class Bar(models.Model):
+    name = models.CharField(max_length=255)
+
+    class Meta:
+        app_label = 'app_waiting_migration'
diff --git a/tests/admin_scripts/tests.py b/tests/admin_scripts/tests.py
index ea7c8a25f0..805be8b144 100644
--- a/tests/admin_scripts/tests.py
+++ b/tests/admin_scripts/tests.py
@@ -26,7 +26,7 @@ from django.db import ConnectionHandler
 from django.db.migrations.exceptions import MigrationSchemaMissing
 from django.db.migrations.recorder import MigrationRecorder
 from django.test import (
-    LiveServerTestCase, SimpleTestCase, mock, override_settings,
+    LiveServerTestCase, SimpleTestCase, TestCase, mock, override_settings,
 )
 from django.test.runner import DiscoverRunner
 from django.utils._os import npath, upath
@@ -1383,6 +1383,36 @@ class ManageRunserver(AdminScriptTestCase):
         self.assertIn("Not checking migrations", self.output.getvalue())
 
 
+class ManageRunserverMigrationWarning(TestCase):
+
+    def setUp(self):
+        from django.core.management.commands.runserver import Command
+        self.stdout = StringIO()
+        self.runserver_command = Command(stdout=self.stdout)
+
+    @override_settings(INSTALLED_APPS=["admin_scripts.app_waiting_migration"])
+    def test_migration_warning_one_app(self):
+        self.runserver_command.check_migrations()
+        output = self.stdout.getvalue()
+        self.assertIn('You have 1 unapplied migration(s)', output)
+        self.assertIn('apply the migrations for app(s): app_waiting_migration.', output)
+
+    @override_settings(
+        INSTALLED_APPS=[
+            "admin_scripts.app_waiting_migration",
+            "admin_scripts.another_app_waiting_migration",
+        ],
+    )
+    def test_migration_warning_multiple_apps(self):
+        self.runserver_command.check_migrations()
+        output = self.stdout.getvalue()
+        self.assertIn('You have 2 unapplied migration(s)', output)
+        self.assertIn(
+            'apply the migrations for app(s): another_app_waiting_migration, '
+            'app_waiting_migration.', output
+        )
+
+
 class ManageRunserverEmptyAllowedHosts(AdminScriptTestCase):
     def setUp(self):
         self.write_settings('settings.py', sdict={