class Cucumber::Filters::Retry

def retry_required?(test_case, event)
  event.test_case == test_case && event.result.failed? && test_case_counts[test_case] < configuration.retry_attempts
end

def test_case(test_case)
  configuration.on_event(:test_case_finished) do |event|
    next unless retry_required?(test_case, event)
    test_case_counts[test_case] += 1
    test_case.describe_to(receiver)
  end
  super
end

def test_case_counts
  @test_case_counts ||= Hash.new { |h, k| h[k] = 0 }
end