Head.for(nil) -> nil.

Author: samuel-williams-shopify

lib/protocol/http/body/head.rb

@@ -12,12 +12,17 @@ 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.
+				#
+				# @parameter body [Readable | Nil] the body to create a head for.
+				# @returns [Head | Nil] the head body, or nil if the body is nil.
 				def self.for(body)
-					head = self.new(body.length)
-					
-					body.close
+					if body
+						head = self.new(body.length)
+						body.close
+						return head
+					end
 
-					return head
+					return nil
 				end
 
 				# Initialize the head body with the given length.

test/protocol/http/body/head.rb

@@ -56,4 +56,11 @@
 			body.close
 		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