Expose Body::Writable#count.

Author: samuel-williams-shopify

lib/protocol/http/body/writable.rb

@@ -28,6 +28,9 @@ def initialize(length = nil, queue: Thread::Queue.new)
 				# @attribute [Integer] The length of the response body if known.
 				attr :length
 
+				# @attribute [Integer] The number of chunks written to the body.
+				attr :count
+				
 				# Stop generating output; cause the next call to write to fail with the given error. Does not prevent existing chunks from being read. In other words, this indicates both that no more data will be or should be written to the body.
 				#
 				# @parameter error [Exception] The error that caused this body to be closed, if any. Will be raised on the next call to {read}.

releases.md

@@ -1,5 +1,9 @@
 # Releases
 
+## Unreleased
+
+  - Expose `Protocol::HTTP::Body::Writable#count` attribute to provide access to the number of chunks written to the body.
+
 ## v0.59.0
 
   - Introduce `Protocol::HTTP::Middleware.load` method for loading middleware applications from files.

test/protocol/http/body/writable.rb

@@ -18,6 +18,45 @@
 		end
 	end
 
+	with "#count" do
+		it "should be zero by default" do
+			expect(body.count).to be == 0
+		end
+		
+		it "should increment when chunks are written" do
+			expect(body.count).to be == 0
+			
+			body.write("Hello")
+			expect(body.count).to be == 1
+			
+			body.write("World")
+			expect(body.count).to be == 2
+			
+			body.write("")
+			expect(body.count).to be == 3
+		end
+		
+		it "should not increment when reading chunks" do
+			body.write("Hello")
+			body.write("World")
+			
+			expect(body.count).to be == 2
+			
+			body.read
+			body.read
+			
+			expect(body.count).to be == 2
+		end
+		
+		it "should not change when body is closed" do
+			body.write("Hello")
+			expect(body.count).to be == 1
+			
+			body.close
+			expect(body.count).to be == 1
+		end
+	end
+	
 	with "#closed?" do
 		it "should not be closed by default" do
 			expect(body).not.to be(:closed?)