retry on known error # ruby

error_handler.rb

module Error
  module ErrorHandler
    def retry_on_known_error(retry_limit:, known_errors: [], &block)
      retry_count = 0

      block.call
    rescue *known_errors => e
      raise e if retry_count == retry_limit

      retry_count += 1

      retry
    end

    module_function :retry_on_known_error
  end
end

error_handler_spec.rb

# frozen_string_literal: true

require 'rails_helper'

class TestError < StandardError
end

class RetryMock
  attr_reader :retry_count

  def initialize
    @retry_count = 0
  end

  def try
    @retry_count += 1

    raise TestError if @retry_count != 3
  end
end

RSpec.describe Error::ErrorHandler do
  describe '#retry_on_known_error' do
    context 'when is raised a unknowed error' do
      it "don't retry" do
        expect do
          subject.retry_on_known_error(retry_limit: 3, known_errors: []) do
            raise TestError
          end
        end.to raise_error(TestError)
      end
    end

    context 'when is raised a knowed error' do
      it 'retry until retry limit' do
        retry_mock = RetryMock.new

        subject.retry_on_known_error(retry_limit: 3, known_errors: [TestError]) do
          retry_mock.try
        end

        expect(retry_mock.retry_count).to eq(3)
      end
    end
  end
end