pester 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f5c9151d53d657939d6506baf8a27e5da2744b3b
-  data.tar.gz: 251a6215cfe9cd275f37bf7df458561116cb7401
+  metadata.gz: c726cd4ef970107604a3cabf1d26ddf04fffbc45
+  data.tar.gz: 3579a4065f03f7791720e9d98d0c6bdc5a3c57f7
 SHA512:
-  metadata.gz: dd350670630f082f40af4ab8290e7eb5f8a75a2137ca61cacc63c72445c28baad2eda3bc595aec14bb2782bc3ac5778a38c5678661aaec77ab75fe9f1fcb7e9f
-  data.tar.gz: 4bd76629e817aca9981340564ec5426ab8e6c6619bbe222b6b7b0d33143e587f21c9b49a31d2dbfb87d9a06ac1863dcf192d550860cf7686ba56448dc38041e0
+  metadata.gz: 3e6af9161e42c15a613748d81845adfa8d786b4a5309b41163b87c5d42945968e984e66e84a95f9ba141b2a713c1c30f1a2d5c6b8b8648e516fc1cb0931b84d0
+  data.tar.gz: a5719117192311422b5bce432f1b53cef9d98f12ef3b4e1df6b244d237709353ce64af6152c015b75910226b010119abadd890cc23a5f15eeebf34c441b0f450

data/lib/pester/version.rb

@@ -1,3 +1,3 @@
 module Pester
-  VERSION = '0.1.0'
+  VERSION = '0.1.1'
 end

data/lib/pester.rb

@@ -26,6 +26,10 @@ module Pester
   # Options:
   #   retry_error_classes       - A single or array of exceptions to retry on. Thrown exceptions not in this list
   #                               (including parent/sub-classes) will be reraised
+  #   retry_error_messages      - A single or array of exception messages to retry on.  If only this options is passed,
+  #                               any exception with a message containing one of these strings will be retried.  If this
+  #                               option is passed along with retry_error_classes, retry will only happen when both the
+  #                               class and the message match the exception.  Strings and regexes are both permitted.
   #   reraise_error_classes     - A single or array of exceptions to always re-raiseon. Thrown exceptions not in
   #                               this list (including parent/sub-classes) will be retried
   #   max_attempts              - Max number of attempts to retry
@@ -33,18 +37,14 @@ module Pester
   #                               passed to retry_with_backoff will retry first after 2 seconds, then 4, then 6, et al.
   #   on_retry                  - A Proc to be called on each successive failure, before the next retry
   #   on_max_attempts_exceeded  - A Proc to be called when attempt_num >= max_attempts - 1
-  #   message                   - String or regex to look for in thrown exception messages. Matches will trigger retry
-  #                               logic, non-matches will cause the exception to be reraised
+  #   logger                    - Where to log the output
   #
   # Usage:
-  #   retry_action do
+  #   retry_action(retry_error_classes: [Mysql2::Error]) do
   #     puts 'trying to remove a directory'
   #     FileUtils.rm_r(directory)
   #   end
-  #
-  #  retryable(error_classes: Mysql2::Error, message: /^Lost connection to MySQL server/, max_attempts: 2) do
-  #    ActiveRecord::Base.connection.execute("LONG MYSQL STATEMENT")
-  #  end
+
   def self.retry_action(opts = {}, &block)
     merge_defaults(opts)
     if opts[:retry_error_classes] && opts[:reraise_error_classes]
@@ -53,16 +53,10 @@ module Pester
 
     opts[:max_attempts].times do |attempt_num|
       begin
-        result = yield block
-        return result
+        return yield(block)
       rescue => e
-        class_reraise = opts[:retry_error_classes] && !opts[:retry_error_classes].include?(e.class)
-        reraise_error = opts[:reraise_error_classes] && opts[:reraise_error_classes].include?(e.class)
-        message_reraise = opts[:message] && !e.message[opts[:message]]
-
-        if class_reraise || message_reraise || reraise_error
-          match_type = class_reraise ? 'class' : 'message'
-          opts[:logger].warn("Reraising exception from inside retry_action because provided #{match_type} was not matched.")
+        unless should_retry?(e, opts)
+          opts[:logger].warn('Reraising exception from inside retry_action.')
           raise
         end
 
