pytest和数据库交互的场景设计与实现

在测试框架中，与数据库的交互是一种常见需求，特别是在接口测试中，可能需要验证接口操作是否正确反映到数据库中，或者从数据库中查询数据，作为接口请求的参数。

以下是一个结合 pytest 测试框架、MySQL 数据库（兼容 OB-MySQL）、以及之前的 YML 用例测试框架 的完整解决方案。

1. 需求分析

1.1 数据库交互场景

• 前置数据准备：
  • 在接口测试之前，需要向数据库插入测试数据。
• 接口参数化：
  • 从数据库中查询特定字段值，动态注入到接口请求参数中。
• 后置验证：
  • 测试接口调用后，验证数据库中的数据是否符合期望。

1.2 设计目标

• 集成 MySQL 数据库（兼容 OB-MySQL）。
• 支持从数据库读取字段作为接口参数。
• 支持在用例中定义数据库操作（如查询语句、验证规则等）。
• 保持与之前的 YML 用例框架兼容。

2. 数据库交互设计

使用 PyMySQL 库与 MySQL 数据库进行交互。

2.1 数据库工具类

实现一个数据库操作工具类，用于执行 SQL 查询、插入数据和验证数据。

import pymysql

class MySQLHandler:
    def __init__(self, host, port, user, password, database):
        self.connection = pymysql.connect(
            host=host,
            port=port,
            user=user,
            password=password,
            database=database,
            cursorclass=pymysql.cursors.DictCursor
        )

    def execute_query(self, query, params=None):
        """
        执行查询语句
        :param query: SQL 查询语句
        :param params: 查询参数
        :return: 查询结果
        """
        with self.connection.cursor() as cursor:
            cursor.execute(query, params)
            result = cursor.fetchall()
        return result

    def execute_update(self, query, params=None):
        """
        执行插入或更新语句
        :param query: SQL 更新语句
        :param params: 更新参数
        """
        with self.connection.cursor() as cursor:
            cursor.execute(query, params)
        self.connection.commit()

    def close(self):
        """关闭数据库连接"""
        self.connection.close()

3. YML 用例扩展

在 YML 用例中增加对数据库操作的支持，包括：

1. 动态参数注入：通过数据库查询结果动态填充接口请求参数。
2. 数据库验证：通过 SQL 查询验证接口调用后的数据状态。

3.1 用例格式示例

扩展后的 YML 用例示例：

test_cases:
  - name: "用户登录"
    request:
      method: POST
      url: "http://example.com/api/login"
      headers:
        Content-Type: "application/json"
      body:
        username: