Fixed #16534 -- Improved ability to customize DiscoverRunner

Added DiscoverRunner.test_suite and .test_runner attributes.

Thanks tomchristie for the suggestion and jcd for the patch.

1 files changed, 10 insertions, 1 deletions

diff --git a/tests/test_runner/test_discover_runner.py b/tests/test_runner/test_discover_runner.py
index 4494b2bd3b..bd26e2e73d 100644
--- a/tests/test_runner/test_discover_runner.py
+++ b/tests/test_runner/test_discover_runner.py
@@ -1,7 +1,7 @@
 from contextlib import contextmanager
 import os
 import sys
-from unittest import expectedFailure
+from unittest import expectedFailure, TestSuite, TextTestRunner, defaultTestLoader
 
 from django.test import TestCase
 from django.test.runner import DiscoverRunner
@@ -68,3 +68,12 @@ class DiscoverRunnerTest(TestCase):
             ).countTestCases()
 
         self.assertEqual(count, 3)
+
+    def test_overrideable_test_suite(self):
+        self.assertEqual(DiscoverRunner().test_suite, TestSuite)
+
+    def test_overrideable_test_runner(self):
+        self.assertEqual(DiscoverRunner().test_runner, TextTestRunner)
+
+    def test_overrideable_test_loader(self):
+        self.assertEqual(DiscoverRunner().test_loader, defaultTestLoader)