Improve connection handling in tests.

ioquatix · ioquatix · commit 7a2887498791 · 2024-09-11T13:21:24.000+12:00
diff --git a/lib/async/websocket/connection.rb b/lib/async/websocket/connection.rb
@@ -42,6 +42,10 @@ def reusable?
 			end
 			
 			attr :protocol
+			
+			def inspect
+				"#<#{self.class} state=#{@state}>"
+			end
 		end
 	end
 end
diff --git a/test/async/websocket/server.rb b/test/async/websocket/server.rb
@@ -19,6 +19,8 @@
 		message = connection.read
 		expect(message.to_str).to be == "Hello World!"
 		
+		connection.shutdown
+	ensure
 		connection.close
 	end
 	
@@ -27,6 +29,8 @@
 			connection.send_text("Hello World!")
 			message = connection.read
 			expect(message.to_str).to be == "Hello World!"
+			
+			connection.shutdown
 		end
 	end
 	
@@ -36,6 +40,8 @@
 				connection.send_text("Hello World!")
 				message = connection.read
 				expect(message.to_str).to be == "Hello World!"
+				
+				connection.shutdown
 			end
 		end
 	end
