如何用RSpec测试方法是否使用正确的数据库副本角色

核心思路 #

通过检查ActiveRecord::Base当前连接的数据库角色，结合RSpec的断言机制，验证方法是否在指定角色的数据库连接下执行。Rails 6.1+的多数据库特性会自动跟踪连接的角色标识，无需依赖实际的读写分离实例。

实现步骤 #

1. 封装角色检查辅助方法 #

先写一个简单的辅助方法，用于获取当前数据库连接的角色：

def current_db_role
  ActiveRecord::Base.connection.pool.db_config.role
end

2. 编写测试用例 #

针对MyService的两个方法分别编写测试：

RSpec.describe MyService do
  let(:service) { described_class.new }

  describe "#call_to_primary" do
    it "使用默认的writing角色（主库）" do
      service.call_to_primary
      # 若要验证该方法未使用读副本，可断言角色不等于:reading
      expect(current_db_role).not_to eq(:reading)
    end
  end

  describe "#call_to_reader" do
    it "切换到reading角色（读副本）执行查询" do
      # 方式1：直接断言执行后的角色
      service.call_to_reader
      expect(current_db_role).to eq(:reading)

      # 方式2：更精准的块内断言（避免外部上下文干扰）
      expect { service.call_to_reader }.to change { current_db_role }.from(:writing).to(:reading)
    end
  end
end

3. 测试环境注意事项 #

即使测试环境中primary和primary_replica指向同一数据库，依然可以通过角色标识完成验证——Rails会基于配置跟踪当前连接的角色，而非实际数据库实例。

关键细节 #

• Rails默认将primary数据库映射为writing角色，primary_replica映射为reading角色，无需额外配置。
• 若要验证代码块内的角色切换，使用expect { ... }.to change的方式更可靠，能确保角色切换发生在方法执行过程中。

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