pytest用yield实现前后置操作

本文深入解析了 pytest 中实现测试前后置操作的两种核心方式：传统命名约定的 setup_method/setup_class 与更强大灵活的 yield fixture 机制，重点强调后者如何通过“yield 前为 setup、yield 后为 teardown”的三段式流程，提供可靠、可作用域控制（function/class/session）、自动异常兜底的资源管理能力；同时直击实践痛点，澄清 autouse 与 scope 组合的常见误区，并提醒开发者注意 os._exit()、SIGKILL 等极端场景下 teardown 的执行边界，助你写出健壮、可维护、真正安全的测试代码。

pytest 中 setup_method 和 setup_class 怎么用

pytest 本身不叫 setup，但行为上完全能替代 unittest 的 setUp 和 setUpClass。直接在测试类里定义 setup_method（每条测试前调用）或 setup_class（整个类首次运行前调用）就行，pytest 会自动识别并执行。

注意：这两个方法名必须拼写准确，且不能带参数（除了 self），否则 pytest 会忽略它们。

• setup_method 接收一个参数 method，可用于判断当前运行的是哪个测试函数（比如跳过某些耗时操作）
• setup_class 只在类内第一个测试开始前执行一次，适合初始化数据库连接、启动 mock server 等重操作
• 它们没有返回值要求，也不支持 yield；想做后置清理，得换别的机制

用 yield fixture 实现 setup + teardown 组合

真正灵活、推荐的方式是用带 yield 的 fixture —— 它天然支持“先 setup，再运行测试，最后 teardown”三段式流程。pytest 在 yield 前的部分当 setup，在 yield 后的部分当 teardown。

示例：

@pytest.fixture
def db_connection():
    conn = create_test_db()  # setup 阶段
    yield conn                # 测试用这个 conn
    conn.close()              # teardown 阶段，一定会执行（哪怕测试崩溃）

• 这个 fixture 默认作用域是 function，即每个测试函数独享一份 setup/teardown
• 加 scope="class" 就等效于 setUpClass + tearDownClass
• yield 后的代码即使测试抛异常也会执行，比 try/finally 更可靠

fixture scope 和 autouse 的坑

很多人误以为加了 autouse=True 就等于全局 setup，结果发现 teardown 没触发、或者被多次执行。关键点在于 scope 和 autouse 的组合逻辑：

• autouse=True + scope="function"：每个测试前 setup，测试后 teardown，安全可控
• autouse=True + scope="session"：整个 pytest 运行只 setup 一次，但 teardown 只在最后执行一次 —— 如果中间某个测试崩溃导致进程退出，teardown 可能根本不会跑
• 多个 autouse fixture 之间无执行顺序保证，别依赖它们的先后关系做资源依赖

更稳妥的做法是显式声明依赖，比如让 db_connection fixture 依赖 tmpdir，pytest 会自动按需调用。

yield fixture 被中断时的 teardown 行为

yield fixture 的 teardown 部分不是“绝对保险”的。以下情况它可能不执行：

• 测试进程中被 os._exit() 强制终止（绕过 Python 正常退出流程）
• pytest 进程被 SIGKILL（kill -9）杀死
• 测试中调用了 sys.exit() 且没被 pytest 的异常捕获机制兜住（少见，但存在）

所以，涉及文件写入、端口绑定、临时进程启动这类操作，建议在 teardown 里加日志或状态检查，而不是假设“只要 yield 写了就一定执行”。

今天关于《pytest用yield实现前后置操作》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！