Pytest中如何mock主函数输入作为所有测试用例基准

问题相关参考 #

• 运行Pytest触发的报错：
  
• 待测试的main函数代码：
  

核心踩坑点（之前各类教程方案失效的根本原因） #

90%的input mock失效都是因为mock路径错误：网上很多教程教你mockbuiltins.input，这个写法只能拦截测试文件里直接调用的input，你自己业务代码文件里的input调用根本拦不住，运行时还是会等真实终端输入，要么卡超时要么抛EOF报错。
要实现全测试用例统一的输入基准，不用每个测试用例重复写mock逻辑，用Pytest自带的fixture做全局自动加载即可。

可直接复用的实现步骤 #

1. 在你的测试文件夹根目录新建conftest.py文件（Pytest会自动识别这个文件里的全局配置，不用手动在测试文件里导入）
2. 把下面的代码粘进去，替换成你自己的模块名和测试输入值：

import pytest
from unittest import mock
# 替换成你自己写main函数的文件名，比如main函数在main.py就写import main
import your_module_name

@pytest.fixture(autouse=True)
def unified_input_mock():
    # 按你main函数里调用input的顺序，把所有测试输入值放这里，全用字符串格式
    test_input_seq = [
        "输入值1", # 对应main里第一次调用input的返回
        "输入值2", # 对应第二次input
        "3", # 比如需要输入数字，就写字符串形式的数字，和真实手动输入格式一致
        "N" # 对应后续的确认类输入
    ]
    # 关键：mock的是你自己业务模块下的input，不是全局builtins.input
    with mock.patch.object(your_module_name, "input", side_effect=test_input_seq):
        # 提前执行一次main，拿到执行结果，所有测试用例共享这个执行上下文
        main_exec_result = your_module_name.main()
        yield main_exec_result

3. 正常写测试用例即可，所有依赖main输入的函数测试，都会自动用上面配置的统一mock输入，不用每个用例单独写mock：

# test_your_code.py 示例测试文件
from your_module_name import calculate_result, check_user_permission

def test_calculate_result():
    # 直接调用依赖main输入的函数即可，已经自动加载mock
    assert calculate_result() == 90 # 替换成对应测试输入下的预期值

def test_check_permission():
    assert check_user_permission() is True

常见报错排查 #

• 报StopIteration：你给的测试输入值数量比main里实际调用input的次数少，补全对应顺序的输入值即可
• 还是卡着等输入/报EOF错误：检查你mock的模块名对不对，必须是main函数所在的业务模块，不要mock builtins.input
• 部分测试用例没用到mock：检查conftest.py是不是放在运行pytest命令的根目录下，fixture的autouse=True有没有加上
• 拿到的输入值不对：检查测试输入的顺序是不是和main里调用input的顺序完全一致，不要直接写数字/布尔值，全部转成字符串格式

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