Unittest：断言正确的 SystemExit 代码

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)