Improved error handling.

socketry · Sep 11, 2024 · eb3b984 · eb3b984
1 parent 5b5288d
commit eb3b984
Showing 1 changed file with 5 additions and 1 deletion.
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