Merge 6a89775 into 38c6892

lib/yt/http_request.rb

@@ -50,7 +50,18 @@ def run
           parse_response!
         end
       else
+        retry_run
+      end
+    end
+
+    def retry_run
+      if @retry
         raise Yt::HTTPError, error_message
+      else
+        @retry = true
+        @response = nil
+        sleep 10
+        run
       end
     end
 

spec/http_request_spec.rb

@@ -32,4 +32,19 @@
       expect{request.run}.to raise_error Yt::ConnectionError
     end
   end
+
+  context 'given that YouTube API raises an error' do
+    context 'only once' do
+      path = '/discovery/v1/apis/youtube/v3/rest'
+      headers = {'User-Agent' => 'Yt::HTTPRequest'}
+      params = {verbose: 1}
+      let(:request) { Yt::HTTPRequest.new path: path, headers: headers, params: params }
+      let(:retry_response) { Net::HTTPSuccess.new '1.0', '200', 'OK' }
+      before { expect(Net::HTTP).to receive(:start).at_least(:once).and_return retry_response }
+      it 'returns the HTTP response with the JSON-parsed body' do
+        allow(retry_response).to receive(:body).and_return('{"fake": "body"}')
+        expect { request.run }.not_to raise_error
+      end
+    end
+  end
 end