如何使用pytest搭配Docker化PostgreSQL数据库测试且不影响生产库？

以下是几个实用且高效的方案，彻底避免测试套件干扰生产数据库：

1. 单次测试专属临时容器 #

每次测试前启动一个全新的数据库容器，测试结束后自动销毁，完全隔离测试数据：

• 启动命令示例（以PostgreSQL为例）：

  docker run --rm -d -p 5432:5432 -e POSTGRES_PASSWORD=test -e POSTGRES_DB=test_db postgres:15
  

  其中--rm参数会在容器停止后自动删除，确保无残留数据。
• 配合测试脚本，在测试开始前启动容器，测试完成后停止容器（或让容器随测试结束自动停止），全程无需手动清理。

2. Docker Compose编排完整测试环境 #

用单独的Compose文件定义测试专属的数据库和测试服务，一键搭建/销毁隔离环境：

• 创建docker-compose.test.yml：

  version: '3.8'
  services:
    test-db:
      image: postgres:15
      environment:
        POSTGRES_PASSWORD: test
        POSTGRES_DB: test_db
      # 无需映射到主机端口，测试服务可通过容器名直接访问
    test-suite:
      build: .
      command: npm run test # 替换为你的测试命令
      depends_on:
        - test-db
      environment:
        DB_CONN_STRING: postgresql://postgres:test@test-db:5432/test_db
  

• 运行测试：

  docker-compose -f docker-compose.test.yml up --build --abort-on-container-exit
  

• 测试完成后清理所有资源：

  docker-compose -f docker-compose.test.yml down -v
  

  -v参数会删除数据库卷，确保下次测试是干净环境。

3. 复用容器+预定义初始化脚本 #

如果启动新容器耗时较长，可以复用测试容器，但每次测试前强制重置数据库状态：

• 在测试套件的前置步骤中执行初始化脚本，比如：

  psql -h test-db-host -U postgres -d test_db -f ./test-reset.sql
  

  test-reset.sql包含清理旧数据、重建表结构、插入基准测试数据的语句。
• 也可以在数据库容器的entrypoint中配置自动执行初始化脚本，确保每次容器启动时都是干净的测试环境。

4. 网络层面彻底隔离 #

通过Docker网络配置，让测试容器完全无法访问生产数据库所在的网络：

• 创建测试专用网络：

  docker network create test-isolate-net
  

• 启动测试相关容器时指定该网络：

  docker run --rm -d --network test-isolate-net postgres:15
  

• 生产数据库放在另一个独立网络，测试网络不与生产网络连通，从根源杜绝误访问。

内容的提问来源于stack exchange，提问作者Hossein Gholami