单元测试是否可以断言某个方法调用了 sys.exit()？

Question

我有一个有时会调用的 Python 2.7 方法

sys.exit(1)

是否可以进行单元测试来验证在满足正确条件时调用了这行代码？

Answer 1

是的。 sys.exit引发SystemExit ，因此您可以使用assertRaises进行检查：

with self.assertRaises(SystemExit):
    your_method()

SystemExit实例有一个属性code设置为建议的退出状态，并且assertRaises返回的上下文管理器将捕获的异常实例作为exception ，因此检查退出状态很容易：

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

self.assertEqual(cm.exception.code, 1)

sys.exit 文档：

退出 Python。 这是通过引发SystemExit异常来实现的……可以在外部级别拦截退出尝试。

Answer 2

这是一个完整的工作示例。 尽管Pavel 的回答很好，但我花了一段时间才弄明白，所以我把它包括在这里，希望它会有所帮助。

import unittest
from glf.logtype.grinder.mapping_reader import MapReader

INCOMPLETE_MAPPING_FILE="test/data/incomplete.http.mapping"

class TestMapReader(unittest.TestCase):

    def test_get_tx_names_incomplete_mapping_file(self):
        map_reader = MapReader()
        with self.assertRaises(SystemExit) as cm:
            tx_names = map_reader.get_tx_names(INCOMPLETE_MAPPING_FILE)
        self.assertEqual(cm.exception.code, 1)

Answer 3

我在Python 单元测试文档搜索“异常测试”中找到了您的问题的答案。 使用您的示例，单元测试如下所示：

self.assertRaises(SystemExit, your_function, argument 1, argument 2)

请记住包含测试函数所需的所有参数。

Answer 4

作为对Pavel 出色回答的附加说明，您还可以检查特定状态（如果它们在您正在测试的函数中提供）。 例如，如果your_method()包含以下sys.exit("Error") ，则可以专门测试“错误”：

with self.assertRaises(SystemExit) as cm:
    your_method()
    self.assertEqual(cm.exception, "Error")