如何减少RSpec错误处理测试中的代码重复？

场景说明 #

我有一个可能触发错误的方法some_cool_method，当内部逻辑抛出异常时，会先执行日志记录、错误处理等逻辑，再重新抛出SpecificException。代码示例如下：

def some_cool_method
  error_prone_logic

rescue StandardError => e
  logging_methods
  
  error_handling

  raise SpecificException.new
end

在RSpec测试中，我通过Stub让error_prone_logic依赖的实体抛出异常：

before do
  allow(some_entity_double).to receive(:work).and_raise(StandardError, 'RIP')
end

当前问题：每个测试用例（无论是验证异常类型、日志调用还是错误处理逻辑）都必须添加expect { subject }.to raise_error(SpecificException)断言，否则测试会因未捕获的异常而失败，导致代码重复。

优化方案 #

当然可以优化，下面提供几种实用的解决思路：

1. 手动捕获异常（简单直接） #

在不需要验证异常的测试用例中，用rescue块包裹方法调用，忽略SpecificException，只关注当前测试的逻辑点：

it '调用日志记录方法' do
  expect(subject).to receive(:logging_methods)
  begin
    subject
  rescue SpecificException
    # 仅捕获目标异常，不影响测试断言
  end
end

it '执行错误处理逻辑' do
  expect(subject).to receive(:error_handling)
  begin
    subject
  rescue SpecificException
    # 忽略异常
  end
end

# 仅保留一个验证异常的测试用例
it '抛出SpecificException异常' do
  expect { subject }.to raise_error(SpecificException)
end

2. 使用Around钩子自动处理异常 #

通过RSpec的around钩子，给除了异常验证用例外的所有测试自动添加异常捕获逻辑，避免重复代码：

describe '#some_cool_method' do
  let(:some_entity_double) { double('SomeEntity') }

  before do
    allow(some_entity_double).to receive(:work).and_raise(StandardError, 'RIP')
  end

  # 单独验证异常的用例
  it '抛出SpecificException异常', :exception_test do
    expect { subject }.to raise_error(SpecificException)
  end

  # 自动为其他用例处理异常断言
  around(:example) do |example|
    if example.metadata[:exception_test]
      example.run
    else
      # 执行测试逻辑时自动捕获并验证异常
      expect { example.run }.to raise_error(SpecificException)
    end
  end

  it '调用日志记录方法' do
    expect(subject).to receive(:logging_methods)
    subject
  end

  it '执行错误处理逻辑' do
    expect(subject).to receive(:error_handling)
    subject
  end
end

3. 利用共享上下文复用断言 #

把异常断言逻辑封装到共享上下文中，在需要的测试组中引入，减少重复：

RSpec.shared_context '自动处理异常断言' do
  before do
    # 先设置方法调用的断言，再执行并验证异常
    yield if block_given?
    expect { subject }.to raise_error(SpecificException)
  end
end

describe '#some_cool_method' do
  let(:some_entity_double) { double('SomeEntity') }

  before do
    allow(some_entity_double).to receive(:work).and_raise(StandardError, 'RIP')
  end

  it '抛出SpecificException异常' do
    expect { subject }.to raise_error(SpecificException)
  end

  describe '错误处理流程' do
    include_context '自动处理异常断言' do
      it '调用日志记录方法' do
        expect(subject).to receive(:logging_methods)
      end

      it '执行错误处理逻辑' do
        expect(subject).to receive(:error_handling)
      end
    end
  end
end

总结 #

以上三种方式都能实现只保留一个异常验证用例，其余用例专注于各自的逻辑点。其中手动捕获异常适合小型测试套件，Around钩子和共享上下文更适合规模较大的测试场景，能进一步减少代码冗余。

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