Rails中如何为含SQL别名的索引页编写RSpec测试？

问题根源 #

你在控制器里通过select添加的class_size是ActiveRecord动态生成的属性——只有当查询包含这个别名时，返回的Teacher实例才会拥有这个方法。测试报错通常是两种情况：要么测试时控制器没执行带该select的查询逻辑，要么测试数据未建立Teacher与Student的关联，导致查询返回的实例未绑定该属性。

具体解决步骤 #

1. 确保测试触发正确的控制器查询逻辑 #

先确认控制器的index方法是完整的（必须搭配group，否则count会统计所有学生的总数，而非单个老师的学生数）：

# app/controllers/teachers_controller.rb
def index
  @teachers = Teacher.joins(:students)
                     .select("teachers.*, count(students.id) as class_size")
                     .group("teachers.id")
end

编写RSpec测试时，需提前建好关联的测试数据，且用let!确保数据在请求前已持久化到数据库：

# spec/controllers/teachers_controller_spec.rb
RSpec.describe TeachersController, type: :controller do
  let(:teacher) { create(:teacher) }
  # 用let!强制立即创建数据，避免查询时数据未入库
  let!(:students) { create_list(:student, 3, teacher: teacher) }

  describe "GET #index" do
    it "returns teachers with valid class_size attribute" do
      get :index
      expect(assigns(:teachers).first.class_size).to eq(3)
    end
  end
end

2. 给Teacher模型添加虚拟属性（可选，提升稳定性） #

如果动态生成的方法在测试中偶尔不稳定，可以在模型里显式定义class_size的读写方法，这样就算不小心使用了普通的Teacher.all，调用class_size也只会返回nil，不会抛出错误：

# app/models/teacher.rb
class Teacher < ApplicationRecord
  has_many :students
  attr_reader :class_size

  def class_size=(value)
    @class_size = value.to_i
  end
end

3. 验证SQL查询是否包含别名（可选，排查复杂场景） #

若不确定控制器是否执行了正确的查询，可以在测试中检查SQL语句：

it "includes class_size in the select clause" do
  expect(Teacher).to receive(:joins).with(:students).and_call_original
  expect(Teacher).to receive(:select).with("teachers.*, count(students.id) as class_size").and_call_original
  get :index
end

关键注意事项 #

• 必须添加group("teachers.id")，否则count统计的是所有学生的总数，而非单个老师的学生数
• 测试数据必须建立Teacher与Student的关联，否则统计结果会是0或nil

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