Add the ability to set Content-ID header for ParamPart

socketry · Jun 26, 2019 · 475c97f · 475c97f
1 parent a7d238f
commit 475c97f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/parts.rb b/lib/parts.rb
@@ -52,6 +52,7 @@ def length
     def build_part(boundary, name, value, headers = {})
       part = ''
       part << "--#{boundary}\r\n"
+      part << "Content-ID: #{headers["Content-ID"]}\r\n" if headers["Content-ID"]
       part << "Content-Disposition: form-data; name=\"#{name.to_s}\"\r\n"
       part << "Content-Type: #{headers["Content-Type"]}\r\n" if headers["Content-Type"]
       part << "\r\n"

diff --git a/spec/parts_spec.rb b/spec/parts_spec.rb
@@ -99,4 +99,9 @@ def content_type; 'application/data'; end
   it "test_correct_length" do
     assert_part_length @part
   end
+
+  it "test_content_id" do
+    part = Parts::ParamPart.new("boundary", "with_content_id", "foobar", "Content-ID" => "id")
+    expect(part.to_io.read).to match(/Content-ID: id/)
+  end
 end