如何用Python和Pytest模拟Redshift查询进行集成测试

以下是几个适合替代Moto、满足你需求的工具和方法：

• PostgreSQL 本地实例
  Redshift基于PostgreSQL构建，绝大多数语法兼容。直接用本地PostgreSQL作为测试数据库：

  • 用psycopg2或sqlalchemy建立连接，创建测试表、插入测试数据
  • 针对Redshift特有语法（如COPY、UNLOAD），要么在测试中替换为PostgreSQL等效逻辑，要么用Mock拦截这类语句
  • 优点：无需额外模拟库，环境接近真实，支持完整的SQL执行和数据操作

• Redshift 本地模拟器（Docker）
  用Docker运行Redshift本地测试镜像，搭配亚马逊官方的redshift-connector连接：

  • 启动容器后，可直接执行Redshift原生SQL，创建表、插入数据、验证查询结果
  • 优点：完全兼容Redshift语法，模拟真实环境最准确

• pytest-redshift 插件
  针对pytest的专用插件，能简化测试流程：

  • 支持测试前自动创建临时表、注入测试数据，测试后自动清理
  • 可配合本地Redshift实例或模拟器使用，专注于SQL逻辑的正确性验证

• Python内置unittest.mock
  如果仅需验证SQL生成逻辑，无需完整执行：

  • 模拟de_core.db.redshift.get_connection()和util.execute_query方法
  • 捕获生成的SQL语句，断言其格式和内容符合预期
  • 优点：轻量快速，适合单元测试层面的SQL校验

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