Merge pull request #48 from varyform/master

Retry based on example metadata

lib/rspec/retry.rb

@@ -13,6 +13,9 @@ def self.setup
         config.add_setting :clear_lets_on_failure, :default => true
         config.add_setting :display_try_failure_messages, :default => false
 
+        # retry based on example metadata
+        config.add_setting :retry_count_condition, :default => ->(_) { nil }
+
         # If a list of exceptions is provided and 'retry' > 1, we only retry if
         # the exception that was raised by the example is NOT in that list. Otherwise
         # we ignore the 'retry' value and fail immediately.
@@ -56,6 +59,7 @@ def retry_count
           (
           ENV['RSPEC_RETRY_RETRY_COUNT'] ||
               ex.metadata[:retry] ||
+              RSpec.configuration.retry_count_condition.call(ex) ||
               RSpec.configuration.default_retry_count
           ).to_i,
           1

spec/lib/rspec/retry_spec.rb

@@ -56,6 +56,15 @@ class SharedError < StandardError; end
       end
     end
 
+    context 'with lambda condition' do
+      before(:all) { set_expectations([false, true]) }
+
+      it "should get retry count from condition call", retry_me_once: true do
+        expect(true).to be(shift_expectation)
+        expect(count).to eq(2)
+      end
+    end
+
     context 'with :retry => 0' do
       after(:all) { @@this_ran_once = nil }
       it 'should still run once', retry: 0 do

spec/spec_helper.rb

@@ -22,4 +22,6 @@
   config.around :each, :overridden do |ex|
     ex.run_with_retry retry: 3
   end
+
+  config.retry_count_condition = ->(example) { example.metadata[:retry_me_once] ? 2 : nil }
 end