Allow non-string value in Headers#[]=.

Author: ioquatix

lib/protocol/http/headers.rb

@@ -198,9 +198,18 @@ def extract(keys)
 			# @parameter key [String] the header key.
 			# @parameter value [String] the header value to assign.
 			def add(key, value)
-				self[key] = value
+				# The value MUST be a string, so we convert it to a string to prevent errors later on.
+				value = value.to_s
+				
+				if @indexed
+					merge_into(@indexed, key.downcase, value)
+				end
+				
+				@fields << [key, value]
 			end
 
+			alias []= add
+			
 			# Set the specified header key to the specified value, replacing any existing header keys with the same name.
 			#
 			# @parameter key [String] the header key to replace.
@@ -214,7 +223,7 @@ def set(key, value)
 			# Merge the headers into this instance.
 			def merge!(headers)
 				headers.each do |key, value|
-					self[key] = value
+					self.add(key, value)
 				end
 
 				return self
@@ -225,18 +234,6 @@ def merge(headers)
 				self.dup.merge!(headers)
 			end
 
-			# Append the value to the given key. Some values can be appended multiple times, others can only be set once.
-			#
-			# @parameter key [String] The header key.
-			# @parameter value [String] The header value.
-			def []= key, value
-				if @indexed
-					merge_into(@indexed, key.downcase, value)
-				end
-				
-				@fields << [key, value]
-			end
-			
 			# The policy for various headers, including how they are merged and normalized.
 			POLICY = {
 				# Headers which may only be specified once:

releases.md

@@ -3,6 +3,7 @@
 ## Unreleased
 
   - `Protocol::HTTP::Headers` now raise a `DuplicateHeaderError` when a duplicate singleton header (e.g. `content-length`) is added.
+  - `Protocol::HTTP::Headers#add` now coerces the value to a string when adding a header, ensuring consistent behaviour.
 
 ## v0.50.0
 

test/protocol/http/headers.rb

@@ -167,27 +167,34 @@
 	end
 
 	with "#[]=" do
-		it "can add field" do
+		it "can add field with a String value" do
+			headers["Content-Length"] = "1"
+			
+			expect(headers.fields.last).to be == ["Content-Length", "1"]
+			expect(headers["content-length"]).to be == "1"
+		end
+		
+		it "can add field with an Integer value" do
 			headers["Content-Length"] = 1
 
-			expect(headers.fields.last).to be == ["Content-Length", 1]
-			expect(headers["content-length"]).to be == 1
+			expect(headers.fields.last).to be == ["Content-Length", "1"]
+			expect(headers["content-length"]).to be == "1"
 		end
 
 		it "can add field with indexed hash" do
 			expect(headers.to_h).not.to be(:empty?)
 
-			headers["Content-Length"] = 1
-			expect(headers["content-length"]).to be == 1
+			headers["Content-Length"] = "1"
+			expect(headers["content-length"]).to be == "1"
 		end
 	end
 
 	with "#add" do
 		it "can add field" do
 			headers.add("Content-Length", 1)
 
-			expect(headers.fields.last).to be == ["Content-Length", 1]
-			expect(headers["content-length"]).to be == 1
+			expect(headers.fields.last).to be == ["Content-Length", "1"]
+			expect(headers["content-length"]).to be == "1"
 		end
 	end
 
@@ -241,7 +248,7 @@
 		it "can merge content-length" do
 			headers.merge!("content-length" => 2)
 
-			expect(headers["content-length"]).to be == 2
+			expect(headers["content-length"]).to be == "2"
 		end
 	end