如何利用pytest-mysql为Python的SQL查询构建无生产数据干扰的单元测试

结论 #

完全可以实现，且绝对不会干扰或访问生产数据。pytest-mysql的核心机制就是启动一个独立的临时MySQL实例，所有测试操作都在这个临时实例里进行，测试结束后自动销毁，和生产环境完全隔离。

具体实现步骤 #

1. 安装依赖 #

先安装必要的包：

pip install pytest-mysql mysqlclient

如果Ubuntu上安装mysqlclient失败，先装系统依赖：

sudo apt-get install libmysqlclient-dev

2. 基础测试示例 #

直接使用pytest-mysql提供的内置fixture，就能快速编写测试：

假设你的应用有一个查询用户总数的逻辑，测试代码如下：

def test_user_count_query(mysql):
    # 1. 创建临时测试表
    mysql.execute("""
        CREATE TABLE users (
            id INT PRIMARY KEY AUTO_INCREMENT,
            username VARCHAR(50) NOT NULL
        )
    """)
    
    # 2. 插入模拟测试数据
    test_data = [("alice",), ("bob",), ("charlie",)]
    mysql.executemany("INSERT INTO users (username) VALUES (%s)", test_data)
    mysql.connection.commit()  # 显式提交数据
    
    # 3. 调用你的应用查询逻辑（这里模拟实现）
    def get_user_count(db_cursor):
        db_cursor.execute("SELECT COUNT(*) FROM users")
        return db_cursor.fetchone()[0]
    
    # 4. 断言结果符合预期
    assert get_user_count(mysql) == 3

3. 关键说明 #

• mysql fixture：自动创建临时数据库、返回游标对象，测试结束后自动销毁整个临时实例，数据不会残留。
• 若需自定义临时实例配置（比如端口、字符集），可在conftest.py中扩展：

import pytest
from pytest_mysql import factories

# 自定义临时MySQL进程，指定端口和字符集
custom_mysql_proc = factories.mysql_proc(port=3307, mysqld_params=["--character-set-server=utf8mb4"])
# 绑定自定义进程到连接fixture
custom_mysql = factories.mysql("custom_mysql_proc")

之后测试函数即可用custom_mysql代替默认的mysql fixture。

4. 注意事项 #

• 无需手动启动本地MySQL服务，pytest-mysql会自动管理临时进程的启停。
• 如果你的应用使用ORM（如SQLAlchemy），可通过mysql fixture获取的连接对象创建ORM会话，完成测试。

内容的提问来源于stack exchange，提问作者André Lourenço