def test_no_args__correct(self):
try:
raise self.OBJECT
except:
with self.assertRaises(self.OBJECT.__class__) as r:
__unit__.raise_() # should re-raise caught exception
self.assertIs(self.OBJECT, r.exception)
def test_no_args__correct(self):
try:
raise self.OBJECT
except:
with self.assertRaises(self.OBJECT.__class__) as r:
__unit__.raise_() # should re-raise caught exception
self.assertIs(self.OBJECT, r.exception)
def test_exception_object(self):
with self.assertRaises(self.OBJECT.__class__) as r:
__unit__.raise_(self.OBJECT)
self.assertIs(self.OBJECT, r.exception)
def test_some_object(self):
with self.assertRaises(TypeError):
__unit__.raise_(object())
def test_none(self):
with self.assertRaises(TypeError):
__unit__.raise_(None)
def test_no_args__incorrect__py3(self):
with self.assertRaises(RuntimeError):
__unit__.raise_()
def test_no_args__incorrect__py2(self):
with self.assertRaises(TypeError):
__unit__.raise_()
class Try(TestCase):
EXCEPTION_CLASS = ArithmeticError
EXCEPTION = EXCEPTION_CLASS("foo")
DIFFERENT_EXCEPTION_CLASS = ReferenceError
BLOCK_RETVAL = -7
BLOCK = staticmethod(lambda: Try.BLOCK_RETVAL)
RAISE = staticmethod(lambda: __unit__.raise_(Try.EXCEPTION))
CATCH = staticmethod(lambda e: e)
RERAISE = staticmethod(lambda _: __unit__.raise_())
ELSE_RETVAL = 42
ELSE = staticmethod(lambda: Try.ELSE_RETVAL)
def test_no_args(self):
with self.assertRaises(TypeError):
__unit__.try_()
def test_block__none(self):
with self.assertRaises(TypeError):
__unit__.try_(None)
def test_block__some_object(self):
with self.assertRaises(TypeError):
__unit__.try_(object())
def test_block__function(self):
with self.assertRaises(TypeError) as r:
__unit__.try_(lambda: None)
msg = str(r.exception)
self.assertIn("at least one of", msg)
self.assertIn("except_", msg)
self.assertIn("else_", msg)
self.assertIn("finally_", msg)
def test_except__some_object(self):
with self.assertRaises(TypeError):
__unit__.try_(self.RAISE, except_=object())
def test_except__pass(self):
__unit__.try_(self.RAISE, except_=__unit__.pass_)
def test_except__catch__exception_raised(self):
self.assertIs(self.EXCEPTION,
__unit__.try_(self.RAISE, except_=self.CATCH))
def test_except__catch__no_exception(self):
retval = __unit__.try_(self.BLOCK, except_=self.CATCH)
self.assertEquals(self.BLOCK_RETVAL, retval)
def test_except__reraise(self):
with self.assertRaises(self.EXCEPTION_CLASS) as r:
__unit__.try_(self.RAISE, except_=self.RERAISE)
self.assertIs(self.EXCEPTION, r.exception)
def test_except__unordered_dict(self):
with self.assertRaises(TypeError):
__unit__.try_(self.BLOCK, except_={Exception: self.CATCH})
@skipIf(IS_PY26, "requires Python 2.7+")
def test_except__ordered_dict(self):
retval = __unit__.try_(self.RAISE,
except_=collections.OrderedDict([
(self.EXCEPTION_CLASS, self.CATCH)
]))
self.assertIs(self.EXCEPTION, retval)
def test_except__handler_list__empty(self):
with self.assertRaises(TypeError):
__unit__.try_(self.RAISE, except_=[])
def test_except__handler_list__singleton(self):
retval = __unit__.try_(self.RAISE,
except_=[(self.EXCEPTION_CLASS, self.CATCH)])
self.assertIs(self.EXCEPTION, retval)
def test_except__handler_list__match_first__exact(self):
retval = __unit__.try_(self.RAISE,
except_=[
(self.EXCEPTION_CLASS, self.CATCH),
(Exception, self.RAISE),
])
self.assertIs(self.EXCEPTION, retval)
def test_except__handler_list__match_first__superclass(self):
retval = __unit__.try_(self.RAISE,
except_=[
(Exception, self.CATCH),
(self.EXCEPTION_CLASS, self.RERAISE),
])
self.assertIs(self.EXCEPTION, retval)
def test_except__handler_list__bypass_first(self):
with self.assertRaises(self.EXCEPTION_CLASS):
__unit__.try_(self.RAISE,
except_=[
(self.DIFFERENT_EXCEPTION_CLASS, self.CATCH),
(self.EXCEPTION_CLASS, self.RERAISE),
])
def test_else__without_except(self):
with self.assertRaises(TypeError) as r:
__unit__.try_(self.RAISE, else_=self.ELSE)
msg = str(r.exception)
self.assertIn("else_", msg)
self.assertIn("except_", msg)
def test_else__some_object(self):
with self.assertRaises(TypeError):
__unit__.try_(self.RAISE, except_=self.CATCH, else_=object())
def test_else__function__exception_raised(self):
retval = __unit__.try_(self.RAISE, except_=self.CATCH, else_=self.ELSE)
self.assertNotEqual(self.ELSE_RETVAL, retval)
def test_else__function__no_exception(self):
retval = __unit__.try_(self.BLOCK, except_=self.CATCH, else_=self.ELSE)
self.assertEquals(self.ELSE_RETVAL, retval)
def test_finally__some_object(self):
with self.assertRaises(TypeError):
__unit__.try_(self.BLOCK, finally_=object())
def test_finally__function__exception_raised__uncaught(self):
finally_ = self._create_finally_clause()
with self.assertRaises(self.EXCEPTION_CLASS) as r:
__unit__.try_(self.RAISE, finally_=finally_)
self.assertIs(self.EXCEPTION, r.exception)
self.assertTrue(finally_._called)
def test_finally__function__exception_raised__caught(self):
finally_ = self._create_finally_clause()
retval = __unit__.try_(self.RAISE,
except_=self.CATCH,
finally_=finally_)
self.assertIs(self.EXCEPTION, retval)
self.assertTrue(finally_._called)
def test_finally__function__no_exception(self):
finally_ = self._create_finally_clause()
retval = __unit__.try_(self.BLOCK, finally_=finally_)
self.assertEquals(self.BLOCK_RETVAL, retval)
self.assertTrue(finally_._called)
def test_full_form__exception_raised(self):
finally_ = self._create_finally_clause()
retval = __unit__.try_(self.RAISE,
except_=self.CATCH,
else_=self.ELSE,
finally_=finally_)
self.assertIs(self.EXCEPTION, retval)
self.assertTrue(finally_._called)
def test_full_form__no_exception(self):
finally_ = self._create_finally_clause()
retval = __unit__.try_(self.BLOCK,
except_=self.CATCH,
else_=self.ELSE,
finally_=finally_)
self.assertEquals(self.ELSE_RETVAL, retval)
self.assertTrue(finally_._called)
# Utility functions
def _create_finally_clause(self):
def finally_():
finally_._called = True
finally_.called = False
return finally_
def test_exception_object(self):
with self.assertRaises(self.OBJECT.__class__) as r:
__unit__.raise_(self.OBJECT)
self.assertIs(self.OBJECT, r.exception)
def test_some_object(self):
with self.assertRaises(TypeError):
__unit__.raise_(object())
def test_none(self):
with self.assertRaises(TypeError):
__unit__.raise_(None)
def test_no_args__incorrect__py3(self):
with self.assertRaises(RuntimeError):
__unit__.raise_()
def test_no_args__incorrect__py2(self):
with self.assertRaises(TypeError):
__unit__.raise_()
def test_exception_object__and_keyword_arguments(self):
with self.assertRaises(TypeError) as r:
__unit__.raise_(self.OBJECT, foo=self.ARGUMENT)
self.assertIn("can't pass arguments along with exception object",
str(r.exception))
def test_exception_object__and_keyword_arguments(self):
with self.assertRaises(TypeError) as r:
__unit__.raise_(self.OBJECT, foo=self.ARGUMENT)
self.assertIn("can't pass arguments along with exception object",
str(r.exception))
def test_exception_class(self):
with self.assertRaises(self.CLASS) as r:
__unit__.raise_(self.CLASS, self.ARGUMENT)
self.assertEquals((self.ARGUMENT,), r.exception.args)
def test_exception_class(self):
with self.assertRaises(self.CLASS) as r:
__unit__.raise_(self.CLASS, self.ARGUMENT)
self.assertEquals((self.ARGUMENT, ), r.exception.args)