Rails多数据库测试：如何将MySQL库纳入PostgreSQL事务？

针对你的Rails 7.1 + RSpec场景，要让PostgreSQL和所有MySQL库的测试操作都被事务包裹并自动回滚，可以按以下步骤配置：

1. 扩展RSpec的事务管理逻辑 #

Rails默认只对primary数据库启用测试事务，你需要手动遍历所有连接池，为每个数据库开启事务并在测试结束后回滚。在spec/rails_helper.rb中添加：

RSpec.configure do |config|
  config.before(:each) do
    ActiveRecord::Base.connection_handler.connection_pools.each_value do |pool|
      next if pool.connected? && pool.current_transaction.open?
      # 开启不可连接的事务（避免嵌套问题）
      pool.connection.begin_transaction(joinable: false)
    end
  end

  config.after(:each) do
    ActiveRecord::Base.connection_handler.connection_pools.each_value do |pool|
      next unless pool.connected? && pool.current_transaction.open?
      pool.connection.rollback_transaction
      # 测试环境下断开连接，避免连接泄漏
      pool.disconnect!
    end
  end
end

2. 确保数据库配置正确 #

检查config/database.yml的test环境，所有MySQL库的配置需正确指向测试数据库，示例：

test:
  primary:
    adapter: postgresql
    database: main_app_test
    host: localhost
    # 其他PostgreSQL配置
  alpha:
    adapter: mysql2
    database: alpha_test
    host: localhost
    username: root
    password: ''
    # 其他MySQL配置
  beta:
    adapter: mysql2
    database: beta_test
    # ... 其他库配置

3. 更优雅的复用Rails内置逻辑 #

如果不想完全手动管理事务，可以重写ActiveRecord::TestFixtures的钩子，补充处理非primary数据库。在spec/support/multidb_transactions.rb中创建模块：

module MultiDbTransactionalTests
  def setup_fixtures
    super
    # 为所有非primary连接开启事务
    ActiveRecord::Base.connection_handler.connection_pools.except(:primary).each_value do |pool|
      pool.connection.begin_transaction(joinable: false) unless pool.current_transaction.open?
    end
  end

  def teardown_fixtures
    # 回滚所有非primary连接的事务
    ActiveRecord::Base.connection_handler.connection_pools.except(:primary).each_value do |pool|
      if pool.connected? && pool.current_transaction.open?
        pool.connection.rollback_transaction
        pool.disconnect!
      end
    end
    super
  end
end

# 将模块注入到需要的测试类型中
RSpec.configure do |config|
  config.include MultiDbTransactionalTests, type: :model
  config.include MultiDbTransactionalTests, type: :request
  config.include MultiDbTransactionalTests, type: :controller
  # 根据你的测试类型添加对应条目
end

4. 验证事务有效性 #

编写测试用例验证跨库操作是否会被回滚：

RSpec.describe "Cross-database transaction rollback", type: :model do
  it "rolls back changes in both PostgreSQL and MySQL" do
    # 创建PostgreSQL记录
    postgres_item = SomePostgresModel.create!(name: "test_postgres")
    # 创建MySQL记录
    mysql_item = Alpha::SomeModel.create!(value: "test_mysql")

    expect(SomePostgresModel.exists?(postgres_item.id)).to be true
    expect(Alpha::SomeModel.exists?(mysql_item.id)).to be true
  end

  after(:example) do
    # 验证测试后数据已回滚
    expect(SomePostgresModel.exists?(postgres_item.id)).to be false
    expect(Alpha::SomeModel.exists?(mysql_item.id)).to be false
  end
end

Rails源码参考位置 #

• 事务核心操作：ActiveRecord::ConnectionAdapters::DatabaseStatements（定义了begin_transaction、rollback_transaction等方法，路径：activerecord/lib/active_record/connection_adapters/abstract/database_statements.rb）
• 测试事务管理：ActiveRecord::TestFixtures（处理测试用例的事务和fixture重置，路径：activerecord/lib/active_record/test_fixtures.rb）
• 多连接池管理：ActiveRecord::ConnectionHandler（管理所有数据库连接池，路径：activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb）

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