Fix handling of rewind after reading all input. (#24)

Author: ioquatix

lib/protocol/rack/input.rb

@@ -64,8 +64,10 @@ def rewind
 				if @body and @body.respond_to?(:rewind)
 					# If the body is not rewindable, this will fail.
 					@body.rewind
+					
 					@buffer = nil
 					@finished = false
+					@closed = false
 
 					return true
 				end
@@ -93,14 +95,18 @@ def read_next
 						# https://github.com/socketry/async-http/issues/183
 						if @body.empty?
 							@body.close
-							@body = nil
+							@closed = true
 						end
 
 						return chunk
 					else
-						# So if we are at the end of the stream, we close it automatically:
-						@body.close
-						@body = nil
+						unless @closed
+							# So if we are at the end of the stream, we close it automatically:
+							@body.close
+							@closed = true
+						end
+						
+						return nil
 					end
 				elsif @closed
 					raise IOError, "Stream is not readable, input has been closed!"

test/protocol/rack/input.rb

@@ -49,7 +49,17 @@
 				expect(input.read).to be == sample_data.join
 				expect(input.read).to be == ""
 
-				expect(input.body).to be_nil
+				expect(input).to be(:closed?)
+			end
+			
+			it "can rewind after reading all input" do
+				expect(input.read).to be == sample_data.join
+				expect(input).to be(:closed?)
+				
+				input.rewind
+				
+				expect(input).not.to be(:closed?)
+				expect(input.read).to be == sample_data.join
 			end
 
 			it "can read exactly the content length" do
@@ -69,13 +79,23 @@
 				expect(input.read(3)).to be == "row"
 			end
 
+			it "can rewind after reading partial input" do
+				expect(input.read(3)).to be == "The"
+				expect(input).not.to be(:closed?)
+				
+				input.rewind
+				
+				expect(input).not.to be(:closed?)
+				expect(input.read(3)).to be == "The"
+			end
+			
 			it "can read all input" do
 				expect(input.read(15)).to be == sample_data.join[0...15]
 				expect(input.read).to be == sample_data.join[15..-1]
 
 				expect(input.read(1)).to be == nil
 
-				expect(input.body).to be_nil
+				expect(input).to be(:closed?)
 			end
 
 			it "can read partial input with buffer" do