issue 6275

Add an "exc_value" attribute to the _AssertRaisesContext context manager in the unittest package.  This allows further tests on the exception that was raised after the context manager exits.
This commit is contained in:
Kristján Valur Jónsson 2009-08-27 22:20:21 +00:00
parent e746daa2cb
commit e2a77980b6
3 changed files with 20 additions and 0 deletions

View File

@ -888,6 +888,10 @@ Test cases
with self.failUnlessRaises(some_error_class):
do_something()
The context manager will store the caught exception object in its
:attr:`exc_value` attribute. This can be useful if the intention
is to perform additional checks on the exception raised.
.. versionchanged:: 2.7
Added the ability to use :meth:`assertRaises` as a context manager.

View File

@ -2821,6 +2821,21 @@ test case
self.assertRaisesRegexp, Exception,
re.compile('^Expected$'), Stub)
def testAssertRaisesExcValue(self):
class ExceptionMock(Exception):
pass
def Stub(foo):
raise ExceptionMock(foo)
v = "particular value"
ctx = self.assertRaises(ExceptionMock)
with ctx:
Stub(v)
e = ctx.exc_value
self.assertTrue(isinstance(e, ExceptionMock))
self.assertEqual(e.args[0], v)
def testSynonymAssertMethodNames(self):
"""Test undocumented method name synonyms.

View File

@ -104,6 +104,7 @@ class _AssertRaisesContext(object):
if not issubclass(exc_type, self.expected):
# let unexpected exceptions pass through
return False
self.exc_value = exc_value #store for later retrieval
if self.expected_regex is None:
return True