Retry capybara tests on specific errors, fix error handler

test/support/retry_on_flaky_tests.rb

@@ -4,26 +4,27 @@
 #   prepend RetryOnFlakyTests[FlakyError, AnotherFlakyError, max_tries: 3]
 # end
 
-class RetryOnFlakyTests < Module
+module RetryOnFlakyTests
   def self.[](*error_classes, max_tries: 3)
     Module.new do
-      define_method :run do
-        tries ||= 1
-        super()
-      rescue *error_classes => err
-        raise unless (tries +=1) <= max_tries
+      define_method :max_tries do
+        max_tries
+      end
 
-        puts "Got #{err.class.name} Error, trying again #{tries}. time"
-        retry
+      define_method :error_classes do
+        error_classes
       end
 
-      define_method :capture_exceptions do |&block|
-        block.call
+      def capture_exceptions
+        tries ||= 1
+
+        yield
       rescue => err
-        if error_classes.include?(err.class)
-          raise
+        if (tries +=1) <= max_tries && error_classes.include?(err.class)
+          puts "Got #{err.message} Error, trying again #{tries}. time"
+          retry
         else
-          super(&->{ raise err })
+          super { raise err }
         end
       end
     end