RSpec：同一上下文多测试场景如何避免重复初始化

我完全懂你的痛点——重复初始化耗时的上下文会拖慢测试速度，但用before(:all)又会碰到stubs失效的问题。下面是几个经过实践验证的可行方案，帮你平衡测试速度和可靠性：

1. 拆分初始化：before(:context)处理昂贵资源 + before(:each)设置Stubs #

这是最常用的方案，核心思路是把不依赖stub、初始化成本高的对象放在before(:context)里只初始化一次，而stubs和消息期望放在before(:each)里每个示例重新设置（因为RSpec会在示例间自动重置mock状态）。

示例代码：

RSpec.describe OrderProcessor do
  # 只执行一次：初始化耗时的数据库对象、外部资源连接等
  before(:context) do
    @large_product_catalog = ProductCatalog.create!(import_huge_csv: 'path/to/big-data.csv')
    @external_service_connection = ExternalInventoryService.new(heavy_config: {})
  end

  # 每个示例执行：重置stub，避免mock状态污染
  before(:each) do
    # 针对外部服务的stub，每个示例重新设置
    allow(@external_service_connection).to receive(:check_stock).and_return(available: true)
    # 针对类方法的stub同理
    allow(NotificationService).to receive(:send_alert).and_return(true)
  end

  it 'successfully processes orders with in-stock items' do
    processor = described_class.new(@large_product_catalog, @external_service_connection)
    result = processor.process(Order.new(item_ids: [1, 2]))
    expect(result.status).to eq(:completed)
  end

  it 'logs a warning when processing empty orders' do
    expect(Rails.logger).to receive(:warn).with('Empty order received')
    processor = described_class.new(@large_product_catalog, @external_service_connection)
    processor.process(Order.new(item_ids: []))
  end
end

关键注意事项： #

• 放在before(:context)里的对象必须是不可变的，或者在before(:each)里手动重置状态，否则示例间的修改会互相污染。比如如果@large_product_catalog会被某个示例修改，要在before(:each)里调用@large_product_catalog.reload或者自定义的重置方法。
• 绝对不要在before(:context)里设置stub/期望，因为RSpec的mock框架会在每个示例启动时清空所有mock状态，这些stub会直接失效。

2. 使用let!配合memoization（适用于中等成本的上下文） #

如果你的上下文初始化成本没那么极端，但还是想避免重复执行，可以用let!（强制立即执行）结合自定义的memoization逻辑，这种方式更符合RSpec的惯用写法：

RSpec.describe InvoiceGenerator do
  # 用实例变量实现跨示例的memoization，只创建一次customer
  let!(:customer) do
    @customer ||= Customer.create!(heavy_profile_data: '...')
  end

  before(:each) do
    allow(TaxCalculator).to receive(:calculate_rate).and_return(0.08)
  end

  it 'generates invoices with correct tax' do
    invoice = described_class.generate(customer, amount: 100)
    expect(invoice.tax_amount).to eq(8.0)
  end

  it 'includes customer details in invoice metadata' do
    invoice = described_class.generate(customer, amount: 50)
    expect(invoice.metadata[:customer_id]).to eq(customer.id)
  end
end

这种方式下，customer只会被创建一次，后续示例复用同一个实例，同时stubs还是每个示例重新设置，保证了测试隔离性。

3. 使用RSpec的shared_context复用初始化逻辑 #

如果多个测试组需要复用同一个昂贵的上下文，可以把初始化逻辑封装到shared_context里，然后在需要的地方引入，避免重复代码：

RSpec.shared_context 'expensive test resources' do
  before(:context) do
    @global_product_db = ProductDB.setup(import_large_dataset: true)
  end

  before(:each) do
    allow(PaymentGateway).to receive(:charge).and_return(success: true)
  end
end

RSpec.describe OrderChecker do
  include_context 'expensive test resources'

  it 'validates orders against product database' do
    checker = described_class.new(@global_product_db)
    expect(checker.valid_order?(Order.new(item_id: 999))).to be_falsey
  end
end

RSpec.describe OrderDiscounter do
  include_context 'expensive test resources'

  it 'applies discounts for bulk orders' do
    discounter = described_class.new(@global_product_db)
    discount = discounter.calculate(Order.new(item_ids: Array(1..100)))
    expect(discount).to be >= 0.15
  end
end

最后要提醒的是，虽然复用上下文能提升速度，但一定要注意测试的隔离性——如果共享对象的状态被示例修改，会导致测试变得不可靠，所以务必确保共享上下文的纯净性。

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