14.3 在单元测试中测试异常情况

    对于异常的测试可使用 方法。例如,如果你想测试某个函数抛出了 ValueError 异常,像下面这样写:

    如果你想测试异常的具体值,需要用到另外一种方法:

    1. import errno
    2.  
    3. class TestIO(unittest.TestCase):
    4. def test_file_not_found(self):
    5. try:
    6. except IOError as e:
    7. self.assertEqual(e.errno, errno.ENOENT)
    8.  
    9. else:
    10. self.fail('IOError not raised')

    这种方法的问题在于它很容易遗漏其他情况,比如没有任何异常抛出的时候。那么你还得需要增加另外的检测过程,如下面这样:

    1. class TestConversion(unittest.TestCase):
    2. def test_bad_int(self):
    3. try:
    4. except ValueError as e:
    5. self.assertEqual(type(e), ValueError)
    6. else:
    7. self.fail('ValueError not raised')

    assertRaises() 方法会处理所有细节,因此你应该使用它。

    assertRaisesRegex()还有一个容易忽略的地方就是它们还能被当做上下文管理器使用:

    1. class TestConversion(unittest.TestCase):
    2. def test_bad_int(self):
    3. with self.assertRaisesRegex(ValueError, 'invalid literal .*'):
    4. r = parse_int('N/A')

    但你的测试涉及到多个执行步骤的时候这种方法就很有用了。