Pytest-Bdd-Playwright 系列教程（8）：pytest的高级代码生成功能

前言

• 在自动化测试开发中，代码的编写和维护往往占据大量时间。特别是对于复杂的业务流程，测试用例多且逻辑复杂，手动编写测试代码效率低且容易出错；
• 在这个背景下，我们可以使用pytest的的高级代码生成工具，通过该工具，我们能够自动生成缺失的测试代码，减少重复劳动并提高效率；
• 本文将以购物车功能为例，介绍如何使用pytest的代码生成工具自动生成缺失的代码，帮助小伙伴们快速上手pytest的高级代码生成功能。

一、pytest高级代码生成功能概述

pytest中有一个智能的代码生成功能，支持自动扫描已有的测试和步骤定义。通过代码生成工具，我们可以自动生成缺失的测试步骤，避免手动编写。这一功能通过以下命令启用：

pytest --generate-missing --feature features/ tests/

参数说明：

• –generate-missing：启用 pytest 的生成缺失代码功能；
• –feature features/：指定存放特性文件（.feature 文件）的路径；
• tests/：指定存放功能测试代码的文件夹路径。

此命令会扫描指定的feature文件，生成缺失的测试代码。接下来我们将通过具体的购物车示例来演示如何使用这一功能。

二、编写购物车特性文件

首先，我们创建一个.feature文件，描述购物车的测试场景和步骤。将以下内容保存为features/cart.feature：

Feature: 购物车功能Scenario: 添加商品到购物车Given 购物车是空的When 我将名称为 "笔记本电脑" 且价格为 1000 的商品添加到购物车Then 购物车中应包含 1 件商品And 总价应为 1000Scenario: 从购物车移除商品Given 购物车中已包含名称为 "笔记本电脑" 且价格为 1000 的商品When 我将名称为 "笔记本电脑" 的商品从购物车移除Then 购物车应为空Scenario: 结算购物车中的商品Given 购物车中包含名称为 "笔记本电脑" 且价格为 1000 的商品And 购物车中还包含名称为 "鼠标" 且价格为 50 的商品When 我进行结算Then 总价应为 1050And 订单应被确认

场景说明：

1. 添加商品到购物车：用户将一件商品（例如“笔记本电脑”）添加到购物车，添加后购物车中应包含该商品，总价应更新。
2. 从购物车移除商品：用户将购物车中已有的商品移除，移除后购物车应为空。
3. 结算购物车中的商品：用户在购物车中有多件商品，结算后总价为各商品价格之和，且订单被成功确认。

三、使用pytest生成缺失的测试代码

在终端中运行以下命令，pytest将扫描cart.feature文件并自动生成测试代码框架：

pytest --generate-missing --feature features/cart.feature tests/

运行结果如下：

四、添加生成的代码到测试文件

创建tests/test_cart.py文件，并将生成的代码粘贴进去，然后对代码进行适当补充、修改：

"""购物车功能 feature tests."""from pytest_bdd import (given,scenario,then,when,
)@scenario('features\cart.feature', '从购物车移除商品')
def test_从购物车移除商品():"""从购物车移除商品."""@scenario('features\cart.feature', '添加商品到购物车')
def test_添加商品到购物车():"""添加商品到购物车."""@scenario('features\cart.feature', '结算购物车中的商品')
def test_结算购物车中的商品():"""结算购物车中的商品."""

五、运行测试并验证

完成测试代码后，运行以下命令验证购物车功能：

pytest tests/test_cart.py

若测试通过，则表示代码生成和自动化建议功能已成功应用。

总结

pytest的代码生成功能为复杂的业务流程测试提供了高效的支持，非常适合BDD风格的行为驱动测试。在项目中合理应用此工具，能够减少手动编写测试代码的工作量，提升测试开发效率。