State machine tests.

Author: ioquatix

lib/protocol/http1/connection.rb

@@ -109,6 +109,26 @@ def initialize(stream, persistent: true, state: :idle)
 			# State transition methods use a trailing "!".
 			attr_accessor :state
 
+			def idle?
+				@state == :idle
+			end
+			
+			def open?
+				@state == :open
+			end
+			
+			def half_closed_local?
+				@state == :half_closed_local
+			end
+			
+			def half_closed_remote?
+				@state == :half_closed_remote
+			end
+			
+			def closed?
+				@state == :closed
+			end
+			
 			# The number of requests processed.
 			attr :count
 

test/protocol/http1/connection.rb

@@ -547,4 +547,25 @@
 			end.to raise_exception(Protocol::HTTP1::BadHeader)
 		end
 	end
+	
+	with "persistent connection" do
+		it "returns back to idle state" do
+			expect(client).to be(:idle?)
+			
+			client.write_request("localhost", "GET", "/", "HTTP/1.1", {})
+			expect(client).to be(:open?)
+			client.write_body("HTTP/1.1", nil)
+			
+			expect(client).to be(:half_closed_local?)
+			
+			expect(server).to be(:idle?)
+			
+			request = server.read_request
+			expect(request).to be == ["localhost", "GET", "/", "HTTP/1.1", {}, nil]
+			expect(server).to be(:open?)
+			
+			server.write_response("HTTP/1.1", 200, {}, [])
+			expect(server).to be(:idle?)
+			
+	end
 end