@@ -72,6 +66,7 @@ module Pester
           opts[:logger].warn("Failure encountered: #{e}, backing off and trying again #{attempts_left} more times. Trace: #{trace}")
           opts[:on_retry].call(attempt_num, opts[:delay_interval])
         else
+          # Careful here because you will get back the return value of the on_max_attempts_exceeded proc!
           return opts[:on_max_attempts_exceeded].call(opts[:logger], opts[:max_attempts], e)
         end
       end
@@ -80,8 +75,29 @@ module Pester
 
   private
 
+  def self.should_retry?(e, opts = {})
+    retry_error_classes = opts[:retry_error_classes]
+    retry_error_messages = opts[:retry_error_messages]
+    reraise_error_classes = opts[:reraise_error_classes]
+
+    if retry_error_classes
+      if retry_error_messages
+        retry_error_classes.include?(e.class) && retry_error_messages.any? { |m| e.message[m] }
+      else
+        retry_error_classes.include?(e.class)
+      end
+    elsif retry_error_messages
+      retry_error_messages.any? { |m| e.message[m] }
+    elsif reraise_error_classes && reraise_error_classes.include?(e.class)
+      false
+    else
+      true
+    end
+  end
+
   def self.merge_defaults(opts)
     opts[:retry_error_classes]      = opts[:retry_error_classes] ? Array(opts[:retry_error_classes]) : nil
+    opts[:retry_error_messages]     = opts[:retry_error_messages] ? Array(opts[:retry_error_messages]) : nil
     opts[:reraise_error_classes]    = opts[:reraise_error_classes] ? Array(opts[:reraise_error_classes]) : nil
     opts[:max_attempts]             ||= 4
     opts[:delay_interval]           ||= 30

data/spec/pester_spec.rb

@@ -55,7 +55,6 @@ shared_examples 'raises an error only in the correct cases with a retry class' d
     let(:actual_error_message) { matching_error_message }
 
     it_has_behavior "doesn't raise an error"
-
     it_has_behavior 'returns and succeeds'
   end
 end
@@ -66,7 +65,6 @@ shared_examples 'raises an error only in the correct cases with a reraise class'
     let(:actual_error_message) { non_matching_error_message }
 
     it_has_behavior "doesn't raise an error"
-
     it_has_behavior 'returns and succeeds'
   end
 
@@ -88,7 +86,6 @@ describe 'retry_action' do
     let(:options) { { delay_interval: 0, logger: null_logger } }
 
     it_has_behavior "doesn't raise an error"
-
     it_has_behavior 'returns and succeeds'
   end
 
@@ -97,7 +94,6 @@ describe 'retry_action' do
     let(:options) { { max_attempts: 3, logger: null_logger } }
 
     it_has_behavior "doesn't raise an error"
-
     it_has_behavior 'returns and succeeds'
   end
 
@@ -111,17 +107,34 @@ describe 'retry_action' do
       it_has_behavior 'raises an error'
     end
 
-    context 'with on_max_attempts_exceeded specified (which does not raise)' do
-      let(:do_nothing_proc) { proc {} }
+    context 'with on_max_attempts_exceeded proc specified' do
       let(:options) do
         {
           max_attempts: max_attempts,
-          on_max_attempts_exceeded: do_nothing_proc,
+          on_max_attempts_exceeded: proc_to_call,
           logger: null_logger
         }
       end
 
-      it_has_behavior "doesn't raise an error"
+      context 'which does not do anything' do
+        let(:proc_to_call) { proc {} }
+        it_has_behavior "doesn't raise an error"
+      end
+
+      context 'which reraises' do
+        let(:proc_to_call) { Behaviors::WarnAndReraise }
+        it_has_behavior 'raises an error'
+      end
+
+      context 'which returns a value' do
+        let(:return_value) { 'return_value' }
+        let(:proc_to_call) { proc { return_value } }
+        it_has_behavior "doesn't raise an error"
+
+        it 'should return the result of the proc' do
+          expect(Pester.retry_action(options) { action }).to eq(return_value)
+        end
+      end
     end
   end
 
@@ -138,7 +151,7 @@ describe 'retry_action' do
       let(:options) do
         {
           retry_error_classes: expected_error_classes,
-          message: /^Lost connection to MySQL server/,
+          retry_error_messages: /^Lost connection to MySQL server/,
           max_attempts: 10,
           logger: null_logger
         }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pester
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Marc Bollinger
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-30 00:00:00.000000000 Z
+date: 2015-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler