pytest capsys无法捕获含sys.exit代码的输出问题咨询

这个问题我之前也碰到过，核心原因是sys.exit()会直接触发Python进程终止，导致stdout的输出缓冲区可能还没来得及把内容同步到capsys监控的流中，同时pytest默认会把SystemExit异常判定为测试失败。下面给你两个可行的解决方案：

方案1：用pytest.raises捕获SystemExit，同时读取输出 #

不需要修改你的原代码，只需要调整测试逻辑，先捕获sys.exit()抛出的SystemExit异常，再用capsys读取输出。这样既可以验证退出码，又能拿到打印的提示信息。

假设你的密码校验代码是这样的：

# password_checker.py
import sys

def validate_password(password):
    if not password:
        print("密码不能为空，请重新输入")
        sys.exit(-1)
    # 其他校验逻辑...

对应的测试代码可以这么写：

# test_password.py
import pytest
from password_checker import validate_password

def test_empty_password(capsys):
    # 用pytest.raises捕获SystemExit异常
    with pytest.raises(SystemExit) as exit_info:
        validate_password("")
    
    # 读取捕获到的输出
    captured_output = capsys.readouterr()
    # 验证提示信息
    assert captured_output.out.strip() == "密码不能为空，请重新输入"
    # 验证退出码
    assert exit_info.value.code == -1

方案2：手动刷新stdout缓冲区（需要修改原代码） #

如果你不想在测试里处理异常，可以在print()之后手动刷新stdout，确保输出被立即写入流中，这样即使调用sys.exit()，capsys也能捕获到内容。

修改后的原代码：

# password_checker.py
import sys

def validate_password(password):
    if not password:
        print("密码不能为空，请重新输入")
        sys.stdout.flush()  # 手动刷新缓冲区
        sys.exit(-1)
    # 其他校验逻辑...

测试代码同样需要捕获SystemExit（否则pytest会认为测试失败）：

# test_password.py
import pytest
from password_checker import validate_password

def test_empty_password(capsys):
    with pytest.raises(SystemExit):
        validate_password("")
    
    captured_output = capsys.readouterr()
    assert captured_output.out.strip() == "密码不能为空，请重新输入"

为什么return能正常捕获？ #

当你用return代替sys.exit()时，只是结束当前函数的执行，整个测试进程还在运行，stdout的缓冲区会被正常刷新，capsys自然能捕获到所有输出。而sys.exit()是直接终止整个Python进程，缓冲区的内容可能还没来得及被处理就被中断了。

内容的提问来源于stack exchange，提问作者Suzanno Hogwarts