Added more tests for issue #27122.

This commit is contained in:
Serhiy Storchaka 2016-06-20 05:29:54 +03:00
parent 5943ea76d5
commit ce1a9f309f
1 changed files with 10 additions and 4 deletions

View File

@ -763,8 +763,9 @@ class TestExitStack(unittest.TestCase):
self.assertIs(stack._exit_callbacks[-1], cm) self.assertIs(stack._exit_callbacks[-1], cm)
def test_dont_reraise_RuntimeError(self): def test_dont_reraise_RuntimeError(self):
"""https://bugs.python.org/issue27122""" # https://bugs.python.org/issue27122
class UniqueException(Exception): pass class UniqueException(Exception): pass
class UniqueRuntimeError(RuntimeError): pass
@contextmanager @contextmanager
def second(): def second():
@ -780,15 +781,20 @@ class TestExitStack(unittest.TestCase):
except Exception as exc: except Exception as exc:
raise exc raise exc
# The RuntimeError should be caught by second()'s exception # The UniqueRuntimeError should be caught by second()'s exception
# handler which chain raised a new UniqueException. # handler which chain raised a new UniqueException.
with self.assertRaises(UniqueException) as err_ctx: with self.assertRaises(UniqueException) as err_ctx:
with ExitStack() as es_ctx: with ExitStack() as es_ctx:
es_ctx.enter_context(second()) es_ctx.enter_context(second())
es_ctx.enter_context(first()) es_ctx.enter_context(first())
raise RuntimeError("please no infinite loop.") raise UniqueRuntimeError("please no infinite loop.")
self.assertEqual(err_ctx.exception.args[0], "new exception") exc = err_ctx.exception
self.assertIsInstance(exc, UniqueException)
self.assertIsInstance(exc.__context__, UniqueRuntimeError)
self.assertIsNone(exc.__context__.__context__)
self.assertIsNone(exc.__context__.__cause__)
self.assertIs(exc.__cause__, exc.__context__)
class TestRedirectStream: class TestRedirectStream: