django/tests/test_suite_override/tests.py

35 lines
906 B
Python
Raw Normal View History

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