35 lines
906 B
Python
35 lines
906 B
Python
|
from django.db.models import get_app
|
||
|
|
from django.test.simple import build_suite
|
||
|
|
from django.utils import unittest
|
||
|
|
|
||
|
|
|
||
|
|
def suite():
|
||
|
|
testSuite = unittest.TestSuite()
|
||
|
|
testSuite.addTest(SuiteOverrideTest('test_suite_override'))
|
||
|
|
return testSuite
|
||
|
|
|
||
|
|
|
||
|
|
class SuiteOverrideTest(unittest.TestCase):
|
||
|
|
def test_suite_override(self):
|
||
|
|
"""
|
||
|
|
Validate that you can define a custom suite when running tests with
|
||
|
|
``django.test.simple.DjangoTestSuiteRunner`` (which builds up a test
|
||
|
|
suite using ``build_suite``).
|
||
|
|
"""
|
||
|
|
|
||
|
|
app = get_app("test_suite_override")
|
||
|
|
suite = build_suite(app)
|
||
|
|
self.assertEqual(suite.countTestCases(), 1)
|
||
|
|
|
||
|
|
|
||
|
|
class SampleTests(unittest.TestCase):
|
||
|
|
"""These tests should not be discovered, due to the custom suite."""
|
||
|
|
def test_one(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_two(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_three(self):
|
||
|
|
pass
|