Add a convenient Reader#buffered! to buffer the body.

Author: ioquatix

lib/protocol/http/body/reader.rb

@@ -10,7 +10,7 @@ module Body
 			# General operations for interacting with a request or response body.
 			module Reader
 				# Read chunks from the body.
-				# @yield [String] read chunks from the body.
+				# @yields {|chunk| ...} chunks from the body.
 				def each(&block)
 					if @body
 						@body.each(&block)
@@ -19,7 +19,7 @@ def each(&block)
 				end
 
 				# Reads the entire request/response body.
-				# @return [String] the entire body as a string.
+				# @returns [String] the entire body as a string.
 				def read
 					if @body
 						buffer = @body.join
@@ -30,7 +30,7 @@ def read
 				end
 
 				# Gracefully finish reading the body. This will buffer the remainder of the body.
-				# @return [Buffered] buffers the entire body.
+				# @returns [Buffered] buffers the entire body.
 				def finish
 					if @body
 						body = @body.finish
@@ -40,6 +40,16 @@ def finish
 					end
 				end
 
+				# Buffer the entire request/response body.
+				# @returns [Readable] the buffered body.
+				def buffered!
+					if @body
+						@body = @body.finish
+						
+						return @body
+					end
+				end
+				
 				# Write the body of the response to the given file path.
 				def save(path, mode = ::File::WRONLY|::File::CREAT|::File::TRUNC, **options)
 					if @body

test/protocol/http/body/reader.rb

@@ -27,6 +27,12 @@ def initialize(body)
 		end
 	end
 
+	with '#buffered!' do
+		it 'buffers the body' do
+			expect(reader.buffered!).to be == body
+		end
+	end
+	
 	with '#close' do
 		it 'closes the underlying body' do
 			expect(body).to receive(:close)