使用Rspec测试重试块

require 'rspec'
require 'retryable'

class ExampleService
  include Retryable

  retryable(:tries => 3, :on => [StandardError], :sleep_interval => 1) do
    def perform_action
      # 模拟可能失败的操作
      raise StandardError, "Temporary failure" if rand(2) == 1
      "Success"
    end
  end
end

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

  it "should succeed after retries" do
    expect(service.perform_action).to eq("Success")
  end

  it "should raise error if all retries fail" do
    allow_any_instance_of(ExampleService).to receive(:perform_action).and_raise(StandardError, "Permanent failure")
    expect { service.perform_action }.to raise_error(StandardError, "Permanent failure")
  end
end