Head.for(nil) -> nil.

Author: samuel-williams-shopify

lib/protocol/http/body/head.rb

@@ -12,12 +12,24 @@ module Body
 			# Represents a body suitable for HEAD requests, in other words, a body that is empty and has a known length.
 			class Head < Readable
 				# Create a head body for the given body, capturing its length and then closing it.
-				def self.for(body)
-					head = self.new(body.length)
-					
-					body.close
+				#
+				# If a body is provided, the length is determined from the body, and the body is closed.
+				# If no body is provided, and the content length is provided, a head body is created with that length.
+				# This is useful for creating a head body when you only know the content length but not the actual body, which may happen in adapters for HTTP applications where the application may not provide a body for HEAD requests, but the content length is known.
+				#
+				# @parameter body [Readable | Nil] the body to create a head for.
+				# @parameter content_length [Integer | Nil] the content length of the body, if known.
+				# @returns [Head | Nil] the head body, or nil if the body is nil.
+				def self.for(body, content_length = nil)
+					if body
+						head = self.new(body.length)
+						body.close
+						return head
+					elsif content_length
+						return self.new(content_length)
+					end
 
-					return head
+					return nil
 				end
 
 				# Initialize the head body with the given length.

test/protocol/http/body/head.rb

@@ -55,5 +55,21 @@
 			expect(body).to have_attributes(length: be == 1)
 			body.close
 		end
+		
+		it "uses the content length if no body is provided" do
+			content_length = 42
+			head_body = subject.for(nil, content_length)
+			
+			expect(head_body).to have_attributes(length: be == content_length)
+			expect(head_body).to be(:empty?)
+			expect(head_body).to be(:ready?)
+		end
+	end
+	
+	with ".for with nil body" do
+		it "returns nil when body is nil" do
+			body = subject.for(nil)
+			expect(body).to be_nil
+		end
 	end
 end