Add tests.

Author: ioquatix

lib/protocol/rack/body.rb

@@ -6,7 +6,9 @@
 require_relative "body/streaming"
 require_relative "body/enumerable"
 require_relative "constants"
+
 require "protocol/http/body/completable"
+require "protocol/http/body/head"
 
 module Protocol
 	module Rack

test/protocol/rack/body.rb

@@ -18,8 +18,8 @@
 	end
 
 	with "#wrap" do
-		let(:env) { {} }
-		let(:headers) { {} }
+		let(:env) {Hash.new}
+		let(:headers) {Hash.new}
 
 		it "handles nil body" do
 			expect(Console).to receive(:warn).and_return(nil)
@@ -28,6 +28,43 @@
 			expect(result).to be_nil
 		end
 
+		it "handles head request with content-length and empty body" do
+			# Rack::Head will result in an enumerable body, and currently does not set content-length either.
+			# So, it's up to the application to set the content-length header if they use `Rack::Head`.
+			# Otherwise, we should basically ignore the body but retain the length if known/provided.
+			headers["content-length"] = "123"
+			
+			result = subject.wrap(env, 200, headers, [], nil, true)
+			
+			expect(result).to be_a(Protocol::HTTP::Body::Head)
+			expect(result.length).to be == 123
+		end
+		
+		it "handles head request with no content-length and empty body" do
+			result = subject.wrap(env, 200, headers, [], nil, true)
+			
+			expect(result).to be_a(Protocol::HTTP::Body::Head)
+			expect(result.length).to be == 0
+		end
+		
+		it "handles head request with content-length and nil body" do
+			headers["content-length"] = "123"
+			
+			expect(Console).to receive(:warn).and_return(nil)
+			result = subject.wrap(env, 200, headers, nil, nil, true)
+			
+			expect(result).to be_a(Protocol::HTTP::Body::Head)
+			expect(result.length).to be == 123
+		end
+		
+		it "handles head request with no content-length and nil body" do
+			expect(Console).to receive(:warn).and_return(nil)
+			
+			result = subject.wrap(env, 200, headers, nil, nil, true)
+			
+			expect(result).to be_nil
+		end
+		
 		with "non-empty body and no-content status" do
 			let(:mock_body) do
 				Protocol::HTTP::Body::Buffered.new(["content"])