pytest如何终止当前测试用例且不影响其余用例运行

问题说明 #

使用pytest测试包含sys.exit(non_zero_code)分支的代码时，直接mock整个sys模块会导致sys.exit调用后的代码继续执行；调用pytest.exit又会终止整个测试进程，无法满足仅终止当前用例、不影响后续用例运行的需求。

复现代码 #

import os
import sys

import pytest
import logging
from pathlib import Path

logging.basicConfig(filename=os.path.join(Path(__file__).parent.resolve(), "pytest_log"),
                    filemode='a',
                    format='%(asctime)s,%(msecs)d %(name)s %(levelname)s %(message)s',
                    datefmt='%H:%M:%S',
                    level=logging.INFO)
logger = logging.getLogger(__name__)


def function_with_exit(value):

    if value == 2:
        sys.exit(2)
        logger.info("Should not execute, we exited")

    logger.info("Value is not 2")


def side_effect():
    logger.info("Sys exit was called")


@pytest.mark.parametrize("value", [1, 2, 3])
def test_with_pytest(mocker, value):
    mocker.patch(__name__ + '.' + sys.__name__)

    function_with_exit(value)

运行上述代码时，value=2的用例会错误触发"Should not execute, we exited"日志输出。

实现目标 #

• 运行value=2的测试用例时，sys.exit调用后的代码不会被执行
• 当前测试用例终止后，后续其余测试用例可以继续正常运行

根因分析 #

原生sys.exit的执行逻辑是抛出SystemExit异常，未被捕获时会终止进程。直接mock整个sys模块后，sys.exit被替换为无任何逻辑的MagicMock对象，调用时不会抛出异常，因此后续代码会继续执行。
pytest框架本身会自动捕获单个测试用例内抛出的SystemExit异常，仅结束当前用例执行，不会中断整体测试流程，不需要额外调用pytest.exit。

修正方案 #

不要mock整个sys模块，仅对sys.exit单独打桩，配置其被调用时抛出SystemExit异常，对齐原生行为即可，同时可配合断言校验退出码是否符合预期。
修正后的测试代码如下：

@pytest.mark.parametrize("value", [1, 2, 3])
def test_with_pytest(mocker, value):
    # 仅mock sys.exit，调用时抛出SystemExit，和原生行为一致
    mock_exit = mocker.patch("sys.exit", side_effect=SystemExit)

    if value == 2:
        # 捕获抛出的SystemExit异常，避免用例标记为异常失败
        with pytest.raises(SystemExit):
            function_with_exit(value)
        # 校验sys.exit被调用时传入了正确的退出码
        mock_exit.assert_called_once_with(2)
    else:
        function_with_exit(value)
        mock_exit.assert_not_called()

效果说明 #

• sys.exit被调用时会立即抛出SystemExit异常，调用点之后的代码完全不会执行，不会出现错误日志
• pytest自动捕获单测内的SystemExit异常，value=2用例执行结束后，value=1、value=3的用例会正常运行，不会终止整个测试进程
• 可直接通过mock对象断言sys.exit的调用情况、入参是否符合预期，满足测试验证需求

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