如何在不重建只读第三方数据库的情况下运行Rails 7测试/RSpec

核心疑问解答 #

Rails 默认只会自动重建主数据库的测试 schema（执行迁移、维护结构），你通过ExtraDbBase这类抽象类连接的次级第三方数据库，除非主动配置迁移或测试任务，否则 Rails 不会尝试重建它们的 schema。

仅注释spec/rails_helper.rb里的ActiveRecord::Migration.maintain_test_schema!并不够——这个方法只负责主数据库的 schema 维护，注释它会让主数据库也跳过重建，但你需要保留主数据库的重建能力。

满足需求的原生配置方案 #

针对你列出的7个要求，用 Rails 原生特性就能搞定，不用折腾老旧的猴子补丁或绕开 Rake 的方案：

1. 仅重建主数据库 #

在config/database.yml的test环境配置中，只为主数据库保留迁移相关配置，次级数据库明确禁用迁移：

# config/database.yml
default: &default
  adapter: postgresql
  encoding: unicode
  pool: <%= ENV.fetch("RAILS_MAX_THREADS") { 5 } %>

development:
  primary:
    <<: *default
    database: my_app_development
  vendor_db_one:
    <<: *default
    database: vendor_db_one_development

test:
  primary:
    <<: *default
    database: my_app_test
    # 主数据库保留迁移配置
    migrations_paths: db/migrate
  vendor_db_one:
    <<: *default
    database: vendor_db_one_test
    # 禁用次级数据库的迁移和schema维护
    migrations_paths: []
    schema_dump: false

然后在spec/rails_helper.rb中，指定仅维护主数据库的测试 schema：

# spec/rails_helper.rb
ActiveRecord::Migration.maintain_test_schema!(only: :primary)

2. 禁止向次级数据库写入数据 #

在ExtraDbBase抽象类中强制开启只读模式，确保测试时不会意外写入：

class ExtraDbBase < ApplicationRecord
  self.abstract_class = true
  establish_connection CONST_SET_BY_INITIALIZER_FROM_YAML

  # 全局禁用写入操作
  before_save { raise ActiveRecord::ReadOnlyRecord }
  before_destroy { raise ActiveRecord::ReadOnlyRecord }
  before_update { raise ActiveRecord::ReadOnlyRecord }

  # 或者仅针对测试环境设置
  if Rails.env.test?
    def readonly?
      true
    end
  end
end

3. 无需引用所有数据库也能启动测试 #

如果某些供应商数据库在测试环境无法访问，可在config/environments/test.rb中动态跳过这些连接的初始化：

# config/environments/test.rb
# 仅在测试环境跳过特定次级数据库的连接
unless ENV["ENABLE_VENDOR_DB"]
  Rails.application.config.after_initialize do
    Object.send(:remove_const, :ExtraDbBase) if defined?(ExtraDbBase)
  end
end

测试时通过环境变量控制是否加载这些数据库，不加载也能正常运行核心功能测试。

4. 复用供应商现有数据做测试 #

直接连接供应商的非生产环境数据库（需确认只读权限），或者导出静态数据快照作为测试数据源：

# spec/support/vendor_db_test_helper.rb
module VendorDbTestHelper
  def vendor_db_connection
    @vendor_db_connection ||= ActiveRecord::Base.establish_connection(
      adapter: "postgresql",
      database: "vendor_uat_db",
      host: "vendor-uat.example.com",
      username: "readonly_user",
      password: "readonly_pass"
    ).connection
  end
end

RSpec.configure do |config|
  config.include VendorDbTestHelper
end

5. 适配不同RDBMS #

如果测试环境无法使用供应商同款数据库，可通过两种方式兼容：

• 用SQLite模拟：在测试环境为次级数据库模型创建匹配字段结构的视图
• 用RSpec Stub：直接 stub 模型的查询方法，返回预设模拟数据

# 示例：Stub供应商模型的查询方法
it "generates report from vendor data" do
  allow(VendorModel).to receive(:where).and_return([
    OpenStruct.new(id: 1, name: "Test Vendor", value: 100)
  ])
  # 执行测试断言
end

6. 处理数据库特定DDL #

针对主数据库的原生DDL，放在db/migrate迁移文件中用execute执行：

# db/migrate/20240520123456_add_postgresql_index.rb
class AddPostgresqlIndex < ActiveRecord::Migration[7.0]
  def up
    execute "CREATE INDEX CONCURRENTLY idx_users_email ON users(email);"
  end

  def down
    execute "DROP INDEX idx_users_email;"
  end
end

Rails会正常执行这些原生DDL，只要主数据库支持即可。

7. 原生SQL与数据库特定特性的测试 #

直接在测试中执行原生SQL查询，验证结果是否符合预期：

it "uses postgresql JSONB query correctly" do
  result = VendorModel.connection.execute(
    "SELECT data->>'key' FROM vendor_items WHERE id = 1;"
  )
  expect(result.first["?column?"]).to eq("expected_value")
end

总结 #

通过以上配置，完全可以用Rails原生特性满足所有测试需求，核心思路是：明确区分主数据库和次级数据库的测试配置，对只读数据库做严格写入限制，灵活选择模拟或复用现有数据的方式降低测试成本。

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