python - 使用 `unittest` 验证是否调用了 `exit` 函数

Question

我定义了一个非常简单的配置管理器（解析配置文件并验证其中是否存在某些键），现在我正在为它编写一些测试。

在配置文件无效的许多情况下，我希望配置处理程序调用exit(). 有什么方法可以编写测试以确保exit被调用并继续测试？我是否可以仅针对测试环境“模拟”该exit功能？

我正在使用 Python 2.7 和unittest.

Answer 1

sys.exit()引发SystemExit异常，应该由 unittest 框架捕获。因此，您可以检查SystemExit为这些测试用例引发的问题。

例如，exit_test.py：

import sys
import unittest


def func():
    sys.exit()


class MyTest(unittest.TestCase):

    def test_func(self):
        self.assertRaises(SystemExit, func)


if __name__ == "__main__":
    unittest.main()

跑：

$ python exit_test.py 
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK