| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387 |
- import io
- import sys
- import textwrap
- from test.support import warnings_helper, captured_stdout, captured_stderr
- import traceback
- import unittest
- from unittest.util import strclass
- class MockTraceback(object):
- class TracebackException:
- def __init__(self, *args, **kwargs):
- self.capture_locals = kwargs.get('capture_locals', False)
- def format(self):
- result = ['A traceback']
- if self.capture_locals:
- result.append('locals')
- return result
- def restore_traceback():
- unittest.result.traceback = traceback
- def bad_cleanup1():
- print('do cleanup1')
- raise TypeError('bad cleanup1')
- def bad_cleanup2():
- print('do cleanup2')
- raise ValueError('bad cleanup2')
- class BufferedWriter:
- def __init__(self):
- self.result = ''
- self.buffer = ''
- def write(self, arg):
- self.buffer += arg
- def flush(self):
- self.result += self.buffer
- self.buffer = ''
- def getvalue(self):
- return self.result
- class Test_TestResult(unittest.TestCase):
- # Note: there are not separate tests for TestResult.wasSuccessful(),
- # TestResult.errors, TestResult.failures, TestResult.testsRun or
- # TestResult.shouldStop because these only have meaning in terms of
- # other TestResult methods.
- #
- # Accordingly, tests for the aforenamed attributes are incorporated
- # in with the tests for the defining methods.
- ################################################################
- def test_init(self):
- result = unittest.TestResult()
- self.assertTrue(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 0)
- self.assertEqual(result.shouldStop, False)
- self.assertIsNone(result._stdout_buffer)
- self.assertIsNone(result._stderr_buffer)
- # "This method can be called to signal that the set of tests being
- # run should be aborted by setting the TestResult's shouldStop
- # attribute to True."
- def test_stop(self):
- result = unittest.TestResult()
- result.stop()
- self.assertEqual(result.shouldStop, True)
- # "Called when the test case test is about to be run. The default
- # implementation simply increments the instance's testsRun counter."
- def test_startTest(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- result = unittest.TestResult()
- result.startTest(test)
- self.assertTrue(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- result.stopTest(test)
- # "Called after the test case test has been executed, regardless of
- # the outcome. The default implementation does nothing."
- def test_stopTest(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- result = unittest.TestResult()
- result.startTest(test)
- self.assertTrue(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- result.stopTest(test)
- # Same tests as above; make sure nothing has changed
- self.assertTrue(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- # "Called before and after tests are run. The default implementation does nothing."
- def test_startTestRun_stopTestRun(self):
- result = unittest.TestResult()
- result.startTestRun()
- result.stopTestRun()
- # "addSuccess(test)"
- # ...
- # "Called when the test case test succeeds"
- # ...
- # "wasSuccessful() - Returns True if all tests run so far have passed,
- # otherwise returns False"
- # ...
- # "testsRun - The total number of tests run so far."
- # ...
- # "errors - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test which raised an
- # unexpected exception. Contains formatted
- # tracebacks instead of sys.exc_info() results."
- # ...
- # "failures - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test where a failure was
- # explicitly signalled using the TestCase.fail*() or TestCase.assert*()
- # methods. Contains formatted tracebacks instead
- # of sys.exc_info() results."
- def test_addSuccess(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- result = unittest.TestResult()
- result.startTest(test)
- result.addSuccess(test)
- result.stopTest(test)
- self.assertTrue(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- # "addFailure(test, err)"
- # ...
- # "Called when the test case test signals a failure. err is a tuple of
- # the form returned by sys.exc_info(): (type, value, traceback)"
- # ...
- # "wasSuccessful() - Returns True if all tests run so far have passed,
- # otherwise returns False"
- # ...
- # "testsRun - The total number of tests run so far."
- # ...
- # "errors - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test which raised an
- # unexpected exception. Contains formatted
- # tracebacks instead of sys.exc_info() results."
- # ...
- # "failures - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test where a failure was
- # explicitly signalled using the TestCase.fail*() or TestCase.assert*()
- # methods. Contains formatted tracebacks instead
- # of sys.exc_info() results."
- def test_addFailure(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- try:
- test.fail("foo")
- except:
- exc_info_tuple = sys.exc_info()
- result = unittest.TestResult()
- result.startTest(test)
- result.addFailure(test, exc_info_tuple)
- result.stopTest(test)
- self.assertFalse(result.wasSuccessful())
- self.assertEqual(len(result.errors), 0)
- self.assertEqual(len(result.failures), 1)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- test_case, formatted_exc = result.failures[0]
- self.assertIs(test_case, test)
- self.assertIsInstance(formatted_exc, str)
- def test_addFailure_filter_traceback_frames(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- def get_exc_info():
- try:
- test.fail("foo")
- except:
- return sys.exc_info()
- exc_info_tuple = get_exc_info()
- full_exc = traceback.format_exception(*exc_info_tuple)
- result = unittest.TestResult()
- result.startTest(test)
- result.addFailure(test, exc_info_tuple)
- result.stopTest(test)
- formatted_exc = result.failures[0][1]
- dropped = [l for l in full_exc if l not in formatted_exc]
- self.assertEqual(len(dropped), 1)
- self.assertIn("raise self.failureException(msg)", dropped[0])
- def test_addFailure_filter_traceback_frames_context(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- def get_exc_info():
- try:
- try:
- test.fail("foo")
- except:
- raise ValueError(42)
- except:
- return sys.exc_info()
- exc_info_tuple = get_exc_info()
- full_exc = traceback.format_exception(*exc_info_tuple)
- result = unittest.TestResult()
- result.startTest(test)
- result.addFailure(test, exc_info_tuple)
- result.stopTest(test)
- formatted_exc = result.failures[0][1]
- dropped = [l for l in full_exc if l not in formatted_exc]
- self.assertEqual(len(dropped), 1)
- self.assertIn("raise self.failureException(msg)", dropped[0])
- def test_addFailure_filter_traceback_frames_chained_exception_self_loop(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- def get_exc_info():
- try:
- loop = Exception("Loop")
- loop.__cause__ = loop
- loop.__context__ = loop
- raise loop
- except:
- return sys.exc_info()
- exc_info_tuple = get_exc_info()
- test = Foo('test_1')
- result = unittest.TestResult()
- result.startTest(test)
- result.addFailure(test, exc_info_tuple)
- result.stopTest(test)
- formatted_exc = result.failures[0][1]
- self.assertEqual(formatted_exc.count("Exception: Loop\n"), 1)
- def test_addFailure_filter_traceback_frames_chained_exception_cycle(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- def get_exc_info():
- try:
- # Create two directionally opposed cycles
- # __cause__ in one direction, __context__ in the other
- A, B, C = Exception("A"), Exception("B"), Exception("C")
- edges = [(C, B), (B, A), (A, C)]
- for ex1, ex2 in edges:
- ex1.__cause__ = ex2
- ex2.__context__ = ex1
- raise C
- except:
- return sys.exc_info()
- exc_info_tuple = get_exc_info()
- test = Foo('test_1')
- result = unittest.TestResult()
- result.startTest(test)
- result.addFailure(test, exc_info_tuple)
- result.stopTest(test)
- formatted_exc = result.failures[0][1]
- self.assertEqual(formatted_exc.count("Exception: A\n"), 1)
- self.assertEqual(formatted_exc.count("Exception: B\n"), 1)
- self.assertEqual(formatted_exc.count("Exception: C\n"), 1)
- # "addError(test, err)"
- # ...
- # "Called when the test case test raises an unexpected exception err
- # is a tuple of the form returned by sys.exc_info():
- # (type, value, traceback)"
- # ...
- # "wasSuccessful() - Returns True if all tests run so far have passed,
- # otherwise returns False"
- # ...
- # "testsRun - The total number of tests run so far."
- # ...
- # "errors - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test which raised an
- # unexpected exception. Contains formatted
- # tracebacks instead of sys.exc_info() results."
- # ...
- # "failures - A list containing 2-tuples of TestCase instances and
- # formatted tracebacks. Each tuple represents a test where a failure was
- # explicitly signalled using the TestCase.fail*() or TestCase.assert*()
- # methods. Contains formatted tracebacks instead
- # of sys.exc_info() results."
- def test_addError(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- pass
- test = Foo('test_1')
- try:
- raise TypeError()
- except:
- exc_info_tuple = sys.exc_info()
- result = unittest.TestResult()
- result.startTest(test)
- result.addError(test, exc_info_tuple)
- result.stopTest(test)
- self.assertFalse(result.wasSuccessful())
- self.assertEqual(len(result.errors), 1)
- self.assertEqual(len(result.failures), 0)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- test_case, formatted_exc = result.errors[0]
- self.assertIs(test_case, test)
- self.assertIsInstance(formatted_exc, str)
- def test_addError_locals(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- 1/0
- test = Foo('test_1')
- result = unittest.TestResult()
- result.tb_locals = True
- unittest.result.traceback = MockTraceback
- self.addCleanup(restore_traceback)
- result.startTestRun()
- test.run(result)
- result.stopTestRun()
- self.assertEqual(len(result.errors), 1)
- test_case, formatted_exc = result.errors[0]
- self.assertEqual('A tracebacklocals', formatted_exc)
- def test_addSubTest(self):
- class Foo(unittest.TestCase):
- def test_1(self):
- nonlocal subtest
- with self.subTest(foo=1):
- subtest = self._subtest
- try:
- 1/0
- except ZeroDivisionError:
- exc_info_tuple = sys.exc_info()
- # Register an error by hand (to check the API)
- result.addSubTest(test, subtest, exc_info_tuple)
- # Now trigger a failure
- self.fail("some recognizable failure")
- subtest = None
- test = Foo('test_1')
- result = unittest.TestResult()
- test.run(result)
- self.assertFalse(result.wasSuccessful())
- self.assertEqual(len(result.errors), 1)
- self.assertEqual(len(result.failures), 1)
- self.assertEqual(result.testsRun, 1)
- self.assertEqual(result.shouldStop, False)
- test_case, formatted_exc = result.errors[0]
- self.assertIs(test_case, subtest)
- self.assertIn("ZeroDivisionError", formatted_exc)
- test_case, formatted_exc = result.failures[0]
- self.assertIs(test_case, subtest)
- self.assertIn("some recognizable failure", formatted_exc)
- def testStackFrameTrimming(self):
- class Frame(object):
- class tb_frame(object):
- f_globals = {}
- result = unittest.TestResult()
- self.assertFalse(result._is_relevant_tb_level(Frame))
- Frame.tb_frame.f_globals['__unittest'] = True
- self.assertTrue(result._is_relevant_tb_level(Frame))
- def testFailFast(self):
- result = unittest.TestResult()
- result._exc_info_to_string = lambda *_: ''
- result.failfast = True
- result.addError(None, None)
- self.assertTrue(result.shouldStop)
- result = unittest.TestResult()
- result._exc_info_to_string = lambda *_: ''
- result.failfast = True
- result.addFailure(None, None)
- self.assertTrue(result.shouldStop)
- result = unittest.TestResult()
- result._exc_info_to_string = lambda *_: ''
- result.failfast = True
- result.addUnexpectedSuccess(None)
- self.assertTrue(result.shouldStop)
- def testFailFastSetByRunner(self):
- stream = BufferedWriter()
- runner = unittest.TextTestRunner(stream=stream, failfast=True)
- def test(result):
- self.assertTrue(result.failfast)
- result = runner.run(test)
- stream.flush()
- self.assertTrue(stream.getvalue().endswith('\n\nOK\n'))
- class Test_TextTestResult(unittest.TestCase):
- maxDiff = None
- def testGetDescriptionWithoutDocstring(self):
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self),
- 'testGetDescriptionWithoutDocstring (' + __name__ +
- '.Test_TextTestResult.testGetDescriptionWithoutDocstring)')
- def testGetSubTestDescriptionWithoutDocstring(self):
- with self.subTest(foo=1, bar=2):
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self._subtest),
- 'testGetSubTestDescriptionWithoutDocstring (' + __name__ +
- '.Test_TextTestResult.testGetSubTestDescriptionWithoutDocstring) (foo=1, bar=2)')
- with self.subTest('some message'):
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self._subtest),
- 'testGetSubTestDescriptionWithoutDocstring (' + __name__ +
- '.Test_TextTestResult.testGetSubTestDescriptionWithoutDocstring) [some message]')
- def testGetSubTestDescriptionWithoutDocstringAndParams(self):
- with self.subTest():
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self._subtest),
- 'testGetSubTestDescriptionWithoutDocstringAndParams '
- '(' + __name__ + '.Test_TextTestResult.testGetSubTestDescriptionWithoutDocstringAndParams) '
- '(<subtest>)')
- def testGetSubTestDescriptionForFalsyValues(self):
- expected = 'testGetSubTestDescriptionForFalsyValues (%s.Test_TextTestResult.testGetSubTestDescriptionForFalsyValues) [%s]'
- result = unittest.TextTestResult(None, True, 1)
- for arg in [0, None, []]:
- with self.subTest(arg):
- self.assertEqual(
- result.getDescription(self._subtest),
- expected % (__name__, arg)
- )
- def testGetNestedSubTestDescriptionWithoutDocstring(self):
- with self.subTest(foo=1):
- with self.subTest(baz=2, bar=3):
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self._subtest),
- 'testGetNestedSubTestDescriptionWithoutDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetNestedSubTestDescriptionWithoutDocstring) '
- '(baz=2, bar=3, foo=1)')
- def testGetDuplicatedNestedSubTestDescriptionWithoutDocstring(self):
- with self.subTest(foo=1, bar=2):
- with self.subTest(baz=3, bar=4):
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self._subtest),
- 'testGetDuplicatedNestedSubTestDescriptionWithoutDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetDuplicatedNestedSubTestDescriptionWithoutDocstring) (baz=3, bar=4, foo=1)')
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testGetDescriptionWithOneLineDocstring(self):
- """Tests getDescription() for a method with a docstring."""
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self),
- ('testGetDescriptionWithOneLineDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetDescriptionWithOneLineDocstring)\n'
- 'Tests getDescription() for a method with a docstring.'))
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testGetSubTestDescriptionWithOneLineDocstring(self):
- """Tests getDescription() for a method with a docstring."""
- result = unittest.TextTestResult(None, True, 1)
- with self.subTest(foo=1, bar=2):
- self.assertEqual(
- result.getDescription(self._subtest),
- ('testGetSubTestDescriptionWithOneLineDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetSubTestDescriptionWithOneLineDocstring) '
- '(foo=1, bar=2)\n'
- 'Tests getDescription() for a method with a docstring.'))
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testGetDescriptionWithMultiLineDocstring(self):
- """Tests getDescription() for a method with a longer docstring.
- The second line of the docstring.
- """
- result = unittest.TextTestResult(None, True, 1)
- self.assertEqual(
- result.getDescription(self),
- ('testGetDescriptionWithMultiLineDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetDescriptionWithMultiLineDocstring)\n'
- 'Tests getDescription() for a method with a longer '
- 'docstring.'))
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testGetSubTestDescriptionWithMultiLineDocstring(self):
- """Tests getDescription() for a method with a longer docstring.
- The second line of the docstring.
- """
- result = unittest.TextTestResult(None, True, 1)
- with self.subTest(foo=1, bar=2):
- self.assertEqual(
- result.getDescription(self._subtest),
- ('testGetSubTestDescriptionWithMultiLineDocstring '
- '(' + __name__ + '.Test_TextTestResult.testGetSubTestDescriptionWithMultiLineDocstring) '
- '(foo=1, bar=2)\n'
- 'Tests getDescription() for a method with a longer '
- 'docstring.'))
- class Test(unittest.TestCase):
- def testSuccess(self):
- pass
- def testSkip(self):
- self.skipTest('skip')
- def testFail(self):
- self.fail('fail')
- def testError(self):
- raise Exception('error')
- @unittest.expectedFailure
- def testExpectedFailure(self):
- self.fail('fail')
- @unittest.expectedFailure
- def testUnexpectedSuccess(self):
- pass
- def testSubTestSuccess(self):
- with self.subTest('one', a=1):
- pass
- with self.subTest('two', b=2):
- pass
- def testSubTestMixed(self):
- with self.subTest('success', a=1):
- pass
- with self.subTest('skip', b=2):
- self.skipTest('skip')
- with self.subTest('fail', c=3):
- self.fail('fail')
- with self.subTest('error', d=4):
- raise Exception('error')
- tearDownError = None
- def tearDown(self):
- if self.tearDownError is not None:
- raise self.tearDownError
- def _run_test(self, test_name, verbosity, tearDownError=None):
- stream = BufferedWriter()
- stream = unittest.runner._WritelnDecorator(stream)
- result = unittest.TextTestResult(stream, True, verbosity)
- test = self.Test(test_name)
- test.tearDownError = tearDownError
- test.run(result)
- return stream.getvalue()
- def testDotsOutput(self):
- self.assertEqual(self._run_test('testSuccess', 1), '.')
- self.assertEqual(self._run_test('testSkip', 1), 's')
- self.assertEqual(self._run_test('testFail', 1), 'F')
- self.assertEqual(self._run_test('testError', 1), 'E')
- self.assertEqual(self._run_test('testExpectedFailure', 1), 'x')
- self.assertEqual(self._run_test('testUnexpectedSuccess', 1), 'u')
- def testLongOutput(self):
- classname = f'{__name__}.{self.Test.__qualname__}'
- self.assertEqual(self._run_test('testSuccess', 2),
- f'testSuccess ({classname}.testSuccess) ... ok\n')
- self.assertEqual(self._run_test('testSkip', 2),
- f"testSkip ({classname}.testSkip) ... skipped 'skip'\n")
- self.assertEqual(self._run_test('testFail', 2),
- f'testFail ({classname}.testFail) ... FAIL\n')
- self.assertEqual(self._run_test('testError', 2),
- f'testError ({classname}.testError) ... ERROR\n')
- self.assertEqual(self._run_test('testExpectedFailure', 2),
- f'testExpectedFailure ({classname}.testExpectedFailure) ... expected failure\n')
- self.assertEqual(self._run_test('testUnexpectedSuccess', 2),
- f'testUnexpectedSuccess ({classname}.testUnexpectedSuccess) ... unexpected success\n')
- def testDotsOutputSubTestSuccess(self):
- self.assertEqual(self._run_test('testSubTestSuccess', 1), '.')
- def testLongOutputSubTestSuccess(self):
- classname = f'{__name__}.{self.Test.__qualname__}'
- self.assertEqual(self._run_test('testSubTestSuccess', 2),
- f'testSubTestSuccess ({classname}.testSubTestSuccess) ... ok\n')
- def testDotsOutputSubTestMixed(self):
- self.assertEqual(self._run_test('testSubTestMixed', 1), 'sFE')
- def testLongOutputSubTestMixed(self):
- classname = f'{__name__}.{self.Test.__qualname__}'
- self.assertEqual(self._run_test('testSubTestMixed', 2),
- f'testSubTestMixed ({classname}.testSubTestMixed) ... \n'
- f" testSubTestMixed ({classname}.testSubTestMixed) [skip] (b=2) ... skipped 'skip'\n"
- f' testSubTestMixed ({classname}.testSubTestMixed) [fail] (c=3) ... FAIL\n'
- f' testSubTestMixed ({classname}.testSubTestMixed) [error] (d=4) ... ERROR\n')
- def testDotsOutputTearDownFail(self):
- out = self._run_test('testSuccess', 1, AssertionError('fail'))
- self.assertEqual(out, 'F')
- out = self._run_test('testError', 1, AssertionError('fail'))
- self.assertEqual(out, 'EF')
- out = self._run_test('testFail', 1, Exception('error'))
- self.assertEqual(out, 'FE')
- out = self._run_test('testSkip', 1, AssertionError('fail'))
- self.assertEqual(out, 'sF')
- def testLongOutputTearDownFail(self):
- classname = f'{__name__}.{self.Test.__qualname__}'
- out = self._run_test('testSuccess', 2, AssertionError('fail'))
- self.assertEqual(out,
- f'testSuccess ({classname}.testSuccess) ... FAIL\n')
- out = self._run_test('testError', 2, AssertionError('fail'))
- self.assertEqual(out,
- f'testError ({classname}.testError) ... ERROR\n'
- f'testError ({classname}.testError) ... FAIL\n')
- out = self._run_test('testFail', 2, Exception('error'))
- self.assertEqual(out,
- f'testFail ({classname}.testFail) ... FAIL\n'
- f'testFail ({classname}.testFail) ... ERROR\n')
- out = self._run_test('testSkip', 2, AssertionError('fail'))
- self.assertEqual(out,
- f"testSkip ({classname}.testSkip) ... skipped 'skip'\n"
- f'testSkip ({classname}.testSkip) ... FAIL\n')
- classDict = dict(unittest.TestResult.__dict__)
- for m in ('addSkip', 'addExpectedFailure', 'addUnexpectedSuccess',
- '__init__'):
- del classDict[m]
- def __init__(self, stream=None, descriptions=None, verbosity=None):
- self.failures = []
- self.errors = []
- self.testsRun = 0
- self.shouldStop = False
- self.buffer = False
- self.tb_locals = False
- classDict['__init__'] = __init__
- OldResult = type('OldResult', (object,), classDict)
- class Test_OldTestResult(unittest.TestCase):
- def assertOldResultWarning(self, test, failures):
- with warnings_helper.check_warnings(
- ("TestResult has no add.+ method,", RuntimeWarning)):
- result = OldResult()
- test.run(result)
- self.assertEqual(len(result.failures), failures)
- def testOldTestResult(self):
- class Test(unittest.TestCase):
- def testSkip(self):
- self.skipTest('foobar')
- @unittest.expectedFailure
- def testExpectedFail(self):
- raise TypeError
- @unittest.expectedFailure
- def testUnexpectedSuccess(self):
- pass
- for test_name, should_pass in (('testSkip', True),
- ('testExpectedFail', True),
- ('testUnexpectedSuccess', False)):
- test = Test(test_name)
- self.assertOldResultWarning(test, int(not should_pass))
- def testOldTestTesultSetup(self):
- class Test(unittest.TestCase):
- def setUp(self):
- self.skipTest('no reason')
- def testFoo(self):
- pass
- self.assertOldResultWarning(Test('testFoo'), 0)
- def testOldTestResultClass(self):
- @unittest.skip('no reason')
- class Test(unittest.TestCase):
- def testFoo(self):
- pass
- self.assertOldResultWarning(Test('testFoo'), 0)
- def testOldResultWithRunner(self):
- class Test(unittest.TestCase):
- def testFoo(self):
- pass
- runner = unittest.TextTestRunner(resultclass=OldResult,
- stream=io.StringIO())
- # This will raise an exception if TextTestRunner can't handle old
- # test result objects
- runner.run(Test('testFoo'))
- class TestOutputBuffering(unittest.TestCase):
- def setUp(self):
- self._real_out = sys.stdout
- self._real_err = sys.stderr
- def tearDown(self):
- sys.stdout = self._real_out
- sys.stderr = self._real_err
- def testBufferOutputOff(self):
- real_out = self._real_out
- real_err = self._real_err
- result = unittest.TestResult()
- self.assertFalse(result.buffer)
- self.assertIs(real_out, sys.stdout)
- self.assertIs(real_err, sys.stderr)
- result.startTest(self)
- self.assertIs(real_out, sys.stdout)
- self.assertIs(real_err, sys.stderr)
- def testBufferOutputStartTestAddSuccess(self):
- real_out = self._real_out
- real_err = self._real_err
- result = unittest.TestResult()
- self.assertFalse(result.buffer)
- result.buffer = True
- self.assertIs(real_out, sys.stdout)
- self.assertIs(real_err, sys.stderr)
- result.startTest(self)
- self.assertIsNot(real_out, sys.stdout)
- self.assertIsNot(real_err, sys.stderr)
- self.assertIsInstance(sys.stdout, io.StringIO)
- self.assertIsInstance(sys.stderr, io.StringIO)
- self.assertIsNot(sys.stdout, sys.stderr)
- out_stream = sys.stdout
- err_stream = sys.stderr
- result._original_stdout = io.StringIO()
- result._original_stderr = io.StringIO()
- print('foo')
- print('bar', file=sys.stderr)
- self.assertEqual(out_stream.getvalue(), 'foo\n')
- self.assertEqual(err_stream.getvalue(), 'bar\n')
- self.assertEqual(result._original_stdout.getvalue(), '')
- self.assertEqual(result._original_stderr.getvalue(), '')
- result.addSuccess(self)
- result.stopTest(self)
- self.assertIs(sys.stdout, result._original_stdout)
- self.assertIs(sys.stderr, result._original_stderr)
- self.assertEqual(result._original_stdout.getvalue(), '')
- self.assertEqual(result._original_stderr.getvalue(), '')
- self.assertEqual(out_stream.getvalue(), '')
- self.assertEqual(err_stream.getvalue(), '')
- def getStartedResult(self):
- result = unittest.TestResult()
- result.buffer = True
- result.startTest(self)
- return result
- def testBufferOutputAddErrorOrFailure(self):
- unittest.result.traceback = MockTraceback
- self.addCleanup(restore_traceback)
- for message_attr, add_attr, include_error in [
- ('errors', 'addError', True),
- ('failures', 'addFailure', False),
- ('errors', 'addError', True),
- ('failures', 'addFailure', False)
- ]:
- result = self.getStartedResult()
- buffered_out = sys.stdout
- buffered_err = sys.stderr
- result._original_stdout = io.StringIO()
- result._original_stderr = io.StringIO()
- print('foo', file=sys.stdout)
- if include_error:
- print('bar', file=sys.stderr)
- addFunction = getattr(result, add_attr)
- addFunction(self, (None, None, None))
- result.stopTest(self)
- result_list = getattr(result, message_attr)
- self.assertEqual(len(result_list), 1)
- test, message = result_list[0]
- expectedOutMessage = textwrap.dedent("""
- Stdout:
- foo
- """)
- expectedErrMessage = ''
- if include_error:
- expectedErrMessage = textwrap.dedent("""
- Stderr:
- bar
- """)
- expectedFullMessage = 'A traceback%s%s' % (expectedOutMessage, expectedErrMessage)
- self.assertIs(test, self)
- self.assertEqual(result._original_stdout.getvalue(), expectedOutMessage)
- self.assertEqual(result._original_stderr.getvalue(), expectedErrMessage)
- self.assertMultiLineEqual(message, expectedFullMessage)
- def testBufferSetUp(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def setUp(self):
- print('set up')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = f'test_foo ({strclass(Foo)}.test_foo)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(str(test_case), description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDown(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def tearDown(self):
- print('tear down')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = f'test_foo ({strclass(Foo)}.test_foo)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(str(test_case), description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferDoCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def setUp(self):
- print('set up')
- self.addCleanup(bad_cleanup1)
- self.addCleanup(bad_cleanup2)
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 2)
- description = f'test_foo ({strclass(Foo)}.test_foo)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(str(test_case), description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up\ndo cleanup2\n', formatted_exc)
- self.assertNotIn('\ndo cleanup1\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(str(test_case), description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferSetUp_DoCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def setUp(self):
- print('set up')
- self.addCleanup(bad_cleanup1)
- self.addCleanup(bad_cleanup2)
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 3)
- description = f'test_foo ({strclass(Foo)}.test_foo)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(str(test_case), description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up\n', formatted_exc)
- self.assertNotIn('\ndo cleanup2\n', formatted_exc)
- self.assertNotIn('\ndo cleanup1\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(str(test_case), description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up\ndo cleanup2\n', formatted_exc)
- self.assertNotIn('\ndo cleanup1\n', formatted_exc)
- test_case, formatted_exc = result.errors[2]
- self.assertEqual(str(test_case), description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDown_DoCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def setUp(self):
- print('set up')
- self.addCleanup(bad_cleanup1)
- self.addCleanup(bad_cleanup2)
- def tearDown(self):
- print('tear down')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up\ntear down\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 3)
- description = f'test_foo ({strclass(Foo)}.test_foo)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(str(test_case), description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up\ntear down\n', formatted_exc)
- self.assertNotIn('\ndo cleanup2\n', formatted_exc)
- self.assertNotIn('\ndo cleanup1\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(str(test_case), description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up\ntear down\ndo cleanup2\n', formatted_exc)
- self.assertNotIn('\ndo cleanup1\n', formatted_exc)
- test_case, formatted_exc = result.errors[2]
- self.assertEqual(str(test_case), description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferSetupClass(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- @classmethod
- def setUpClass(cls):
- print('set up class')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up class\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = f'setUpClass ({strclass(Foo)})'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDownClass(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- @classmethod
- def tearDownClass(cls):
- print('tear down class')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down class\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = f'tearDownClass ({strclass(Foo)})'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferDoClassCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- @classmethod
- def setUpClass(cls):
- print('set up class')
- cls.addClassCleanup(bad_cleanup1)
- cls.addClassCleanup(bad_cleanup2)
- @classmethod
- def tearDownClass(cls):
- print('tear down class')
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down class\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 2)
- description = f'tearDownClass ({strclass(Foo)})'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(test_case.description, description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferSetupClass_DoClassCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- @classmethod
- def setUpClass(cls):
- print('set up class')
- cls.addClassCleanup(bad_cleanup1)
- cls.addClassCleanup(bad_cleanup2)
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up class\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 3)
- description = f'setUpClass ({strclass(Foo)})'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up class\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- test_case, formatted_exc = result.errors[2]
- self.assertEqual(test_case.description, description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDownClass_DoClassCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- @classmethod
- def setUpClass(cls):
- print('set up class')
- cls.addClassCleanup(bad_cleanup1)
- cls.addClassCleanup(bad_cleanup2)
- @classmethod
- def tearDownClass(cls):
- print('tear down class')
- 1/0
- def test_foo(self):
- pass
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down class\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 3)
- description = f'tearDownClass ({strclass(Foo)})'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\ntear down class\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- test_case, formatted_exc = result.errors[2]
- self.assertEqual(test_case.description, description)
- self.assertIn('TypeError: bad cleanup1', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferSetUpModule(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def test_foo(self):
- pass
- class Module(object):
- @staticmethod
- def setUpModule():
- print('set up module')
- 1/0
- Foo.__module__ = 'Module'
- sys.modules['Module'] = Module
- self.addCleanup(sys.modules.pop, 'Module')
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up module\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = 'setUpModule (Module)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDownModule(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def test_foo(self):
- pass
- class Module(object):
- @staticmethod
- def tearDownModule():
- print('tear down module')
- 1/0
- Foo.__module__ = 'Module'
- sys.modules['Module'] = Module
- self.addCleanup(sys.modules.pop, 'Module')
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down module\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = 'tearDownModule (Module)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferDoModuleCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def test_foo(self):
- pass
- class Module(object):
- @staticmethod
- def setUpModule():
- print('set up module')
- unittest.addModuleCleanup(bad_cleanup1)
- unittest.addModuleCleanup(bad_cleanup2)
- Foo.__module__ = 'Module'
- sys.modules['Module'] = Module
- self.addCleanup(sys.modules.pop, 'Module')
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 1)
- description = 'tearDownModule (Module)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferSetUpModule_DoModuleCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def test_foo(self):
- pass
- class Module(object):
- @staticmethod
- def setUpModule():
- print('set up module')
- unittest.addModuleCleanup(bad_cleanup1)
- unittest.addModuleCleanup(bad_cleanup2)
- 1/0
- Foo.__module__ = 'Module'
- sys.modules['Module'] = Module
- self.addCleanup(sys.modules.pop, 'Module')
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\nset up module\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 2)
- description = 'setUpModule (Module)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\nset up module\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertIn(expected_out, formatted_exc)
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- def testBufferTearDownModule_DoModuleCleanups(self):
- with captured_stdout() as stdout:
- result = unittest.TestResult()
- result.buffer = True
- class Foo(unittest.TestCase):
- def test_foo(self):
- pass
- class Module(object):
- @staticmethod
- def setUpModule():
- print('set up module')
- unittest.addModuleCleanup(bad_cleanup1)
- unittest.addModuleCleanup(bad_cleanup2)
- @staticmethod
- def tearDownModule():
- print('tear down module')
- 1/0
- Foo.__module__ = 'Module'
- sys.modules['Module'] = Module
- self.addCleanup(sys.modules.pop, 'Module')
- suite = unittest.TestSuite([Foo('test_foo')])
- suite(result)
- expected_out = '\nStdout:\ntear down module\ndo cleanup2\ndo cleanup1\n'
- self.assertEqual(stdout.getvalue(), expected_out)
- self.assertEqual(len(result.errors), 2)
- description = 'tearDownModule (Module)'
- test_case, formatted_exc = result.errors[0]
- self.assertEqual(test_case.description, description)
- self.assertIn('ZeroDivisionError: division by zero', formatted_exc)
- self.assertNotIn('ValueError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn('\nStdout:\ntear down module\n', formatted_exc)
- test_case, formatted_exc = result.errors[1]
- self.assertEqual(test_case.description, description)
- self.assertIn('ValueError: bad cleanup2', formatted_exc)
- self.assertNotIn('ZeroDivisionError', formatted_exc)
- self.assertNotIn('TypeError', formatted_exc)
- self.assertIn(expected_out, formatted_exc)
- if __name__ == '__main__':
- unittest.main()
|