lib/calabash-android/wait_helpers.rb in calabash-android-0.5.9.pre2 vs lib/calabash-android/wait_helpers.rb in calabash-android-0.5.9

@@ -46,11 +46,11 @@
             sleep(retry_frequency)
             return screenshot_and_retry(msg, &block)
           else
             raise wait_error(msg)
           end
-        rescue Exception => e
+        rescue => e
           handle_error_with_options(e, nil, screenshot_on_error)
         end
       end
 
       def screenshot_and_retry(msg, &block)
@@ -114,17 +114,16 @@
           elements_arr.none? { |q| element_exists(q) }
         end
       end
 
       def handle_error_with_options(ex, timeout_message, screenshot_on_error)
-        msg = (timeout_message || ex)
-        if ex
-          msg = "#{msg} (#{ex.class})"
-        end
+        error_class = (ex && ex.class) || RuntimeError
+        error = error_class.new(timeout_message || ex.message)
+
         if screenshot_on_error
-          screenshot_and_raise msg
+          screenshot_and_raise error
         else
-          raise msg
+          raise error
         end
       end
 
       def wait_error(msg)
         (msg.is_a?(String) ? WaitError.new(msg) : msg)