AI编写自动化测试脚本

编写pytest + selenium的接口自动化脚本格式 参考资料：https://www.byhy.net/auto/pyatframework/pytest-01/ 接口脚本模版： 在这里插入图片描述

UI测试用例模版

快速生成自动化测试用例的实践指南

一、环境准备

1. 安装依赖库

   pip install pytest requests selenium webdriver-manager allure-pytest

2. Allure 环境配置

   • 下载
   • 添加 Allure 的 bin 目录到系统环境变量。

二、项目结构规划

project/
├── api/                  # 接口封装模块
│   ├── config.py         # 环境配置
│   └── urls.py          # 接口 URL 管理
├── tests/                # 测试用例模块
│   ├── conftest.py       # 全局 Fixture
│   ├── test_api/         # API 测试
│   └── test_ui/          # UI 测试
├── requirements.txt      # 依赖库清单
└── pytest.ini            # Pytest 配置

三、快速生成用例的步骤

1. 解析接口文档，封装接口

• 根据接口文档提取关键信息 ：

  • 请求方法（GET/POST/PUT/DELETE）
  • URL 路径（如 /api/login ）
  • 请求参数（必填项、类型、取值范围）
  • 响应结构（状态码、字段定义）

• 封装到 api/urls.py ：

  # api/urls.py
  from .config import EnvConfig
  
  class MerchantUrls:
      @property
      def login(self):
          return f"{EnvConfig.BASE_URL}/api/login"
  
      def get_user_info(self, user_id):
          return f"{EnvConfig.BASE_URL}/users/{user_id}"

2. 编写数据工厂（Data Factory）

• 动态生成测试数据 ：

  # tests/data_factory.py
  from faker import Faker
  
  fake = Faker('zh_CN')
  
  class TestData:
      @staticmethod
      def generate_user():
          return {
              "username": fake.user_name(),
              "phone": fake.phone_number(),
              "password": fake.password()
          }

• 参数化测试用例 ：

  # tests/test_api/test_login.py
  import pytest
  @pytest.mark.parametrize("username, password", [
      ("admin", "valid_password"),
      ("", "invalid_password"),  # 空用户名
      ("admin", "")              # 空密码
  ])
  def test_login(username, password):
      # 调用接口并断言

3. 实现基础 Fixture

• 全局 Token 管理 ：

  # tests/conftest.py
  @pytest.fixture(scope="session")
  def auth_token():
      response = requests.post(MerchantUrls().login, json=VALID_CREDENTIALS)
      return response.json()["token"]

• 浏览器驱动管理（UI 测试） ：

  # tests/conftest.py
  @pytest.fixture
  def browser():
      driver = webdriver.Chrome()
      yield driver
      driver.quit()

4. 编写测试用例模板

• API 测试用例模板 ：

  # tests/test_api/test_user.py
  def test_get_user_info(auth_token):
      headers = {"Authorization": f"Bearer {auth_token}"}
      response = requests.get(MerchantUrls().get_user_info(1), headers=headers)
      assert response.status_code == 200
      assert "id" in response.json()

• UI 测试用例模板（POM 模式） ：

  # tests/test_ui/pages/login_page.py
  class LoginPage:
      def login(self, username, password):
          self.driver.find_element(By.ID, "username").send_keys(username)
          self.driver.find_element(By.ID, "password").send_keys(password)
          self.driver.find_element(By.ID, "login-btn").click()
  
  # tests/test_ui/test_login.py
  def test_ui_login(browser):
      login_page = LoginPage(browser)
      login_page.login("admin", "password")
      assert "dashboard" in browser.current_url

5. 集成 Allure 报告

• 添加 Allure 注解 ：

  @allure.epic("用户管理")
  @allure.feature("登录功能")
  @allure.story("成功登录")
  def test_login_success():
      with allure.step("输入用户名密码"):
          # 操作步骤
      with allure.step("点击登录按钮"):
          # 操作步骤

• 生成并查看报告 ：

  pytest --alluredir=./reports   # 生成报告数据
  allure serve ./reports         # 启动本地报告服务

四、效率提升技巧

1. 使用模板引擎

   • 通过 Jinja2 或 Cookiecutter 生成基础代码，减少重复编写。

2. Excel/YAML 数据驱动

   # 从 Excel 读取测试数据
   import pandas as pd
   test_data = pd.read_excel("test_cases.xlsx").to_dict("records")
   
   @pytest.mark.parametrize("data", test_data)
   def test_api(data):
       requests.post(url, json=data)

3. 接口自动化工具集成

   • 使用 Postman 导出用例为 JSON ，通过脚本转换为 pytest 用例。

4. 代码片段复用

   • 将常用操作封装为函数（如数据库清理、文件上传）。

五、常见问题解决

1. 依赖接口的测试用例

   • 方案 ：使用 pytest-dependency 标记用例依赖关系。
   @pytest.mark.dependency(name="login")
   def test_login():
       pass
   
   @pytest.mark.dependency(depends=["login"])
   def test_user_info():
       pass

2. 环境切换问题

   • 方案 ：通过环境变量动态加载配置。
   ENV=prod pytest tests/  # 运行生产环境测试

3. 测试数据清理

   • 方案 ：在 fixture 中实现后置清理逻辑。
   @pytest.fixture
   def temp_user(api_client):
       user_id = api_client.create_user()
       yield user_id
       api_client.delete_user(user_id)

六、总结

通过以下步骤可快速生成自动化测试用例：

1. 解析文档 → 2. 封装接口 → 3. 数据驱动 → 4. 模板化用例 → 5. 集成报告

   关键点 ：

• 模块化设计（API/UI 分离）
• 数据工厂动态生成参数
• 利用 Allure 增强报告可读性
• 使用 Fixture 管理依赖和资源

扩展方向 ：

• 集成 CI/CD（如 Jenkins、GitHub Actions）
• 添加性能测试（Locust）
• 实现智能断言（JSON Schema 校验）