Improved error handling.

ioquatix · ioquatix · commit eb3b98426fff · 2024-09-11T13:26:00.000+12:00
diff --git a/test/async/websocket/client.rb b/test/async/websocket/client.rb
@@ -17,9 +17,13 @@
 					connection.write(message)
 				end
 				
-				connection.close
+				connection.shutdown
 			rescue Protocol::WebSocket::ClosedError
 				# Ignore this error.
+			rescue Errno::EPIPE
+				# Ignore this error: the client has closed the connection ungracefully.
+			ensure
+				connection.close
 			end or Protocol::HTTP::Response[404, {}, []]
 		end
 	end
