在 Python 3 中，使用 Pytest，我们如何测试 python 程序的退出代码:exit(1) 和 exit(

Question

我是python中Pytest的新手。

我面临着一个棘手的情况，我需要使用 Pytest 模块测试退出代码 - exit(1( 和 exit(0(。 下面是python程序：

def sample_script():
count_file  = 0
if count_file == 0:
print("The count of files is zero")
exit(1)
else:
print("File are present")
exit(0)

现在我想测试上述程序的退出代码，退出(1(和退出(0(。使用 Pytest 我们如何构建测试代码，以便我们可以测试或资产函数sample_script的退出代码？

请帮助我。

Answer 1

按照建议将exit(1)放入 if 块后，您可以测试 SystemExit 异常：

from some_package import sample_script

def test_exit():
with pytest.raises(SystemExit) as pytest_wrapped_e:
sample_script()
assert pytest_wrapped_e.type == SystemExit
assert pytest_wrapped_e.value.code == 42

示例取自此处：https://medium.com/python-pandemonium/testing-sys-exit-with-pytest-10c6e5f7726f

更新：

这是一个完整的工作示例，您可以复制/粘贴以进行测试：

import pytest
def sample_func():
exit(1)
def test_exit():
with pytest.raises(SystemExit) as e:
sample_func()
assert e.type == SystemExit
assert e.value.code == 1
if __name__ == '__main__':
test_exit()