Merge pull request #245 from noir-cr/with-headers-bug-fix

🐛 #244 Fixed bug in --with-headers

Author: hahwul

spec/unit_test/models/deliver_spec.cr

@@ -18,6 +18,12 @@ describe "Initialize" do
     object.headers["X-API-Key"].should eq("abcdssss")
   end
 
+  it "Deliver with headers (bearer case)" do
+    options[:send_with_headers] = "Authorization: Bearer gAAAAABl3qwaQqol243Np"
+    object = Deliver.new options
+    object.headers["Authorization"].should eq("Bearer gAAAAABl3qwaQqol243Np")
+  end
+
   it "Deliver with matchers" do
     options[:use_matchers] = "/admin"
     object = Deliver.new options

src/models/deliver.cr

@@ -26,7 +26,18 @@ class Deliver
         if header.includes? ":"
           @logger.debug "Adding '#{header}' to headers."
           splited = header.split(":")
-          @headers[splited[0]] = splited[1].gsub(/\s/, "")
+          value = ""
+          begin
+            if splited[1][0].to_s == " "
+              value = splited[1][1..-1].to_s
+            else
+              value = splited[1].to_s
+            end
+          rescue
+            value = splited[1].to_s
+          end
+
+          @headers[splited[0]] = value
         end
       end
       @logger.info_sub "#{@headers.size} headers added."