From 4ba373840a7b299fff4a7bcc1001e32dffceee86 Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Tue, 10 Sep 2013 09:49:39 -0400 Subject: 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. --- tests/test_runner/test_discover_runner.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'tests') 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) -- cgit v1.3