Commit 6f3e1397 authored by Preston Timmons's avatar Preston Timmons Committed by Carl Meyer
Browse files

Modified test_client_override to work with unittest2 discovery.

parent 751b007c
Loading
Loading
Loading
Loading
+0 −21
Original line number Diff line number Diff line
from django.utils import unittest


def suite():
    """
    Validate that you can override the default test suite
    Define a suite that deliberately ignores a test defined in
    this module.
    """

    testSuite = unittest.TestSuite()
    testSuite.addTest(SampleTests('testGoodStuff'))
    return testSuite


class SampleTests(unittest.TestCase):
    def testGoodStuff(self):
        pass

    def testBadStuff(self):
        self.fail("This test shouldn't run")
+34 −0
Original line number Diff line number Diff line
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