python - 单元测试 : assert right SystemExit code

Question

我正在使用 unittest断言我的脚本引发了正确的 SystemExit 代码。

基于 http://docs.python.org/3.3/library/unittest.html#unittest.TestCase.assertRaises 中的示例

with self.assertRaises(SomeException) as cm:
    do_something()

the_exception = cm.exception
self.assertEqual(the_exception.error_code, 3)

我这样编码:

with self.assertRaises(SystemExit) as cm:
    do_something()

the_exception = cm.exception
self.assertEqual(the_exception.error_code, 3)

但是，这不起作用。出现以下错误:

AttributeError: 'SystemExit' object has no attribute 'error_code'

Answer 1

SystemExit直接派生自 BaseException 而不是 StandardError，因此它没有属性 error_code。

您必须使用属性 code 而不是 error_code。该示例如下所示:

with self.assertRaises(SystemExit) as cm:
    do_something()

the_exception = cm.exception
self.assertEqual(the_exception.code, 3)