lib/httpi/request.rb in httpi-2.0.1 vs lib/httpi/request.rb in httpi-2.0.2

@@ -76,12 +76,18 @@
     # Adds a header information to accept gzipped content.
     def gzip
       headers["Accept-Encoding"] = "gzip,deflate"
     end
 
-    # Sets the cookies from a given +http_response+.
-    def set_cookies(http_response)
-      cookie_store.add *http_response.cookies
+    # Sets the cookies from an object responding to `cookies` (e.g. `HTTPI::Response`)
+    # or an Array of `HTTPI::Cookie` objects.
+    def set_cookies(object_or_array)
+      if object_or_array.respond_to?(:cookies)
+        cookie_store.add *object_or_array.cookies
+      else
+        cookie_store.add *object_or_array
+      end
+
       cookies = cookie_store.fetch
       headers["Cookie"] = cookies if cookies
     end
 
     attr_accessor :open_timeout, :read_timeout