Add .retriable feature to Http

lib/http/retriable/performer.rb

@@ -23,7 +23,7 @@ class Performer
         IOError
       ].freeze
 
-      RETRIABLE_STATUSES = [500].freeze
+      RETRIABLE_STATUSES = (500...Float::INFINITY).freeze
 
       # Default retry delay proc
       DELAY_PROC = ->(attempt) {
@@ -168,11 +168,20 @@ def out_of_retries_error(request, response, exception)
       end
 
       def build_retry_proc(exception_classes, retry_statuses)
+        retry_statuses = [retry_statuses].flatten
+
         ->(_req, err, res, _i) {
           if err
             exception_classes.any? { |e| err.is_a?(e) }
           else
-            retry_statuses.include?(res.status.to_i)
+            response_status = res.status.to_i
+            retry_statuses.any? do |matcher|
+              case matcher
+              when Range then matcher.include?(response_status)
+              when Numeric then matcher == response_status
+              else matcher.call(response_status)
+              end
+            end
           end
         }
       end

spec/lib/http/retriable/performer_spec.rb

@@ -73,13 +73,47 @@ def measure_wait
     end
 
     describe "expected status codes" do
+
+      def response(options = {})
+        HTTP::Response.new(
+          {
+            :status  => 200,
+            :version => "1.1",
+            :headers => {},
+            :body    => "Hello world!",
+            :uri     => "http://example.com/",
+            :request => request
+          }.merge(options)
+        )
+      end
+
       it "retries the request" do
         expect do
           perform(:retry_statuses => [200], :tries => 2)
         end.to raise_error HTTP::OutOfRetriesError
 
         expect(counter_spy).to eq 2
       end
+
+      describe "status codes can be expressed in many ways" do
+
+        [
+          301,
+          [200, 301, 485],
+          250...400,
+          [250...Float::INFINITY],
+          ->(status_code) { status_code == 301  },
+          [->(status_code) { status_code == 301  }]
+        ].each do |retry_statuses|
+          it retry_statuses.to_s do
+            expect do
+              perform(:retry_statuses => retry_statuses, :tries => 2) do
+                response(:status => 301)
+              end
+            end.to raise_error HTTP::OutOfRetriesError
+          end
+        end
+      end
     end
 
     describe "unexpected status code" do