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:
parent
e746daa2cb
commit
e2a77980b6
|
@ -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.
|
||||
|
||||
|
|
|
@ -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.
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue