Python测试MySQL插入函数：纯Python搭建测试库的实现疑问

方案1：用Docker容器临时启动MySQL #

直接通过Python代码调用Docker创建临时MySQL容器，完全模拟真实环境，语法100%兼容，测试后自动清理，无需本地预装MySQL：

1. 先安装依赖：pip install docker
2. 测试代码示例：

   import docker
   import mysql.connector
   import time
   from your_module import insert_function
   
   def test_insert():
       # 初始化Docker客户端
       client = docker.from_env()
       # 启动临时MySQL容器，设置密码和测试库
       container = client.containers.run(
           "mysql:8.0",
           environment={"MYSQL_ROOT_PASSWORD": "testpass", "MYSQL_DATABASE": "testdb"},
           ports={"3306/tcp": 33067},
           detach=True,
           remove=True  # 容器停止后自动删除
       )
       # 等待容器初始化完成
       time.sleep(5)
   
       try:
           # 连接临时数据库
           conn = mysql.connector.connect(
               host="localhost",
               port=33067,
               user="root",
               password="testpass",
               database="testdb"
           )
           # 创建和业务代码一致的表结构
           cursor = conn.cursor()
           cursor.execute("CREATE TABLE test_table (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), value INT)")
           conn.commit()
   
           # 执行待测试的插入函数
           test_data = [{"name": "foo", "value": 1}, {"name": "bar", "value": 2}]
           insert_function(test_data)
   
           # 验证数据库内容是否符合预期
           cursor.execute("SELECT name, value FROM test_table ORDER BY id")
           results = cursor.fetchall()
           assert results == [("foo", 1), ("bar", 2)], f"结果不符：预期{[('foo',1),('bar',2)]}，实际{results}"
       finally:
           # 清理资源
           container.stop()
           conn.close()
   

方案2：用testcontainers-python简化容器操作 #

这个库专门为测试场景封装了Docker操作，代码更简洁，无需手动处理容器启停、等待逻辑：

1. 安装依赖：pip install testcontainers[mysql]
2. 测试代码示例：

   from testcontainers.mysql import MySqlContainer
   import mysql.connector
   from your_module import insert_function
   
   def test_insert():
       with MySqlContainer("mysql:8.0") as container:
           # 直接获取容器的连接参数
           conn = mysql.connector.connect(**container.get_connection_params())
           cursor = conn.cursor()
           # 创建测试表
           cursor.execute("CREATE TABLE test_table (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), value INT)")
           conn.commit()
   
           # 执行测试函数
           test_data = [{"name": "foo", "value": 1}, {"name": "bar", "value": 2}]
           insert_function(test_data)
   
           # 验证结果
           cursor.execute("SELECT name, value FROM test_table ORDER BY id")
           assert cursor.fetchall() == [("foo", 1), ("bar", 2)]
   

注意事项 #

以上方案需要本地安装Docker，Docker是跨平台工具，安装后无需额外配置，属于测试环境的通用依赖，比手动搭建MySQL成本低得多。

内容的提问来源于stack exchange，提问作者J.N.