如何在RSpec中临时扩展ActiveRecord连接类以测试辅助脚本？

首先得先修正你的补丁模块一个小问题：你现在用的是included钩子，但当你给实例（比如ActiveRecord::Base.connection）调用extend时，included不会触发——得换成extended钩子（或者同时兼容两种场景），这样alias :old_execute :execute才能生效：

module App::SomePatch
  # 给实例extend时触发
  def self.extended(mod)
    mod.send(:alias_method, :old_execute, :execute)
  end

  # 兼容给类include的场景（如果需要的话）
  def self.included(mod)
    mod.send(:alias_method, :old_execute, :execute)
  end

  def execute(*args)
    # 你的补丁逻辑
    # 比如可以调用old_execute(*args)来保留原行为
  end

  def some_storage
    @some_storage ||= []
  end
end

接下来是RSpec里的临时补丁实现，分两种常用方案：

方案一：手动备份+恢复（简单直接） #

核心思路是在测试前备份原方法，打完补丁测试后，把实例的方法和状态还原回去，避免影响其他测试：

RSpec.describe "入口点B的补丁行为" do
  let(:db_connection) { ActiveRecord::Base.connection }
  # 备份原execute方法的引用
  let(:original_execute) { db_connection.method(:execute) }

  before do
    # 给当前连接实例打补丁
    db_connection.extend(App::SomePatch)
  end

  after do
    # 1. 恢复execute方法为原始版本
    db_connection.define_singleton_method(:execute, original_execute)
    # 2. 移除补丁新增的实例方法
    db_connection.singleton_class.send(:undef_method, :some_storage)
    # 3. 清理补丁留下的实例变量
    db_connection.instance_variable_set(:@some_storage, nil)
  end

  it "修改execute行为并记录到some_storage" do
    # 执行测试逻辑
    db_connection.execute("SELECT 1")
    
    # 验证补丁效果
    expect(db_connection.some_storage).not_to be_empty
  end
end

方案二：隔离测试连接（更安全） #

如果担心修改全局的ActiveRecord::Base.connection会影响其他测试，可以从连接池单独 checkout 一个连接来测试，用完再归还：

RSpec.describe "入口点B的隔离测试" do
  let(:test_connection) { ActiveRecord::Base.connection_pool.checkout }

  after do
    # 归还连接到池里
    ActiveRecord::Base.connection_pool.checkin(test_connection)
  end

  it "补丁仅作用于测试连接" do
    # 给独立的测试连接打补丁
    test_connection.extend(App::SomePatch)
    
    # 运行入口点B的逻辑（建议把入口点B改成可传入连接参数的方法）
    EntryPointB.run(connection: test_connection)
    
    # 验证补丁效果
    expect(test_connection.some_storage).to include("SELECT 1")
    # 验证全局连接不受影响
    expect(ActiveRecord::Base.connection).not_to respond_to(:some_storage)
  end
end

为什么不用单纯的mock？ #

你提到方法别名让mock变复杂——确实，因为你的补丁是通过alias替换原方法，单纯用allow(connection).to receive(:execute)会和补丁的alias逻辑冲突。而上面的方案是直接操作实例的单例方法，能完美模拟真实场景下的猴子补丁行为。

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