From 9535ffa44cd44dfab633aa6ab9d071c65678d979 Mon Sep 17 00:00:00 2001
From: Tony Novak <tony@amitree.com>
Date: Fri, 20 Nov 2015 09:32:43 -0500
Subject: [PATCH] No longer swallowing errors from block passed to execute

Fixes #309
---
 lib/google/apis/core/http_command.rb       | 20 +++++++++++++-------
 spec/google/apis/core/http_command_spec.rb |  9 +++++----
 2 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/lib/google/apis/core/http_command.rb b/lib/google/apis/core/http_command.rb
index 4827f7d1f..ee8915815 100644
--- a/lib/google/apis/core/http_command.rb
+++ b/lib/google/apis/core/http_command.rb
@@ -91,7 +91,6 @@ module Google
         # @raise [Google::Apis::AuthorizationError] Authorization is required
         def execute(client)
           prepare!
-          proc = block_given? ? Proc.new : nil
           begin
             Retriable.retriable tries: options.retries + 1,
                                 base_interval: 1,
@@ -104,11 +103,19 @@ module Google
               Retriable.retriable tries: auth_tries,
                                   on: [Google::Apis::AuthorizationError],
                                   on_retry: proc { |*| refresh_authorization } do
-                return execute_once(client, &proc)
+                execute_once(client).tap do |result|
+                  if block_given?
+                    yield result, nil
+                  end
+                end
               end
             end
           rescue => e
-            raise e if proc.nil?
+            if block_given?
+              yield nil, e
+            else
+              raise e
+            end
           end
         ensure
           release!
@@ -244,12 +251,11 @@ module Google
         # @private
         # @param [Hurley::Client] client
         #   HTTP client
-        # @yield [result, err] Result or error if block supplied
         # @return [Object]
         # @raise [Google::Apis::ServerError] An error occurred on the server and the request can be retried
         # @raise [Google::Apis::ClientError] The request is invalid and should not be retried without modification
         # @raise [Google::Apis::AuthorizationError] Authorization is required
-        def execute_once(client, &block)
+        def execute_once(client)
           body.rewind if body.respond_to?(:rewind)
           begin
             logger.debug { sprintf('Sending HTTP %s %s', method, url) }
@@ -264,10 +270,10 @@ module Google
             logger.debug { response.status_code }
             logger.debug { response.inspect }
             response = process_response(response.status_code, response.header, response.body)
-            success(response, &block)
+            success(response)
           rescue => e
             logger.debug { sprintf('Caught error %s', e) }
-            error(e, rethrow: true, &block)
+            error(e, rethrow: true)
           end
         end
 
diff --git a/spec/google/apis/core/http_command_spec.rb b/spec/google/apis/core/http_command_spec.rb
index d7a7f6d6c..c9445f791 100644
--- a/spec/google/apis/core/http_command_spec.rb
+++ b/spec/google/apis/core/http_command_spec.rb
@@ -146,10 +146,7 @@ RSpec.describe Google::Apis::Core::HttpCommand do
 
     context('with callbacks') do
       it 'should return the response body after retries' do
-        expect { |b| command.execute(client, &b) }.to yield_successive_args(
-          [nil, an_instance_of(Google::Apis::ServerError)],
-          [nil, an_instance_of(Google::Apis::ServerError)],
-          ['Hello world', nil])
+        expect { |b| command.execute(client, &b) }.to yield_with_args('Hello world', nil)
       end
     end
   end
@@ -254,6 +251,10 @@ RSpec.describe Google::Apis::Core::HttpCommand do
     it 'should call block if present' do
       expect { |b| command.execute(client, &b) }.to yield_with_args(nil, an_instance_of(Google::Apis::ClientError))
     end
+
+    it 'should not swallow errors raised in block' do
+      expect { command.execute(client) { raise "Potatoes detected in tailpipe" } }.to raise_error("Potatoes detected in tailpipe")
+    end
   end
 
   it 'should send repeated query parameters' do