Add tests for inflation failures.

Author: ioquatix

lib/protocol/http/body/inflate.rb

@@ -23,17 +23,20 @@ def read
 					return if @stream.finished?
 
 					# The stream might have been closed while waiting for the chunk to come in.
-					if chunk = super
+					while chunk = super
 						@input_length += chunk.bytesize
 
 						# It's possible this triggers the stream to finish.
 						chunk = @stream.inflate(chunk)
 
+						break unless chunk&.empty?
+					end
+					
+					if chunk
+						@output_length += chunk.bytesize
+					elsif !@stream.closed?
+						chunk = @stream.finish
 						@output_length += chunk.bytesize
-					# elsif [email protected]?
-					# 	chunk = @stream.finish
-						
-					# 	@output_length += chunk.bytesize
 					end
 
 					if chunk.empty? and @stream.finished?

test/protocol/http/body/inflate.rb

@@ -0,0 +1,35 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2019-2023, by Samuel Williams.
+
+require 'protocol/http/body/buffered'
+require 'protocol/http/body/deflate'
+require 'protocol/http/body/inflate'
+
+require 'securerandom'
+
+describe Protocol::HTTP::Body::Inflate do
+	let(:sample) {"The quick brown fox jumps over the lazy dog."}
+	let(:chunks) {[sample] * 1024}
+	
+	let(:body) {Protocol::HTTP::Body::Buffered.for(chunks)}
+	let(:deflate_body) {Protocol::HTTP::Body::Deflate.for(body)}
+	let(:compressed_chunks) {deflate_body.join.each_char.to_a}
+	let(:compressed_body_chunks) {compressed_chunks}
+	let(:compressed_body) {Protocol::HTTP::Body::Buffered.new(compressed_body_chunks)}
+	let(:decompressed_body) {subject.for(compressed_body)}
+	
+	it "can decompress a body" do
+		expect(decompressed_body.join).to be == chunks.join
+	end
+	
+	with "incomplete input" do
+		let(:compressed_body_chunks) {compressed_chunks.first(compressed_chunks.size/2)}
+		
+		it "raises error when input is incomplete" do
+			expect{decompressed_body.join}.to raise_exception(Zlib::BufError)
+		end
+	end
+end