Kaydet (Commit) e2a77980 authored tarafından Kristján Valur Jónsson's avatar Kristján Valur Jónsson

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.
üst e746daa2
...@@ -888,6 +888,10 @@ Test cases ...@@ -888,6 +888,10 @@ Test cases
with self.failUnlessRaises(some_error_class): with self.failUnlessRaises(some_error_class):
do_something() 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 .. versionchanged:: 2.7
Added the ability to use :meth:`assertRaises` as a context manager. Added the ability to use :meth:`assertRaises` as a context manager.
......
...@@ -2821,6 +2821,21 @@ test case ...@@ -2821,6 +2821,21 @@ test case
self.assertRaisesRegexp, Exception, self.assertRaisesRegexp, Exception,
re.compile('^Expected$'), Stub) 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): def testSynonymAssertMethodNames(self):
"""Test undocumented method name synonyms. """Test undocumented method name synonyms.
......
...@@ -104,6 +104,7 @@ class _AssertRaisesContext(object): ...@@ -104,6 +104,7 @@ class _AssertRaisesContext(object):
if not issubclass(exc_type, self.expected): if not issubclass(exc_type, self.expected):
# let unexpected exceptions pass through # let unexpected exceptions pass through
return False return False
self.exc_value = exc_value #store for later retrieval
if self.expected_regex is None: if self.expected_regex is None:
return True return True
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment