Make Status comparable.

Author: ioquatix

lib/protocol/grpc/header/message.rb

@@ -14,6 +14,10 @@ module Header
 			# This header is optional and typically only present when there's an error (non-zero status code).
 			# This header can appear both as an initial header (for trailers-only responses) and as a trailer.
 			class Message < String
+				# Parse a message from a header value.
+				#
+				# @parameter value [String] The header value to parse.
+				# @returns [Message] A new Message instance.
 				def self.parse(value)
 					new(value)
 				end

lib/protocol/grpc/header/status.rb

@@ -12,6 +12,10 @@ module Header
 			# Status code 0 indicates success (OK), while other codes indicate various error conditions.
 			# This header can appear both as an initial header (for trailers-only responses) and as a trailer.
 			class Status
+				# Parse a status code from a header value.
+				#
+				# @parameter value [String] The header value to parse.
+				# @returns [Status] A new Status instance.
 				def self.parse(value)
 					new(value.to_i)
 				end
@@ -37,6 +41,30 @@ def to_s
 					@value.to_s
 				end
 
+				# Check equality with another status or integer value.
+				#
+				# @parameter other [Status | Integer] The value to compare with.
+				# @returns [Boolean] if the status codes are equal.
+				def ==(other)
+					@value == other.to_i
+				end
+				
+				alias eql? ==
+				
+				# Generate hash for use in Hash/Set collections.
+				#
+				# @returns [Integer] The hash value based on the status code.
+				def hash
+					@value.hash
+				end
+				
+				# Check if this status represents success (status code 0).
+				#
+				# @returns [Boolean] `true` if the status code is 0 (OK).
+				def ok?
+					@value == 0
+				end
+				
 				# Merge another status value (takes the new value, as status should only appear once)
 				# @parameter value [String | Integer] The new status code
 				def <<(value)

test/protocol/grpc/header/status.rb

@@ -49,6 +49,69 @@
 		end
 	end
 
+	with "#==" do
+		it "compares equal status instances" do
+			status1 = subject.new(0)
+			status2 = subject.new(0)
+			expect(status1).to be == status2
+		end
+		
+		it "compares unequal status instances" do
+			status1 = subject.new(0)
+			status2 = subject.new(1)
+			expect(status1).not.to be == status2
+		end
+		
+		it "compares status with integer" do
+			status = subject.new(5)
+			expect(status).to be == 5
+		end
+		
+		it "compares status with different integer" do
+			status = subject.new(5)
+			expect(status).not.to be == 3
+		end
+	end
+	
+	with "#eql?" do
+		it "works same as ==" do
+			status1 = subject.new(0)
+			status2 = subject.new(0)
+			expect(status1.eql?(status2)).to be == true
+		end
+	end
+	
+	with "#hash" do
+		it "returns consistent hash for same status code" do
+			status1 = subject.new(0)
+			status2 = subject.new(0)
+			expect(status1.hash).to be == status2.hash
+		end
+		
+		it "can be used as hash key" do
+			status = subject.new(0)
+			hash = {status => "success"}
+			expect(hash[subject.new(0)]).to be == "success"
+		end
+	end
+	
+	with "#ok?" do
+		it "returns true for status code 0" do
+			status = subject.new(0)
+			expect(status).to be(:ok?)
+		end
+		
+		it "returns false for non-zero status code" do
+			status = subject.new(1)
+			expect(status.ok?).to be == false
+		end
+		
+		it "returns false for error status" do
+			status = subject.new(14)
+			expect(status.ok?).to be == false
+		end
+	end
+	
 	with "#<<" do
 		it "merges new integer value" do
 			status = subject.new(0)