3 files changed, 108 insertions, 0 deletions
diff --git a/tests/check/__init__.py b/tests/check/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/check/__init__.py
diff --git a/tests/check/models.py b/tests/check/models.py
new file mode 100644
index 0000000000..78a10abba6
--- /dev/null
+++ b/tests/check/models.py
@@ -0,0 +1 @@
+# Stubby.
diff --git a/tests/check/tests.py b/tests/check/tests.py
new file mode 100644
index 0000000000..98495e38ae
--- /dev/null
+++ b/tests/check/tests.py
@@ -0,0 +1,107 @@
+from django.core.checks.compatibility import base
+from django.core.checks.compatibility import django_1_6_0
+from django.core.management.commands import check
+from django.core.management import call_command
+from django.test import TestCase
+
+
+class StubCheckModule(object):
+    # Has no ``run_checks`` attribute & will trigger a warning.
+    __name__ = 'StubCheckModule'
+
+
+class FakeWarnings(object):
+    def __init__(self):
+        self._warnings = []
+
+    def warn(self, message):
+        self._warnings.append(message)
+
+
+class CompatChecksTestCase(TestCase):
+    def setUp(self):
+        super(CompatChecksTestCase, self).setUp()
+
+        # We're going to override the list of checks to perform for test
+        # consistency in the future.
+        self.old_compat_checks = base.COMPAT_CHECKS
+        base.COMPAT_CHECKS = [
+            django_1_6_0,
+        ]
+
+    def tearDown(self):
+        # Restore what's supposed to be in ``COMPAT_CHECKS``.
+        base.COMPAT_CHECKS = self.old_compat_checks
+        super(CompatChecksTestCase, self).tearDown()
+
+    def test_check_test_runner_new_default(self):
+        with self.settings(TEST_RUNNER='django.test.runner.DiscoverRunner'):
+            result = django_1_6_0.check_test_runner()
+            self.assertTrue("You have not explicitly set 'TEST_RUNNER'" in result)
+
+    def test_check_test_runner_overridden(self):
+        with self.settings(TEST_RUNNER='myapp.test.CustomRunnner'):
+            self.assertEqual(django_1_6_0.check_test_runner(), None)
+
+    def test_run_checks_new_default(self):
+        with self.settings(TEST_RUNNER='django.test.runner.DiscoverRunner'):
+            result = django_1_6_0.run_checks()
+            self.assertEqual(len(result), 1)
+            self.assertTrue("You have not explicitly set 'TEST_RUNNER'" in result[0])
+
+    def test_run_checks_overridden(self):
+        with self.settings(TEST_RUNNER='myapp.test.CustomRunnner'):
+            self.assertEqual(len(django_1_6_0.run_checks()), 0)
+
+    def test_check_compatibility(self):
+        with self.settings(TEST_RUNNER='django.test.runner.DiscoverRunner'):
+            result = base.check_compatibility()
+            self.assertEqual(len(result), 1)
+            self.assertTrue("You have not explicitly set 'TEST_RUNNER'" in result[0])
+
+        with self.settings(TEST_RUNNER='myapp.test.CustomRunnner'):
+            self.assertEqual(len(base.check_compatibility()), 0)
+
+    def test_check_compatibility_warning(self):
+        # First, we're patching over the ``COMPAT_CHECKS`` with a stub which
+        # will trigger the warning.
+        base.COMPAT_CHECKS = [
+            StubCheckModule(),
+        ]
+
+        # Next, we unfortunately have to patch out ``warnings``.
+        old_warnings = base.warnings
+        base.warnings = FakeWarnings()
+
+        self.assertEqual(len(base.warnings._warnings), 0)
+
+        with self.settings(TEST_RUNNER='myapp.test.CustomRunnner'):
+            self.assertEqual(len(base.check_compatibility()), 0)
+
+        self.assertEqual(len(base.warnings._warnings), 1)
+        self.assertTrue("The 'StubCheckModule' module lacks a 'run_checks'" in base.warnings._warnings[0])
+
+        # Restore the ``warnings``.
+        base.warnings = old_warnings
+
+    def test_management_command(self):
+        # Again, we unfortunately have to patch out ``warnings``. Different
+        old_warnings = check.warnings
+        check.warnings = FakeWarnings()
+
+        self.assertEqual(len(check.warnings._warnings), 0)
+
+        # Should not produce any warnings.
+        with self.settings(TEST_RUNNER='myapp.test.CustomRunnner'):
+            call_command('check')
+
+        self.assertEqual(len(check.warnings._warnings), 0)
+
+        with self.settings(TEST_RUNNER='django.test.runner.DiscoverRunner'):
+            call_command('check')
+
+        self.assertEqual(len(check.warnings._warnings), 1)
+        self.assertTrue("You have not explicitly set 'TEST_RUNNER'" in check.warnings._warnings[0])
+
+        # Restore the ``warnings``.
+        base.warnings = old_warnings