Fix hound

ermaker · Aug 10, 2016 · 97dfce2 · 97dfce2
1 parent 012cdee
commit 97dfce2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/cool/retriable.rb b/lib/cool/retriable.rb
@@ -1,8 +1,7 @@
 module Cool
   # Retriable
   module Retriable
-    def retry_(new_method, on: StandardError, times: 3, &condition) # rubocop:disable Metrics/MethodLength, Metrics/LineLength
-      condition ||= ->(*) { false }
+    def retry_(new_method, on: StandardError, times: 3) # rubocop:disable Metrics/MethodLength, Metrics/LineLength
       orig_method = "__retry_orig_#{new_method}".to_sym
       alias_method orig_method, new_method
       define_method(new_method) do |*args, &block|
@@ -13,7 +12,7 @@ def retry_(new_method, on: StandardError, times: 3, &condition) # rubocop:disabl
             rescue on
               next
             end
-          return retval unless condition.call(retval)
+          return retval unless block_given? && yield(retval)
         end
         return method(orig_method).call(*args, &block)
       end