Refactor doc polling as new method

lib/gini-api/client.rb

@@ -160,15 +160,9 @@ def upload(file, &block)
         # Start polling (0.5s) when document has been uploaded successfully
         if response.status == 201
           doc = Gini::Api::Document.new(self, response.headers['location'])
+          duration[:processing] = poll_document(doc, &block)
 
-          timeout(@processing_timeout) do
-            duration[:processing] = Benchmark.realtime do
-              doc.poll(&block)
-            end
-          end
-
-          # Combine duration values and update doc object
-          duration[:total] = duration[:upload] + duration[:processing]
+          duration[:total] = duration.values.inject(:+)
           doc.duration = duration
 
           doc
@@ -278,6 +272,22 @@ def resource_to_location(resource)
         )
       end
 
+      # Poll document and duration
+      #
+      # @param [Gini::Api::Document] doc Document instance to poll
+      #
+      # @return [Integer] Processing duration
+      #
+      def poll_document(doc, &block)
+        duration = 0
+        timeout(@processing_timeout) do
+          duration = Benchmark.realtime do
+            doc.poll(&block)
+          end
+        end
+        duration
+      end
+
       # Helper to upload document
       #
       # @param [String] file location of document to be uploaded